o

|��g.��@
sdZd
dl
mZ
d
dlZd
dlmZ
d
dlmZmZm	Z	
d
dl
mZ
d
dlm
Z
mZ
d
dlmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZm 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/
d
d	l0m1Z1
d
d
l2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8
d
dl9m:Z:
d
dl;m<Z<
d
d
l=m>Z>
d
dl?m@Z@
d
dlAmBZB
d
dlCmDZDmEZEmFZFmGZGmHZHmIZImJZJmKZKmLZLmMZM
d
dlNmOZO
d
dlPmQZQ
d
dlRmSZT
d
dlUmVZV
d
dlWmXZX
dZYdZZdZ[dZ\dZ]dZ^dZ_dZ`dZahd�
ZbeXeT�
Zcd Zdd!ZSdod&d'�ZeGd(d)�d)e6�ZfGd*d+�d+�Zgdpd/d0�ZhGd1d2�d2�ZiGd3d4�d4�ZjGd5d6�d6�ZkGd7d8�d8�Zled9d:d;�Zmed<d=d;�Zned>d?d;�Zoed@dAd;�ZpedBdCd;�ZqedBdDd;�ZredBdEd;�ZsdqdLdM�ZtdrdPdQ�ZudsdRdS�ZvdtdVdW�ZwdsdXdY�ZxdudZd[�Zy			\dvdwdidj�Zzdxdmdn�Z{dS)yz^This module includes classes and functions designed specifically for use with the mypy plugin.�)
�annotationsN)
�ConfigParser)�Any�Callable�Iterator)
�	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)
�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�0pydantic.functional_serializers.model_serializer�3pydantic.deprecated.class_validators.root_validatorrF�builtins��version�str�return�type[Plugin]c


C
stS)
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)
rN�rS�8/usr/local/lib/python3.10/dist-packages/pydantic/mypy.py�pluginjsrUc
s^eZ
dZd
Zd�f
dd�Zddd�Zdd
d�Zddd�Zddd�Zd dd�Z	d dd�Z
�ZS)!rRzThe Pydantic mypy plugin.�optionsr*rP�Nonec
s&t|
�
|_
|j
��|_t��|
�

dS�
N)�PydanticPluginConfig�
plugin_config�to_data�_plugin_data�super�__init__)�selfrV�
�	__class__rSrTr^|s



zPydanticPlugin.__init__�fullnamerO�(Callable[[ClassDefContext], None] | NonecC
s:|�|
�
}|rt
|jt�rtd
d�|jjD�
�
r|jSdS)zUpdate Pydantic model class.c
s
��|]}
|
jt
kV
qdSrX�rb�BASEMODEL_FULLNAME��.0�baserSrSrT�	<genexpr>���z5PydanticPlugin.get_base_class_hook.<locals>.<genexpr>N)�lookup_fully_qualified�
isinstance�noder(�any�mro�$_pydantic_model_class_maker_callback)r_rb�symrSrSrT�get_base_class_hook�s




z"PydanticPlugin.get_base_class_hookcC
s|
tkr|j
Sd
S)z,Update Pydantic `ModelMetaclass` definition.N)�MODEL_METACLASS_FULLNAME�)_pydantic_model_metaclass_marker_callback�r_rbrSrSrT�get_metaclass_hook�s

z!PydanticPlugin.get_metaclass_hook�&Callable[[MethodContext], Type] | NonecC
s|
�d
�
rt
SdS)z-Adjust return type of `from_orm` method call.z	.from_ormN)�endswith�from_attributes_callbackrvrSrSrT�get_method_hook�s


zPydanticPlugin.get_method_hook�ctxr/�dict[str, Any]c
C
s|jS)
zjReturn all plugin config data.

        Used by mypy to determine if cache needs to be discarded.
        )
r\)r_r|rSrSrT�report_config_data�sz!PydanticPlugin.report_config_datar,cC
s"t|
j
|
j|
j|j�}|��
dSrX)�PydanticModelTransformer�cls�reason�apirZ�	transform)r_r|�transformerrSrSrTrq�s

z3PydanticPlugin._pydantic_model_class_maker_callbackcC
s@|jj
rd
S|
jjj}|sJd�
�
t|jdd
�rd
|j_d
Sd
S)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)rZ�debug_dataclass_transformr��info�declared_metaclass�getattr�typer�)r_r|�info_metaclassrSrSrTru�s





�z8PydanticPlugin._pydantic_model_metaclass_marker_callback�rVr*rPrW)rbrOrPrc)rbrOrPrx)r|r/rPr})r|r,rPrW)�__name__�
__module__�__qualname__�__doc__r^rsrwr{r~rqru�
__classcell__rSrSr`rTrRys


	


