HEX
Server: LiteSpeed
System: Linux php-prod-1.spaceapp.ru 5.15.0-157-generic #167-Ubuntu SMP Wed Sep 17 21:35:53 UTC 2025 x86_64
User: xnsbb3110 (1041)
PHP: 8.1.33
Disabled: NONE
$ pwd: /usr/local/CyberPanel/lib/python3.10/site-packages/pydantic/__pycache__/
Upload Files
File: //usr/local/CyberPanel/lib/python3.10/site-packages/pydantic/__pycache__/mypy.cpython-310.pyc
o

��h���@s>dZddlmZddlZddlmZddlmZddlm	Z	m
Z
ddlmZddl
mZmZdd	lmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0dd
l1m2Z2ddl3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9ddl:m;Z;dd
l<m=Z=ddl>m?Z?ddl@mAZAddlBmCZCddlDmEZEddlFmGZGmHZHmIZImJZJmKZKmLZLmMZMmNZNmOZOmPZPddlQmRZRddlSmTZTddlUmVZWddlXmYZYddlZm[Z[dZ\dZ]dZ^dZ_dZ`dZadZbdZcd Zdhd!�Zeeed"hZfe[eW�Zgd#Zhd$ZVdxd)d*�ZiGd+d,�d,e7�ZjGd-d.�d.�Zkdyd2d3�ZlGd4d5�d5�ZmGd6d7�d7�ZnGd8d9�d9�ZoGd:d;�d;eC�ZpGd<d=�d=�ZqdzdAdB�ZredCdDdE�ZsedFdGdE�ZtedHdIdE�ZuedJdKdE�ZvedLdMdE�ZwedLdNdE�ZxedLdOdE�Zyd{dVdW�Zzd|dZd[�Z{d}d\d]�Z|d~d`da�Z}d}dbdc�Z~dddde�Z			fd�d�drds�Z�d�dvdw�Z�dS)�z^This module includes classes and functions designed specifically for use with the mypy plugin.�)�annotationsN)�Iterator)�ConfigParser)�Any�Callable)�	ErrorCode)�expand_type�expand_type_by_instance) �	ARG_NAMED�
ARG_NAMED_OPT�ARG_OPT�ARG_POS�	ARG_STAR2�	INVARIANT�MDEF�Argument�AssignmentStmt�Block�CallExpr�ClassDef�Context�	Decorator�DictExpr�EllipsisExpr�
Expression�FuncDef�IfStmt�JsonDict�
MemberExpr�NameExpr�PassStmt�PlaceholderNode�RefExpr�	Statement�StrExpr�SymbolTableNode�TempNode�	TypeAlias�TypeInfo�Var)�Options)�CheckerPluginInterface�ClassDefContext�
MethodContext�Plugin�ReportConfigContext�SemanticAnalyzerPluginInterface)�deserialize_and_fixup_type)�set_callable_name)�make_wildcard_trigger)�state)�TypeTranslator)�map_type_from_supertype)
�AnyType�CallableType�Instance�NoneType�Type�	TypeOfAny�TypeType�TypeVarType�	UnionType�get_proper_type)�
fill_typevars)�get_unique_redefinition_name)�__version__)�_fields)�parse_mypy_version�
pydantic-mypyzpydantic-mypy-metadatazpydantic.main.BaseModelz#pydantic_settings.main.BaseSettingszpydantic.root_model.RootModelz5pydantic._internal._model_construction.ModelMetaclasszpydantic.fields.Fieldzpydantic.dataclasses.dataclass�.pydantic.functional_validators.model_validator>�*pydantic.functional_serializers.serializer�.pydantic.deprecated.class_validators.validator�.pydantic.functional_validators.field_validator�3pydantic.deprecated.class_validators.root_validatorrG�0pydantic.functional_serializers.model_serializerrL�builtins��version�str�return�type[Plugin]cCstS)a!`version` is the mypy version string.

    We might want to use this to print a warning if the mypy version being used is
    newer, or especially older, than we expect (or need).

    Args:
        version: The mypy version string.

    Return:
        The Pydantic mypy plugin type.
    )�PydanticPlugin)rO�rT�C/usr/local/CyberPanel/lib/python3.10/site-packages/pydantic/mypy.py�pluginmsrVcs^eZdZdZd�fdd�Zddd�Zdd