rRc@
sJeZ
dZUd
ZdZded<ded<ded<ded<ddd
�Zddd�ZdS)rYa�
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�r�rVr*rPrWcC
s�|
jdurdSt
|
j�
}|du
r:|�d
i��di�}|jD]}|�|d�}t|t�s1td|���
�
t|||�
qdSt�}|�	|
j�

|jD]}|j
t|dd�}t|||�
qFdS)N�toolrEFz/Configuration value must be a boolean for key: )
�fallback)�config_file�
parse_toml�get�	__slots__rmr��
ValueError�setattrr�read�
getboolean�CONFIGFILE_KEY)r_rV�toml_config�config�key�settingrZrSrSrTr^�s"












�




�zPydanticPluginConfig.__init__r}c


s�f
d
d��jD�
S)z/Returns a dict of config names to their values.c

si|]}
|
t�|
��qSrS)
r�)rhr��
r_rSrT�
<dictcomp>��z0PydanticPluginConfig.to_data.<locals>.<dictcomp>)
r�r�rSr�rTr[��zPydanticPluginConfig.to_dataNr��rPr})r�r�r�r�r��__annotations__r^r[rSrSrSrTrY�s






rYr|r-r:c
C
s�|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Stdd�|jjD�
�
sT|jS|�di��d�
}|d	u
rkt|jj|j
|j�
|jS)
z1Raise an error if from_attributes is not enabled.z
ctx.type: �
 (of type �
)Nc
s
rdrXrergrSrSrTrj�rkz+from_attributes_callback.<locals>.<genexpr>r��from_attributesT)r�rmr<�itemr7�ret_typer8rar��error_unexpected_behaviorr��context�default_return_type�metadatar��METADATA_KEYrorp�error_from_attributes�name)r|�ctx_type�
model_type�detail�pydantic_metadatar�rSrSrTrz�s&

















rzc@
sbeZ
dZd
Zd3dd�Zd4d d!�Z	"d5d6d#d$�Z	"d5d7d&d'�Zd8d)d*�Ze	d9d,d-��
Z
d:d0d1�Zd2S);�PydanticModelFieldz5Based on mypy.plugins.dataclasses.DataclassAttribute.r�rO�alias�
str | None�	is_frozenr��has_dynamic_alias�has_default�strict�bool | None�line�int�columnr��Type | Noner�r(cC
s@|
|_||_
||_||_||_||_||_||_|	|_|
|_	dSrX)
r�r�r�r�r�r�r�r�r�r�)r_r�r�r�r�r�r�r�r�r�r�rSrSrTr^�s










zPydanticModelField.__init__�current_info�typed�model_strict�force_optional�	use_aliasr�r0�force_typevars_invariant�is_root_model_rootrPrc	C
sr|�|
|||�}	|j
d
ur|n|j
}
|s|
r|�|
|�}nttj�
}t|	|d
|r,td�S|s1|jr5t	d�St
d�S)zABased on mypy.plugins.dataclasses.DataclassAttribute.to_argument.N)�variable�type_annotation�initializer�kind)�to_varr�rr6r;�explicitrrr�rr
)r_r�r�r�r�r�r�r�r�r�r�r�rSrSrT�to_argument
s 





����zPydanticModelField.to_argumentFcC
s�|rt|j
t�r|j
��}t|_||_
|j
d
u
r[|jjd
u
r[t�	|j
j�
�.
t|
�
}t|t
�s1J�
|rA|jD]
}t|t�r@t|_q6t|j
|jjj|i
�Wd
�
S1sVw



Y
|j
S)zABased on mypy.plugins.dataclasses.DataclassAttribute.expand_type.N)rmr�r=�
copy_modifiedr�variancer��	self_typer4�strict_optional_setrV�strict_optionalr@r8�argsr�id)r_r�r�r��
modified_type�filled_with_typevars�argrSrSrTr/
s"










�
 �	zPydanticModelField.expand_typer)cC
s0|r|jd
u
r|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�rSrSrTr�M
s
zPydanticModelField.to_varrc


C
s8|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�rSrSrTr�\
s








�zPydanticModelField.serialize�datacC
s,|��}t
|�d
�
|�}|d||
d�|�
�
S)�ABased on mypy.plugins.dataclasses.DataclassAttribute.deserialize.r�)r�r�NrS)�copyr1�pop)r�r�r�r��typrSrSrT�deserializek
s

zPydanticModelField.deserialize�sub_typerWcC
sT|jd
u
r(t
�|jj�
�
t|j|
|j�|_Wd
�
d
S1s!w



Y
d
Sd
S)zxExpands type vars in the context of a subtype when an attribute is inherited
        from a generic super type.
        N)r�r4r�rVr�r5r�)r_r�r�rSrSrT�expand_typevar_from_subtyper
s



"��z.PydanticModelField.expand_typevar_from_subtypeN)r�rOr�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�rPr�
F)r�r(r�r0r�r�rPr�)
r�r(r�r0r�r�r�r�rPr)�rPr)r�r(r�rr�r0rPr�)r�r(r�r0rPrW)r�r�r�r�r^r�rr�r��classmethodr�r�rSrSrSrTr��s


�#�

r�c@
s0eZ
dZd
Zdd�Zeddd��
Zd
d	d
�ZdS)�PydanticModelClassVarz�Based on mypy.plugins.dataclasses.DataclassAttribute.

    ClassVars are ignored by subclasses.

    Attributes:
        name: the ClassVar name
    cC
s
|
|_dSrX�
r�)r_r�rSrSrTr^�
s

zPydanticModelClassVar.__init__r�rrPcC
s|
��}
|di|
�
�
S)r�NrS)
r�)r�r�rSrSrTr��
s
z!PydanticModelClassVar.deserializec

C
s
d
|ji
S)r�r�r�r�rSrSrTr��
s�zPydanticModelClassVar.serializeN)r�rrPr�r�)r�r�r�r�r^r�r�r�rSrSrSrTr�{
s

r�c@
s
eZ
dZUd
Zhd�
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)srz�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_namezset[str]�tracked_config_fieldsr�rr��Expression | Statementr�r0rZrYrPrWcC
s|
|_||_
||_||_dSrX)�_cls�_reason�_apirZ)r_r�r�r�rZrSrSrTr^�
s


z!PydanticModelTransformer.__init__r�c
C
s�|jj
}
td
d�|
jdd�D�
�
}|��}|�||�\}}|dus&|dur(dS|D]
}|jdur4
dSq*tdd�|
jdd�D�
�
}|�||||�
|�||||�
|j	||j
|jdud�
|��
d	d
�|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
        c