d�Zddd�Zddd�Zd dd�Z	d dd�Z
�ZS)!rSzThe Pydantic mypy plugin.�optionsr*rQ�Nonecs&t|�|_|j��|_t��|�dS�N)�PydanticPluginConfig�
plugin_config�to_data�_plugin_data�super�__init__)�selfrW��	__class__rTrUr_s
zPydanticPlugin.__init__�fullnamerP�(Callable[[ClassDefContext], None] | NonecCs0|�|�}|rt|jt�r|j�t�r|jSdS)zUpdate Pydantic model class.N)�lookup_fully_qualified�
isinstance�noder(�has_base�BASEMODEL_FULLNAME�$_pydantic_model_class_maker_callback)r`rc�symrTrTrU�get_base_class_hook�s

z"PydanticPlugin.get_base_class_hookcCs|tkr|jSdS)z,Update Pydantic `ModelMetaclass` definition.N)�MODEL_METACLASS_FULLNAME�)_pydantic_model_metaclass_marker_callback�r`rcrTrTrU�get_metaclass_hook�sz!PydanticPlugin.get_metaclass_hook�&Callable[[MethodContext], Type] | NonecCs|�d�rtSdS)z-Adjust return type of `from_orm` method call.z	.from_ormN)�endswith�from_attributes_callbackrorTrTrU�get_method_hook�s
zPydanticPlugin.get_method_hook�ctxr/�dict[str, Any]cCs|jS)zjReturn all plugin config data.

        Used by mypy to determine if cache needs to be discarded.
        )r])r`rurTrTrU�report_config_data�sz!PydanticPlugin.report_config_datar,cCs"t|j|j|j|j�}|��dSrY)�PydanticModelTransformer�cls�reason�apir[�	transform)r`ru�transformerrTrTrUrj�sz3PydanticPlugin._pydantic_model_class_maker_callbackcCs@|jjrdS|jjj}|sJd��t|jdd�rd|j_dSdS)z�Reset dataclass_transform_spec attribute of ModelMetaclass.

        Let the plugin handle it. This behavior can be disabled
        if 'debug_dataclass_transform' is set to True', for testing purposes.
        Nz-callback not passed from 'get_metaclass_hook'�dataclass_transform_spec)r[�debug_dataclass_transformry�info�declared_metaclass�getattr�typer~)r`ru�info_metaclassrTrTrUrn�s
�z8PydanticPlugin._pydantic_model_metaclass_marker_callback�rWr*rQrX)rcrPrQrd)rcrPrQrq)rur/rQrv)rur,rQrX)�__name__�
__module__�__qualname__�__doc__r_rlrprtrwrjrn�
__classcell__rTrTrarUrS|s

	


rSc@sJeZdZUdZdZded<ded<ded<ded<ddd
�Zddd�ZdS)rZa�A Pydantic mypy plugin config holder.

    Attributes:
        init_forbid_extra: Whether to add a `**kwargs` at the end of the generated `__init__` signature.
        init_typed: Whether to annotate fields in the generated `__init__`.
        warn_required_dynamic_aliases: Whether to raise required dynamic aliases error.
        debug_dataclass_transform: Whether to not reset `dataclass_transform_spec` attribute
            of `ModelMetaclass` for testing purposes.
    )�init_forbid_extra�
init_typed�warn_required_dynamic_aliasesr�boolr�r�r�rrWr*rQrXcCs�|jdurdSt|j�}|dur:|�di��di�}|jD]}|�|d�}t|t�s1td|����t|||�qdSt�}|�	|j�|jD]}|j
t|dd�}t|||�qFdS)N�toolrFFz/Configuration value must be a boolean for key: )�fallback)�config_file�
parse_toml�get�	__slots__rfr��
ValueError�setattrr�read�
getboolean�CONFIGFILE_KEY)r`rW�toml_config�config�key�settingr[rTrTrUr_�s"



�
�zPydanticPluginConfig.__init__rvcs�fdd��jD�S)z/Returns a dict of config names to their values.csi|]}|t�|��qSrT)r�)�.0r��r`rTrU�
<dictcomp>��z0PydanticPluginConfig.to_data.<locals>.<dictcomp>)r�r�rTr�rUr\��zPydanticPluginConfig.to_dataNr��rQrv)r�r�r�r�r��__annotations__r_r\rTrTrTrUrZ�s


rZrur-r;cCs�|j}t|t�r|j}t|t�rt|jt�r|j}nt|t�r"|}nd|�d|jj�d�}t	||j
|j�|jS|jj
�t�}|durF|jS|j�t�sO|jS|�di��d�}|durft|jj|j
|j�|jS)z1Raise an error if from_attributes is not enabled.z
ctx.type: �
 (of type �)Nr��from_attributesT)r�rfr=�itemr8�ret_typer9rbr��error_unexpected_behaviorr{�context�default_return_type�metadatar��METADATA_KEYrhri�error_from_attributes�name)ru�ctx_type�
model_type�detail�pydantic_metadatar�rTrTrUrs�s&

rsc@sdeZdZdZd4dd�Zd5d d!�Z	"	"d6d7d$d%�Z	"d8d9d'd(�Zd:d*d+�Ze	d;d-d.��Z
d<d1d2�Zd3S)=�PydanticModelFieldz5Based on mypy.plugins.dataclasses.DataclassAttribute.r�rP�alias�
str | None�	is_frozenr��has_dynamic_alias�has_default�strict�bool | None�line�int�columnr��Type | Noner�r(cCs@||_||_||_||_||_||_||_||_|	|_|
|_	dSrY)
r�r�r�r�r�r�r�r�r�r�)r`r�r�r�r�r�r�r�r�r�r�rTrTrUr_�s