s
s�|]}
t|
j
vV
qdSrX)�ROOT_MODEL_FULLNAMErbrgrSrSrTrj�
rkz5PydanticModelTransformer.transform.<locals>.<genexpr>N���Fc
s
rdrX)rb�BASESETTINGS_FULLNAMErgrSrSrTrj�
rkT)
r�c
S
�i|]}
|
j|
�
��qSrS�r�r��rh�fieldrSrSrTr��
r�z6PydanticModelTransformer.transform.<locals>.<dictcomp>c
S
r�rSr�)rh�	class_varrSrSrTr��
r�)�fields�
class_varsr�)r�r�rorp�collect_config�collect_fields_and_class_varsr��add_initializer�add_model_construct_method�
set_frozenr�r��adjust_decorator_signatures�get_values_dictr�r�)r_r��
is_root_modelr�r�r
r��is_settingsrSrSrTr��
s(








�




�z"PydanticModelTransformer.transformc

s�|jj
j��D]8}
t|
jt�r?|
jjd
�t�t�r?t�j	t
�r?�j	jtvr?�j	jt
kr:t�f
dd�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`.
        r
c
3
s6�|]\}
}�j|
dkot
|t�o|jd
kV
qdS)�mode�afterN)�	arg_namesrmr$�value)rh�
ir��
�	first_decrSrTrj�
s
� �
�zGPydanticModelTransformer.adjust_decorator_signatures.<locals>.<genexpr>TN)r�r��names�valuesrmrnr�original_decoratorsr�calleerrb�DECORATOR_FULLNAMES�MODEL_VALIDATOR_FULLNAMEro�	enumerater��func�is_class)r_rrrSr
rTr
�
s 

�
�
��
��z4PydanticModelTransformer.adjust_decorator_signatures�ModelConfigDatac
	C
s |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�|jjr�t|j|�
|
jjd	d�D]&}t|j v
r�q�|j�!t"|j#�
�

|j td
��D]\}}|�$||�
�
q
q�|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�bodyrmrr�lvaluesrr��rvaluer�zipr
r�rr$r

r��fail�has_alias_generatorr�rZr��error_required_dynamic_aliasesr�rpr�r��add_plugin_dependencyr3rb�
setdefault)r_r�r��has_config_kwargs�has_config_from_namespacer��expr�config_data�stmt�lhs�arg_namer��key_expr�
value_expr�substmtr�r

rSrSrTr

�
s�






�











�




�
















�����




�z'PydanticModelTransformer.collect_configr
r
�Jtuple[list[PydanticModelField] | None, list[PydanticModelClassVar] | None]cC
s�
|j}i}i}t
|jjd
d��
D]b}t|jv
rq|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�r�z7BaseModel field may only be overridden by another fieldr
Nr
�root) r��reversedr�rpr�r�r�r+
r3rbr 
r�r�r�r
r�rnrmr)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_r
r
r��found_fields�found_class_varsr�r�r�r��sym_node�current_field_names�current_class_vars_namesr1
�maybe_fieldr2
rSrSrTr
<sN









��
�
















�z6PydanticModelTransformer.collect_fields_and_class_varsr1
r�Iterator[AssignmentStmt]cc
sT�|
jD]
}|j
s|�|�
EdH
q|
jdu
r&|
jj
s(|�|
j�
EdH
dSdSdSrX)r$
�is_unreachabler;
�	else_body)r_r1
r$
rSrSrT�,_get_assignment_statements_from_if_statement�s�



�

�zEPydanticModelTransformer._get_assignment_statements_from_if_statement�blockrcc
s>�|
jD]}t
|t�r|V
qt
|t�r|�|�
EdH
qdSrX)r$
rmrrrI
)r_rJ
r1
rSrSrTr;
�s�







��z>PydanticModelTransformer._get_assignment_statements_from_blockrr
� dict[str, PydanticModelClassVar]�1PydanticModelField | PydanticModelClassVar | NonecC
s,|j}|
j
d
}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|
j
d
}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#�
|_|�$|
�
\}}
|
r�|j%s�|j&j'r�t(|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`.
        r
r
NzFType 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%
rmrrC�is_valid_field_namer��
new_syntaxr&
rr
rbr
�error_untyped_fieldsr�r�r
r�rnr!r'r(
r)�is_classvarr�r?r�r8�get_has_default�
get_strictrM
�is_inferred�analyze_simple_literal_typer6r;�
from_error�get_alias_infor�rZr�r*
�is_field_frozen�_infer_dataclass_attr_init_typer�r�r�)r_r1
r
r
r�r2
rrrn�	node_typer�r�r�r�r�r��	init_typerSrSrTr<
�s�
 


���













�







�






�














�z=PydanticModelTransformer.collect_field_or_class_var_from_stmtrrr%r�rOr�rr�c	C
s�|
j}|
j
r|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__rMz(Unsupported signature for "__set__" in "�
"zUnsupported "__set__" in ")r��implicitr?rmr8r�rnr�get_containing_type_infor5r6r;�unannotatedr7�	arg_kindsr
r	�	arg_typesr�r(
r�)	r_rrr�r��default�
t�setter�
super_info�setter_typerSrSrTrY
s0

















��z8PydanticModelTransformer._infer_dataclass_attr_init_typer��list[PydanticModelField]r�r	
c
C
s�
d
|jj
jvr|jj
jd
jsdS|jj}t|j�
}|jdu
}t|j	o&|j�
}|j
|
||||||dd�}	|r�|j�t
�
j}
t|
t�sEJ�
d
|
jvr�|
jd
j}t|t�sWJ�
|du
r�|jdu
r�|j}t|t�sjJ�
t|j�
D]+\}
}|dus�|�d�
s�|�d�
s�qo|j�|j|
�
}t||�}|	�t||dt��

qo|�|
|�s�td�
}|	�t|ttj �
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^NT)r�r��requires_dynamic_aliasesr�r	
r
r��__�
_�kwargs)r��return_type)$r�r�r
�plugin_generatedrZr�r�r�r�r)
�get_field_argumentsr�rlr�rnrmr(rr�r7r
r
�
startswith�	anal_typerb
r)�appendrr�should_init_forbid_extrar6r;r�r�
add_methodr9)r_r�r�r	
r
r�r�r�ri
r��base_settings_node�base_settings_init_node�	func_type�arg_idxr3
�analyzed_variable_typer��varrSrSrTr
5sF














�

















z(PydanticModelTransformer.add_initializerc

C
s�|j�
t�d
�|j�
t�d��
g
�}t|t�g�
}ttd|�|dt�}t�	|jj
j�
�
|j|
dt
|j�
dd||d�}Wd�
n1sEw



Y
|�|
|�satd�
}	|�t|	ttj�
dt��

|rh||g
n|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�ri
r�r	
r
rl
�model_construct)r�rm
�is_classmethod)r��
named_type�
BUILTINS_NAMEr>r9rr)rr4r�rVr�ro
r�r�rs
rr
r6r;r�rrt
r�r@r�)
r_r�r�r	
r
�set_str�optional_set_str�fields_set_argumentr�rz
rSrSrTr
ds4$










��










�z3PydanticModelTransformer.add_model_construct_methodr�c
	C
s�|jj
}|
D]r}|j�|j�
}|d
u
rX|j}t|t�r"|p|j|_	qt|t
�r1|jjs1|j�
�
qzt|�
}Wn
tyD


t|�
}Yn
wd|�d|j�d�}	t|	|j|j�
q|j||dd�}||_
||_	|jd|j|_tt|�|j|j<qd
S)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�r
r�r�rnrmr)r��is_propertyr!r��final_iteration�deferrO�	TypeError�reprrar�r�rb�	_fullnamer%r)
r_r�r�r�r�r�rB
rz
�var_strr�rSrSrTr
�s,









�




�z#PydanticModelTransformer.set_frozenFr�rr
�ModelConfigData | NonecC
s�|
|jv
rd
S|
dkr1t
|t�r|jdk}nt
|t�r!|jdk}n|s*t|
|j|�
d
St|d�
S|
dkrHd}t
|t	�rC|j
dkrCd}t|d	�
St
|t	�r^|j
d
vr^tdi|
|j
dki
�
�
St|
|j|�
d
S)
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�
rS)r�rmr$r

rr��error_invalid_config_valuer�r
rrb)r_r�r�r
r�
r)
rSrSrTr!
�s(








	












z*PydanticModelTransformer.get_config_updatec
C
s�|j}
t
|
t�r
d
St
|
t�rIt
|
jt�rI|
jjtkrIt|
j	|
j
�D]$\}}|dus.|dkr5|jtu

S|dkrFt
|t
�oB|jdk
Sq"d
St
|
t�S)zUReturns a boolean indicating whether the field defined in `stmt` is a required field.FNrc
�default_factoryr�
)r&
rmr&rr
r"rb�FIELD_FULLNAMEr'
r�r
rarr�r1
r/
r�r�rSrSrTrR
�s