zPydanticModelField.__init__�current_info�typed�model_strict�force_optional�	use_aliasr{r0�force_typevars_invariant�is_root_model_rootrQrc	Csv|�||||�}	|jdur|n|j}
|s|
r|j||dd�}nttj�}t|	|d|r.td�S|s3|jr7t	d�St
d�S)zABased on mypy.plugins.dataclasses.DataclassAttribute.to_argument.NT)�include_root_type)�variable�type_annotation�initializer�kind)�to_varr�rr7r<�explicitrrr�rr
)r`r�r�r�r�r�r{r�r�r�r�r�rTrTrU�to_arguments 
����zPydanticModelField.to_argumentFr�cCs|rt|jt�r|j��}t|_||_|jdur�|jjdur�t�	|j
j��Zt|�}t|t
�s1J�|rA|jD]
}t|t�r@t|_q6t|j|jjj|i�}|rvt|t
�rvt|j�rv|jdj}	|	durk|Wd�St|	|�}
t||
g�}|Wd�S1s�wY|jS)zABased on mypy.plugins.dataclasses.DataclassAttribute.expand_type.N�root)rfr�r>�
copy_modifiedr�variancer��	self_typer4�strict_optional_setrW�strict_optionalrAr9�argsr�id�
is_root_modelr	r?)r`r�r{r�r��
modified_type�filled_with_typevars�arg�
expanded_type�	root_type�expanded_root_typerTrTrUr2s2


��
 �zPydanticModelField.expand_typer)cCs0|r|jdur|j}n|j}t||�|||��S)z<Based on mypy.plugins.dataclasses.DataclassAttribute.to_var.N)r�r�r)r)r`r�r{r�r�r�rTrTrUr�_szPydanticModelField.to_varrc
Cs8|jsJ�|j|j|j|j|j|j|j|j|j�	�d�	S)�?Based on mypy.plugins.dataclasses.DataclassAttribute.serialize.)	r�r�r�r�r�r�r�r�r�)
r�r�r�r�r�r�r�r�r��	serializer�rTrTrUr�ns
�zPydanticModelField.serialize�datacCs,|��}t|�d�|�}|d||d�|��S)�ABased on mypy.plugins.dataclasses.DataclassAttribute.deserialize.r�)r�r�NrT)�copyr1�pop)ryr�r�r{�typrTrTrU�deserialize}szPydanticModelField.deserialize�sub_typerXcCsT|jdur(t�|jj��t|j||j�|_Wd�dS1s!wYdSdS)zxExpands type vars in the context of a subtype when an attribute is inherited
        from a generic super type.
        N)r�r4r�rWr�r6r�)r`r�r{rTrTrU�expand_typevar_from_subtype�s

"��z.PydanticModelField.expand_typevar_from_subtypeN)r�rPr�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r()r�r(r�r�r�r�r�r�r�r�r{r0r�r�r�r�rQr)FF)
r�r(r{r0r�r�r�r�rQr��F)
r�r(r{r0r�r�r�r�rQr)�rQr)r�r(r�rr{r0rQr�)r�r(r{r0rQrX)r�r�r�r�r_r�rr�r��classmethodr�r�rTrTrTrUr��s

!�2�
r�c@s0eZdZdZdd�Zeddd��Zd
d	d
�ZdS)�PydanticModelClassVarz�Based on mypy.plugins.dataclasses.DataclassAttribute.

    ClassVars are ignored by subclasses.

    Attributes:
        name: the ClassVar name
    cCs
||_dSrY�r�)r`r�rTrTrUr_�s
zPydanticModelClassVar.__init__r�rrQcCs|��}|di|��S)r�NrT)r�)ryr�rTrTrUr��sz!PydanticModelClassVar.deserializecCs
d|jiS)r�r�r�r�rTrTrUr��s�zPydanticModelClassVar.serializeN)r�rrQr�r�)r�r�r�r�r_r�r�r�rTrTrTrUr��sr�c@seZdZUdZhd�Zded<d`dd�Zdadd�Zdbdd�Zdcdd�Z	dddd�Z
ded!d"�Zdfd%d&�Zdgd+d,�Z
dhd4d5�Zdid:d;�Zdid<d=�Zdjd?d@�ZdkdldFdG�ZedmdHdI��ZedndKdL��ZedodNdO��ZedmdPdQ��Z	AdkdpdXdY�ZdqdZd[�Zedrd]d^��Zd_S)srxz�Transform the BaseModel subclass according to the plugin settings.

    Attributes:
        tracked_config_fields: A set of field configs that the plugin has to track their value.
    >�extra�frozenr��alias_generatorr��populate_by_name�validate_by_name�validate_by_aliaszset[str]�tracked_config_fieldsryrrz�Expression | Statementr{r0r[rZrQrXcCs||_||_||_||_dSrY)�_cls�_reason�_apir[)r`ryrzr{r[rTrTrUr_�s
z!PydanticModelTransformer.__init__r�cCs�|jj}t|�}|��}|�||�\}}|dus|durdS|D]
}|jdur*dSq |�t�}|�||||�|�	||||�|j
||j|jdud�|�
�dd�|D�dd�|D�|��d�|jt<dS)	a�Configures the BaseModel subclass according to the plugin settings.

        In particular:

        * determines the model config and fields,
        * adds a fields-aware signature for the initializer and construct methods
        * freezes the class if frozen = True
        * stores the fields, config, and if the class is settings in the mypy metadata for access by subclasses
        NFT)r�cS�i|]}|j|���qSrT�r�r��r��fieldrTrTrUr��r�z6PydanticModelTransformer.transform.<locals>.<dictcomp>cSr�rTr�)r��	class_varrTrTrUr��r�)�fields�
class_varsr�)r�r�r��collect_config�collect_fields_and_class_varsr�rh�BASESETTINGS_FULLNAME�add_initializer�add_model_construct_method�
set_frozenr�r��adjust_decorator_signatures�get_values_dictr�r�)r`r��is_a_root_modelr�r�r�r��is_settingsrTrTrUr|�s(

�
�z"PydanticModelTransformer.transformcs�|jjj��D]8}t|jt�r?|jjd�t�t�r?t�j	t
�r?�j	jtvr?�j	jt
kr:t�fdd�t�j�D��s?d|jj_qdS)a�When we decorate a function `f` with `pydantic.validator(...)`, `pydantic.field_validator`
        or `pydantic.serializer(...)`, mypy sees `f` as a regular method taking a `self` instance,
        even though pydantic internally wraps `f` with `classmethod` if necessary.

        Teach mypy this by marking any function whose outermost decorator is a `validator()`,
        `field_validator()` or `serializer()` call as a `classmethod`.
        rc3s6�|]\}}�j|dkot|t�o|jdkVqdS)�mode�afterN)�	arg_namesrfr$�value)r��ir���	first_decrTrU�	<genexpr>�s
� �
�zGPydanticModelTransformer.adjust_decorator_signatures.<locals>.<genexpr>TN)r�r��names�valuesrfrgr�original_decoratorsr�calleerrc�(IMPLICIT_CLASSMETHOD_DECORATOR_FULLNAMES�MODEL_VALIDATOR_FULLNAME�any�	enumerater��func�is_class)r`rkrTrrUr�s �
���
��z4PydanticModelTransformer.adjust_decorator_signatures�ModelConfigDatac	Cs&|j}t�}d}d}|j��D]\}}|�||�}|r"d}|�|�qd}|jjD]�}t|t	t
f�s3q)t|t	�r�|jd}	t|	t�rG|	j
dkrHq)t|jt�rmt|jj|jj�D]\}
}|
dur`qW|�|j|
|dd��qWnNt|jt�r�|jjD]\}}
t|t�s�qw|�|�|j|
��qwn.t|t
�r�|j
dkr�q)|jjD]}t|t	�s�q�|jd}	t|	t�s�q�|�|�|	j
|j��q�|r�|j�d|�nd}q)|s�|r�|r�|jr�|js�|js�|jjr�t|j|�|jjd	d�D]&}t |j!vr�q�|j�"t#|j$��|j!t d
��D]\}}|�%||��qq�|S)zhCollects the values of the config attributes that are used by the plugin, accounting for parent classes.FTNr�model_config)�	lax_extra�ConfigzYSpecifying config in two places is ambiguous, use either Config attribute or class kwargs�r�)&r�r�keywords�items�get_config_update�update�defs�bodyrfrr�lvaluesrr��rvaluer�ziprr�rr$r
r��fail�has_alias_generatorr�r�r[r��error_required_dynamic_aliasesr��mror�r��add_plugin_dependencyr3rc�
setdefault)r`ryr��has_config_kwargs�has_config_from_namespacer��expr�config_data�stmt�lhs�arg_namer��key_expr�
value_expr�substmtr�r
rTrTrUrs�
�

�
�




������
�z'PydanticModelTransformer.collect_configrr��Jtuple[list[PydanticModelField] | None, list[PydanticModelClassVar] | None]cCs�|j}i}i}t|jjdd��D]b}t|jvrq|j�t|j	��|jtd�
�D]2\}}t�|||j�}	|	�
|j|j�|	||<|jj�|�}
|
r]|
jr]t|
jt�s]|j�d|
j�q+|jtd�
�D]\}}t�|�||<qgqt�}t�}|�|j�D]I}
|�|
||�}|dur�q�|
jd}t|t�s�J�t|t�r�|r�|jdkr�t|j|
�q�|�|j�|||j<q�t|t�r�|�|j�|||j<q�t|���t|���fS)	zACollects the fields for the model, accounting for parent classes.r ���r�z7BaseModel field may only be overridden by another fieldr�Nrr�) r��reversedr�r-r�r�r�r.r3rcr"r�r�r�rr�rgrfr)r*r��set�%_get_assignment_statements_from_blockr%�$collect_field_or_class_var_from_stmtr'rr�� error_extra_fields_on_root_model�add�listr)r`rr�ry�found_fields�found_class_varsr�r�r�r��sym_node�current_field_names�current_class_vars_namesr4�maybe_fieldr5rTrTrUrPsN
���



�z6PydanticModelTransformer.collect_fields_and_class_varsr4r�Iterator[AssignmentStmt]ccsT�|jD]
}|js|�|�EdHq|jdur&|jjs(|�|j�EdHdSdSdSrY)r&�is_unreachabler>�	else_body)r`r4r&rTrTrU�,_get_assignment_statements_from_if_statement�s�
��zEPydanticModelTransformer._get_assignment_statements_from_if_statement�blockrccs>�|jD]}t|t�r|Vqt|t�r|�|�EdHqdSrY)r&rfrrrL)r`rMr4rTrTrUr>�s�


��z>PydanticModelTransformer._get_assignment_statements_from_blockrr�� dict[str, PydanticModelClassVar]�1PydanticModelField | PydanticModelClassVar | NonecCsN|j}|jd}t|t�rt�|j�r|jdkrdS|jsKt|jt	�r<t|jj
t	�r<t|jj
j
t�r<|jj
j
jtvr<dS|j|vrCdSt
|j|�dS|jd}t|t�sWdSt�|j�rb|jdkrddS|jj�|j�}|durrdS|j}t|t�r|dSt|t�r�|j�d|�dSt|t�s�dS|jr�t|j�St|j�}t|t�r�|jjdkr�|j�d|�|�|�}	|�|�}
|jdur�|jr�|jr�|jj |jdd�}|r�||_n
|j�d	|�t!t"j#�|_|jr�|	r�t|j�S|�$|�\}}
|
�r|j%�s|j&�s|j'j(�rt)|j|�|�*|�}|�+||j|�}t,|j|
|	|
|||j-|j.||jd
�
S)a^Get pydantic model field from statement.

        Args:
            stmt: The statement.
            model_config: Configuration settings for the model.
            class_vars: ClassVars already known to be defined on the model.

        Returns:
            A pydantic model field if it could find the field in statement. Otherwise, `None`.
        rrNzFType aliases inside BaseModel definitions are not supported at runtimezdataclasses.InitVarz%InitVar is not supported in BaseModelT)�is_finalzGNeed type argument for Final[...] with non-literal default in BaseModel)
r�r�r�r�r�r�r�r�r�r�)/r�r'rfrrD�is_valid_field_namer��
new_syntaxr(rrrc�DECORATOR_FULLNAMES�error_untyped_fieldsr�r�rr�rgr!r'r*r)�is_classvarr�r@r�r9�get_has_default�
get_strictrP�is_inferred�analyze_simple_literal_typer7r<�
from_error�get_alias_infor�r�r[r�r,�is_field_frozen�_infer_dataclass_attr_init_typer�r�r�)r`r4rr�ryr5rkrg�	node_typer�r�r�r�r�r��	init_typerTrTrUr?�s�
 
���




�


�

�

����
�z=PydanticModelTransformer.collect_field_or_class_var_from_stmtrkr%r�rPr�rr�c	Cs�|j}|jr|St|j�}t|t�s|S|j�d�}|rpt|jt�rc|j�d�}|s,J�|jr:tt	|j|j|��}nt
tj�St|t
�rT|jtttgkrTt|jd|�S|j�d|jj�d�|�|S|j�d|jj�d�|�|S)zvInfer __init__ argument type for an attribute.

        In particular, possibly use the signature of __set__.
        �__set__rNz(Unsupported signature for "__set__" in "�"zUnsupported "__set__" in ")r��implicitr@rfr9r�rgr�get_containing_type_infor6r7r<�unannotatedr8�	arg_kindsr
r	�	arg_typesr�r*r�)	r`rkr�r��default�t�setter�
super_info�setter_typerTrTrUr]+s0


��z8PydanticModelTransformer._infer_dataclass_attr_init_typer��list[PydanticModelField]r�r	c
Cs�d|jjjvr|jjjdjsdS|jj}t|j�}|jp|j	o%|j
du}t|jo-|j�}|j|||||||dd�}	|r�|j
�t�j}
t|
t�sLJ�d|
jvr�|
jdj}t|t�s^J�|dur�|jdur�|j}t|t�sqJ�t|j�D];\}
}|dus�|�d�s�|�d�s�qv|j
�|j|
�}|dur�|dkr�|�ttj��}t||�}|	� t!||dt"��qv|�#||�s�td	�}|	� t!|t$tj%�dt&��t'|j
|jd|	t(�d
�dS)z�Adds a fields-aware `__init__` method to the class.

        The added `__init__` will be annotated with types vs. all `Any` depending on the plugin settings.
        r_NFT)r�r��requires_dynamic_aliasesr�r	r�r��__�_�_cli_settings_source�kwargs)r��return_type))r�r�r�plugin_generatedr[r�r�r�r�r�r�r+�get_field_argumentsr�rerrgrfr(rr�r8rr�
startswith�	anal_typerf�accept�ChangeExplicitTypeOfAnyr<�from_omitted_genericsr)�appendrr�should_init_forbid_extrar7r�r�
add_methodr:)r`r�r�r	r�r�r�r�rmr��base_settings_node�base_settings_init_node�	func_type�arg_idxr6�analyzed_variable_typer��varrTrTrUrQsN
�
�
z(PydanticModelTransformer.add_initializerc

Cs�|j�t�d�|j�t�d��g�}t|t�g�}ttd|�|dt�}t�	|jj
j��|j|dt
|j�dd||d�}Wd�n1sEwY|�||�satd�}	|�t|	ttj�dt��|rh||gn|g|}t|j|jd	|t|jj�dd
�dS)z�Adds a fully typed `model_construct` classmethod to the class.

        Similar to the fields-aware __init__ method, but always uses the field names (not aliases),
        and does not treat settings fields as optional.
        z.setz.str�_fields_setNTF)r�r�rmr�r	r�rq�model_construct)r�rr�is_classmethod)r��
named_type�
BUILTINS_NAMEr?r:rr)rr4r�rWr�rtr�r�r{rzr7r<r�rr|r�rAr�)
r`r�r�r	r��set_str�optional_set_str�fields_set_argumentr�r�rTrTrUr�s4$��


�z3PydanticModelTransformer.add_model_construct_methodr�c
	Cs�|jj}|D]r}|j�|j�}|durX|j}t|t�r"|p|j|_	qt|t
�r1|jjs1|j�
�qzt|�}Wn
tyDt|�}Ynwd|�d|j�d�}	t|	|j|j�q|j||dd�}||_||_	|jd|j|_tt|�|j|j<qdS)z�Marks all fields as properties so that attempts to set them trigger mypy errors.

        This is the same approach used by the attrs and dataclasses plugins.
        Nzsym_node.node: r�r�F)r��.)r�r�rr�r�rgrfr)r��is_propertyr!r��final_iteration�deferrP�	TypeError�reprrbr�r�rc�	_fullnamer%r)
r`r�r{r�r�r�rEr��var_strr�rTrTrUr�s,
��z#PydanticModelTransformer.set_frozenFr�rr�ModelConfigData | NonecCs�||jvrdS|dkr1t|t�r|jdk}nt|t�r!|jdk}n|s*t||j|�dSt|d�S|dkrHd}t|t	�rC|j
dkrCd}t|d	�St|t	�r^|j
d
vr^tdi||j
dki��St||j|�dS)
z�Determines the config update due to a single kwarg in the ConfigDict definition.

        Warns if a tracked config attribute is set to a value the plugin doesn't know how to interpret (e.g., an int)
        Nr��forbid)�forbid_extrar�T�
builtins.NoneF)r+)�
builtins.True�builtins.Falser�rT)r�rfr$r
rr��error_invalid_config_valuer�rrrc)r`r�r�rr�r+rTrTrUr#�s(


	

z*PydanticModelTransformer.get_config_updatecCs�|j}t|t�r
dSt|t�rIt|jt�rI|jjtkrIt|j	|j
�D]$\}}|dus.|dkr5|jtuS|dkrFt|t
�oB|jdkSq"dSt|t�S)zUReturns a boolean indicating whether the field defined in `stmt` is a required field.FNrg�default_factoryr�)r(rfr&rrr"rc�FIELD_FULLNAMEr)r�rrbrr�r4r2r�r�rTrTrUrV�s
"�z(PydanticModelTransformer.get_has_defaultr�cCs||j}t|t�r<t|jt�r<|jjtkr<t|j|j	�D] \}}|dkr$qt|t
�r9|jdkr1dS|jdkr9dSdSdS)zEReturns a the `strict` value of a field if defined, otherwise `None`.r�r�Tr�FN)r(rfrrr"rcr�r)r�rrr�rTrTrUrWs"


z#PydanticModelTransformer.get_strict�tuple[str | None, bool]cCs�|j}t|t�r
dSt|t�rt|jt�r|jjtksdSd|jvr,|j	|j�
d�}nd|jvr;|j	|j�
d�}ndSt|t�rG|jdfSdS)aReturns a pair (alias, has_dynamic_alias), extracted from the declaration of the field defined in `stmt`.

        `has_dynamic_alias` is True if and only if an alias is provided, but not as a string literal.
        If `has_dynamic_alias` is True, `alias` will be None.
        )NF�validation_aliasr�F)NT)