"



�z(PydanticModelTransformer.get_has_defaultr�c
C
s||j}
t
|
t�r<t
|
jt�r<|
jjtkr<t|
j|
j	�D] \}}|d
kr$qt
|t
�r9|jdkr1
dS|jdkr9
dS
dSdS)zEReturns a the `strict` value of a field if defined, otherwise `None`.r�r�
Tr�
FN)r&
rmrr
r"rbr�
r'
r�r
rr�
rSrSrTrS
�s"













z#PydanticModelTransformer.get_strict�tuple[str | None, bool]c
C
s�|j}
t
|
t�r
d
St
|
t�rt
|
jt�r|
jjtksd
St|
j	�
D]\}}|dkr+q"|
j
|}t
|t�r<|jdf
S
dSd
S)a
Returns 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.
        )NFr�F)NT)
r&
rmr&rr
r"rbr�
r
r
r�r$r

�r1
r/
r
r3
r�rSrSrTrW
�s"

�

�








z'PydanticModelTransformer.get_alias_infoc
C
s||j}
t
|
t�r
d
St
|
t�rt
|
jt�r|
jjtksd
St|
j	�
D]\}}|dkr;|
j
|}t
|t�o8|jdk
Sq"d
S)al
Returns 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&
rmr&rr
r"rbr�
r
r
r�rr�
rSrSrTrX
s

�

�




�z(PydanticModelTransformer.is_field_frozenr�r�r�ri
r��list[Argument]c	
	
s.�jj
�
��
�������f	d
d�|
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)r8
)r�r�r�r�r�r�r�)r�r�r�r�r��	r�r�r
r	
r�ri
r_r�r�rSrT�
<listcomp>9s"
��








��z@PydanticModelTransformer.get_field_arguments.<locals>.<listcomp>)r�r�)
r_r�r�r�r�ri
r	
r
r��	argumentsrSr�
rTro
)s

�z,PydanticModelTransformer.get_field_argumentscC
s.|js|�
|
t|j�
�rd
S|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��is_dynamic_alias_presentr�r)
r�
rZr�)r_r�r�rSrSrTrs
Is




z1PydanticModelTransformer.should_init_forbid_extrar)
cC
s8|D]}|jr

d
Sq|
r|D]
}|j
dur
d
SqdS)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�rSrSrTr�
Vs

�



�z1PydanticModelTransformer.is_dynamic_alias_presentN)
r�rr�r�r�r0rZrYrPrW)rPr�)rPrW)rPr
)r
r
r
r�rPr7
)r1
rrPrF
)rJ
rrPrF
)r1
rr
r
r
rK
rPrL
)rrr%r�rOr�rrPr�)
r�rh
r�r
r	
r�r
r�rPrW)r�rh
r�r0r�r�rPrWr�)r�rOr�rr
r�rPr�
)r1
rrPr�)r1
rrPr�)r1
rrPr�
)r�rh
r�r�r�r�r�r�ri
r�r	
r�r
r�r�r�rPr�
)r�rh
r�r
rPr�)r�rh
r)
r�rPr�)r�r�r�r�r�r�r^r�r
r

r
rI
r;
r<
rY
r
r
r
r!
�staticmethodrR
rS
rW
rX
ro
rs
r�
rSrSrSrTr�
s<


	

$

L
D



&
/
(#



!�
 

rc@
sFeZ
dZd
Z						ddd
d�Zddd�Zddd�Zddd�ZdS) r
z(Pydantic mypy plugin model config class.Nr�
r�r�r�r�r)
r�cC
s(|
|_||_
||_||_||_||_dSrX)r�
r�r�r�r)
r�)r_r�
r�r�r�r)
r�rSrSrTr^hs	





zModelConfigData.__init__rPr}c

C
sd
d�|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`.
        c
S
si|]\}
}|du
r|
|�qSrXrS)rh�
k�
vrSrSrTr�}sz3ModelConfigData.get_values_dict.<locals>.<dictcomp>)�__dict__r 
r�rSrSrTr
x�zModelConfigData.get_values_dictr�r�
rWcC
s2|
d
urd
S|
���
�D]
\}}t|||�
qd
S)z$Update Pydantic model config values.N)r
r 
r�)r_r�r�
r�
rSrSrTr"
s



�zModelConfigData.updater�rOr

rcC
s"t||
�d
urt
||
|�
d
Sd
S)zFSet default value for Pydantic model config if config value is `None`.N)r�r�)r_r�r

rSrSrTr,
�s
�zModelConfigData.setdefault)NNNNNN)r�
r�r�r�r�r�r�r�r)
r�r�r�r�)r�r�
rPrW)r�rOr

rrPrW)r�r�r�r�r^r
r"
r,
rSrSrSrTr
es





�

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�rrWcC
�|
jd
|�d�|t
d�
dS)zCEmits an error when the model does not have `from_attributes=True`.r]
z$" does not have from_attributes=True�
�codeN)r(
�	ERROR_ORM)r�
r�r�rSrSrTr���r�r�r0cC
r�
)z0Emits an error when the config value is invalid.zInvalid value for "Config.r]
r�
N)r(
�ERROR_CONFIG)r�r�r�rSrSrTr�
�r�
r�
cC
�|jd
|
t
d�
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�rSrSrTr*
�r�
r*
r��8CheckerPluginInterface | SemanticAnalyzerPluginInterfacecC
s4d
}d|�d�}|d|�d�7}|
j||t
d�
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_messagerSrSrTr��s


r�cC
r�
)z;Emits an error when there is an untyped field in the model.zUntyped fields disallowedr�
N)r(
�
ERROR_UNTYPEDr�
rSrSrTrP
�r�rP
cC
r�
)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�
rSrSrTr=
�r�r=
F�8SemanticAnalyzerPluginInterface | CheckerPluginInterfacer�rr�r�
rm
r�r��tvar_def�TypeVarType | Noner}
r�cC
s|
j}||j
vr|j
|}	|	jrt|	jt�r|
jj�|	j�

t|t	�r)|�
d
�
}
n|�d
g�}
|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|_||j
vr�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�r
rn
rmrnrr#
r$
�remover0r~
�named_generic_typer<r@rr)r
r�rr
r�r�r�r7�	variablesrr r2r�r
rbr�
r�rA�is_decoratedr}
rrr%r�defn)r�r�r�r�rm
r�r�
r}
r�rr�
function_type�firstrb
r
ra
r��	signaturer
�r_namer�
�decrSrSrTrt
�sZ



































rt
r��dict[str, Any] | Nonec
C
s�|�d
�
sdSt
jdkrddl}
nzddl}
Wnty*


ddl}|�d�

YdSwt|d��
}|
�	|�
Wd�
S1s@w



Y
dS)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)��r
zJNo TOML parser installed, cannot read configuration from `pyproject.toml`.�rb)
ry�sys�version_info�tomllib�tomli�ImportError�warnings�warn�open�load)r��toml_r�
�rfrSrSrTr�s









�
$�r�)rNrOrPrQ)r|r-rPr:)r�
rOr�r+r�rrPrW)r�rOr�r0r�rrPrW)r�r0r�rrPrW)r�rOr�r�
r�rrPrW)r�r+r�rrPrW)NNF)r�r�
r�rr�rOr�r�
rm
r:r�r�r�
r�
r}
r�rPrW)r�rOrPr�
)|r��
__future__rr�
�configparserr�typingrrr�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.typeopsr5�
mypy.typesr6r7r8r9r:r;r<r=r>r?�
mypy.typevarsr@�	mypy.utilrA�mypy.versionrB�mypy_version�pydantic._internalrC�pydantic.versionrDr�r�rfr�r�rtr�
�DATACLASS_FULLNAMEr
r
�MYPY_VERSION_TUPLEr
rUrRrYrzr�r�rr
r�
r�
r�
r�
r�
�ERROR_FIELD_DEFAULTSr�
r�r�
r*
r�rP
r=
rt
r�rSrSrSrT�<module>
s�


�
" 



0















6
.V'













�X