r(rfr&rrr"rcr�rr��indexr$r
)r4r2r�rTrTrUr[s"
�
�



z'PydanticModelTransformer.get_alias_infocCs||j}t|t�r
dSt|t�rt|jt�r|jjtksdSt|j	�D]\}}|dkr;|j
|}t|t�o8|jdkSq"dS)alReturns whether the field is frozen, extracted from the declaration of the field defined in `stmt`.

        Note that this is only whether the field was declared to be frozen in a `<field_name> = Field(frozen=True)`
        sense; this does not determine whether the field is frozen because the entire model is frozen; that is
        handled separately.
        Fr�r�)r(rfr&rrr"rcr�rrr�r)r4r2rr6r�rTrTrUr\5s
�
�
�z(PydanticModelTransformer.is_field_frozenr�r�r�rmr��list[Argument]c	
	s.�jj����������f	dd�|D�}	|	S)z�Helper function used during the construction of the `__init__` and `model_construct` method signatures.

        Returns a list of mypy Argument instances for use in the generated signatures.
        cs@g|]}�r	|js|j����p���j��o|jdkd��qS)r�)r�r�r�r�r{r�r�)r�r�r�r�r��	r�r�r�r	r�rmr`r�r�rTrU�
<listcomp>^s"����z@PydanticModelTransformer.get_field_arguments.<locals>.<listcomp>)r�r�)
r`r�r�r�r�rmr	r�r��	argumentsrTr�rUrtNs
�z,PydanticModelTransformer.get_field_argumentscCs4|js|js|�|t|j��rdS|jrdS|jjS)a@Indicates whether the generated `__init__` should get a `**kwargs` at the end of its signature.

        We disallow arbitrary kwargs if the extra config setting is "forbid", or if the plugin config says to,
        *unless* a required dynamic alias is present (since then we can't determine a valid signature).
        FT)r�r��is_dynamic_alias_presentr�r+r�r[r�)r`r�r�rTrTrUr{nsz1PydanticModelTransformer.should_init_forbid_extrar+cCs8|D]}|jr
dSq|r|D]
}|jdurdSqdS)z�Returns whether any fields on the model have a "dynamic alias", i.e., an alias that cannot be
        determined during static analysis.
        TNF)r�r�)r�r+r�rTrTrUr�{s�
�z1PydanticModelTransformer.is_dynamic_alias_presentN)
ryrrzr�r{r0r[rZrQrX)rQr�)rQrX)rQr)rrr�r�rQr:)r4rrQrI)rMrrQrI)r4rrrr�rNrQrO)rkr%r�rPr�rrQr�)
r�rlr�rr	r�r�r�rQrX)r�rlr{r0r�r�rQrXr�)r�rPr�rrr�rQr�)r4rrQr�)r4rrQr�)r4rrQr�)r�rlr�r�r�r�r�r�rmr�r	r�r�r�r�r�rQr�)r�rlr�rrQr�)r�rlr+r�rQr�)r�r�r�r�r�r�r_r|rrrrLr>r?r]rrrr#�staticmethodrVrWr[r\rtr{r�rTrTrTrUrx�s<



$

L
D




&
6
(#!�
 
rxcs,eZdZdZd
�fdd�Zddd�Z�ZS)rxz<A type translator used to change type of Any's, if explicit.�type_of_anyr�rQrXcs||_t���dSrY)�_type_of_anyr^r_)r`r�rarTrUr_�sz ChangeExplicitTypeOfAny.__init__rhr7r;cCs|jtjkr
|j|jd�S|S)N)r�)r�r<r�r�r�)r`rhrTrTrU�	visit_any�sz!ChangeExplicitTypeOfAny.visit_any)r�r�rQrX)rhr7rQr;)r�r�r�r�r_r�r�rTrTrarUrx�srxc@sJeZdZdZ								dddd
�Zddd�Zd dd�Zd!dd�ZdS)"rz(Pydantic mypy plugin model config class.Nr�r�r�r�r�r�r�r+r�c		Cs4||_||_||_||_||_||_||_||_dSrY)r�r�r�r�r�r�r+r�)	r`r�r�r�r�r�r�r+r�rTrTrUr_�s
zModelConfigData.__init__rQrvcCsdd�|j��D�S)z�Returns a dict of Pydantic model config names to their values.

        It includes the config if config value is not `None`.
        cSsi|]\}}|dur||�qSrYrT)r��k�vrTrTrUr��sz3ModelConfigData.get_values_dict.<locals>.<dictcomp>)�__dict__r"r�rTrTrUr��zModelConfigData.get_values_dictr�r�rXcCs2|durdS|����D]
\}}t|||�qdS)z$Update Pydantic model config values.N)rr"r�)r`r�r�r�rTrTrUr$�s
�zModelConfigData.updater�rPr
rcCs"t||�durt|||�dSdS)zFSet default value for Pydantic model config if config value is `None`.N)r�r�)r`r�r
rTrTrUr/�s�zModelConfigData.setdefault)NNNNNNNN)r�r�r�r�r�r�r�r�r�r�r�r�r+r�r�r�r�)r�r�rQrX)r�rPr
rrQrX)r�r�r�r�r_rr$r/rTrTrTrUr�s�

rr�r(r�cCs
|�t�S)zXReturn whether the type info is a root model subclass (or the `RootModel` class itself).)rh�ROOT_MODEL_FULLNAME)r�rTrTrUr��s
r�zpydantic-ormzInvalid from_attributes call�Pydanticzpydantic-configzInvalid config valuezpydantic-aliaszDynamic alias disallowedzpydantic-unexpectedzUnexpected behaviorzpydantic-fieldzUntyped field disallowedzInvalid Field defaultsz!Extra field on RootModel subclass�
model_namer{r+r�rrXcC�|jd|�d�|td�dS)zCEmits an error when the model does not have `from_attributes=True`.raz$" does not have from_attributes=True��codeN)r*�	ERROR_ORM)r�r{r�rTrTrUr���r�r�r0cCr�)z0Emits an error when the config value is invalid.zInvalid value for "Config.rar�N)r*�ERROR_CONFIG)r�r{r�rTrTrUr��r�r�cC�|jd|td�dS)znEmits required dynamic aliases error.

    This will be called when `warn_required_dynamic_aliases=True`.
    z#Required dynamic aliases disallowedr�N)r*�ERROR_ALIAS�r{r�rTrTrUr,�r�r,r��8CheckerPluginInterface | SemanticAnalyzerPluginInterfacecCs4d}d|�d�}|d|�d�7}|j||td�dS)z Emits unexpected behavior error.z6https://github.com/pydantic/pydantic/issues/new/choosez7The pydantic mypy plugin ran into unexpected behavior: �
z&Please consider reporting this bug at z so we can try to fix it!r�N)r*�ERROR_UNEXPECTED)r�r{r��link�full_messagerTrTrUr��sr�cCr�)z;Emits an error when there is an untyped field in the model.zUntyped fields disallowedr�N)r*�
ERROR_UNTYPEDr�rTrTrUrT�r�rTcCr�)z]Emits an error when there is more than just a root field defined for a subclass of RootModel.z2Only `root` is allowed as a field of a `RootModel`r�N)r*�ERROR_EXTRA_FIELD_ROOT_MODELr�rTrTrUr@�r�r@F�8SemanticAnalyzerPluginInterface | CheckerPluginInterfaceryrr�r�rrr�r��tvar_def�TypeVarType | Noner�cCs|j}||jvr|j|}	|	jrt|	jt�r|jj�|	j�t|t	�r)|�
d�}
n|�dg�}
|rE|p8tt
|��}ttd�|dtd�g}n|pJt
|�}ttd�|dt�g}||}ggg}}
}|D]}|jslJd��|�|j�|
�|jj�|�|j�qct|||
||
�}|r�|g|_t||tt�g��}||_t||�|_||_|jd||_|j|_||jvr�t ||j�}|j||j|<|r�d|_!t||j�}||_|j|_d|_"t#|t$d�g|�}|j|_t%t&|�}	nt%t&|�}	d|	_|	|j|<|j'jj�|�dS)	zhVery closely related to `mypy.plugins.common.add_method_to_class`, with a few pydantic-specific changes.zbuiltins.functionr�NT�__pydantic_self__z"All arguments must be fully typed.r�r�)(r�rrsrfrgrr%r&�remover0r��named_generic_typer=rArr)r
r�rzr�r�r�r8�	variablesrr r2r�rrcr�r�rB�is_decoratedr�rrr%r�defn)r{ryr�r�rrr�r�r�r�rk�
function_type�firstrfrrer��	signaturer�r_namer��decrTrTrUr|�sZ





r|r��dict[str, Any] | NonecCs�|�d�sdStjdkrddl}nzddl}Wnty*ddl}|�d�YdSwt|d��
}|�	|�Wd�S1s@wYdS)z�Returns a dict of config keys to values.

    It reads configs from toml file and returns `None` if the file is not a toml file.
    z.tomlN)��rzJNo TOML parser installed, cannot read configuration from `pyproject.toml`.�rb)
rr�sys�version_info�tomllib�tomli�ImportError�warnings�warn�open�load)r��toml_r��rfrTrTrUr�Ps



�$�r�)rOrPrQrR)rur-rQr;)r�r(rQr�)r�rPr{r+r�rrQrX)r�rPr{r0r�rrQrX)r{r0r�rrQrX)r�rPr{r�r�rrQrX)r{r+r�rrQrX)NNF)r{r�ryrr�rPr�r�rrr;r�r�r�r�r�r�rQrX)r�rPrQr�)�r��
__future__rr��collections.abcr�configparserr�typingrr�mypy.errorcodesr�mypy.expandtyperr	�
mypy.nodesr
rrr
rrrrrrrrrrrrrrrrrrr r!r"r#r$r%r&r'r(r)�mypy.optionsr*�mypy.pluginr+r,r-r.r/r0�mypy.plugins.commonr1�mypy.semanalr2�mypy.server.triggerr3�
mypy.stater4�mypy.type_visitorr5�mypy.typeopsr6�
mypy.typesr7r8r9r:r;r<r=r>r?r@�
mypy.typevarsrA�	mypy.utilrB�mypy.versionrC�mypy_version�pydantic._internalrD�pydantic.versionrEr�r�rirr�rmr��DATACLASS_FULLNAMErrSr�MYPY_VERSION_TUPLEr�rVrSrZrsr�r�rxrxrr�r�r�r�r�r��ERROR_FIELD_DEFAULTSr�r�r�r,r�rTr@r|r�rTrTrTrU�<module>s��" 0

6
.i
+





�X
@ Telegram