
    +DCf                       d Z ddlmZ ddlZddlmZ ddlmZmZm	Z	 ddl
mZ ddlmZmZ ddl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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-m.Z. dd	l/m0Z0 dd
l1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8 ddl9m:Z: ddl;m<Z< ddl=m>Z> ddl?m@Z@ ddlAmBZB ddlCmDZD ddlEmFZFmGZGmHZHmIZImJZJmKZKmLZLmMZMmNZNmOZOmPZP ddlQmRZR ddlSmTZT ddlUmVZW ddlXmYZY ddlZm[Z[ 	 ddlEm\Z\ dZ^dZ_dZ`dZadZbdZcdZdd Zed!Zfh d"Zg e[eW      Zhehd#k\  rd$nd%Zid&ZVdKd'Zj G d( d)e6      Zk G d* d+      ZldLd,Zm G d- d.      Zn G d/ d0      Zo G d1 d2      Zp G d3 d4      Zq ed5d6d7      Zr ed8d9d7      Zs ed:d;d7      Zt ed<d=d7      Zu ed>d?d7      Zv ed>d@d7      Zw ed>dAd7      ZxdMdBZydNdCZzdOdDZ{	 	 	 	 	 	 	 	 dPdEZ|dOdFZ}dQdGZ~dQdHZ	 	 	 dR	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 dSdIZdTdJZy# e]$ r
 ddlEmNZ\ Y w xY w)Uz^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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FunctionContextMethodContextPluginReportConfigContextSemanticAnalyzerPluginInterface)dataclasses)deserialize_and_fixup_type)set_callable_name)make_wildcard_trigger)state)map_type_from_supertype)AnyTypeCallableTypeInstanceNoneType
OverloadedType	TypeOfAnyTypeTypeTypeVarType	UnionTypeget_proper_type)fill_typevars)get_unique_redefinition_name)__version__)_fields)parse_mypy_version)
TypeVarDef)r@   pydantic-mypyzpydantic-mypy-metadatazpydantic.main.BaseModelz#pydantic_settings.main.BaseSettingszpydantic.root_model.RootModelz5pydantic._internal._model_construction.ModelMetaclasszpydantic.fields.Fieldzpydantic.dataclasses.dataclass.pydantic.functional_validators.model_validator>   *pydantic.functional_serializers.serializer.pydantic.deprecated.class_validators.validator.pydantic.functional_validators.field_validator0pydantic.functional_serializers.model_serializer3pydantic.deprecated.class_validators.root_validatorrJ   )r   i  builtins__builtins__   c                    t         S )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)versions    M/var/www/highfloat_scraper/venv/lib/python3.12/site-packages/pydantic/mypy.pypluginrW   r   s
         c                  l     e Zd ZdZd fdZddZddZddZddZddZ	ddZ
dd	Zdd
ZddZ xZS )rT   zThe Pydantic mypy plugin.c                    t        |      | _        | j                  j                         | _        t        |   |       y N)PydanticPluginConfigplugin_configto_data_plugin_datasuper__init__)selfoptions	__class__s     rV   ra   zPydanticPlugin.__init__   s4    1': ..668!rX   c                    | j                  |      }|rLt        |j                  t              r2t	        d |j                  j
                  D              r| j                  S y)zUpdate Pydantic model class.c              3  B   K   | ]  }|j                   t        k(    y wr[   )fullnameBASEMODEL_FULLNAME.0bases     rV   	<genexpr>z5PydanticPlugin.get_base_class_hook.<locals>.<genexpr>   s     P44==$66P   N)lookup_fully_qualified
isinstancenoder(   anymro$_pydantic_model_class_maker_callbackrb   rg   syms      rV   get_base_class_hookz"PydanticPlugin.get_base_class_hook   sG    ))(3:chh1P388<<PP@@@rX   c                .    |t         k(  r| j                  S y)z,Update Pydantic `ModelMetaclass` definition.N)MODEL_METACLASS_FULLNAME)_pydantic_model_metaclass_marker_callbackrb   rg   s     rV   get_metaclass_hookz!PydanticPlugin.get_metaclass_hook   s    //AAArX   c                h    | j                  |      }|r|j                  t        k(  r| j                  S y)z/Adjust the return type of the `Field` function.N)rn   rg   FIELD_FULLNAME_pydantic_field_callbackrt   s      rV   get_function_hookz PydanticPlugin.get_function_hook   s/    ))(33<<>1000rX   c                2    |j                  d      rt        S y)z-Adjust return type of `from_orm` method call.z	.from_ormN)endswithfrom_attributes_callbackrz   s     rV   get_method_hookzPydanticPlugin.get_method_hook   s    [)++rX   c                H    |t         k(  rt        dk  rt        j                  S y)zMark pydantic.dataclasses as dataclass.

        Mypy version 1.1.1 added support for `@dataclass_transform` decorator.
        )   r   N)DATACLASS_FULLNAMEMYPY_VERSION_TUPLEr2   dataclass_class_maker_callbackrz   s     rV   get_class_decorator_hookz'PydanticPlugin.get_class_decorator_hook   s$    
 )).@6.I===rX   c                    | j                   S )zjReturn all plugin config data.

        Used by mypy to determine if cache needs to be discarded.
        )r_   )rb   ctxs     rV   report_config_dataz!PydanticPlugin.report_config_data   s    
    rX   c                    t        |j                  |j                  |j                  | j                        }|j                         S r[   )PydanticModelTransformerclsreasonapir]   	transform)rb   r   transformers      rV   rs   z3PydanticPlugin._pydantic_model_class_maker_callback   s3    .sww

CGGTM_M_`$$&&rX   c                    | j                   j                  ry|j                  j                  j                  }|sJ d       t        |j                  dd      rd|j                  _        yy)zReset 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_transformr   infodeclared_metaclassgetattrtyper   )rb   r   info_metaclasss      rV   ry   z8PydanticPlugin._pydantic_model_metaclass_marker_callback   s^     7788NNN~>&&(BDI;?N8 JrX   c                |  	 |j                   	|j                  d   dk(  sJ d       |j                  d   dk(  sJ d       |j                  d   }|j                  d   }|r$|r"t        |j                  |j
                         	S |r+|j                  d   d   }|d   }t        |t              s|S 	S |r|j                  d   d   }t        |t              r|j                  d   }t        |t              rH|j                  }t        |dd      }|r+t        d	 |D              rt        	fd
|D              |_        |S 	S )ao  Extract the type of the `default` argument from the Field function, and use it as the return type.

        In particular:
        * Check whether the default and default_factory argument is specified.
        * Output an error if both are specified.
        * Retrieve the type of the argument which is specified, and use it as return type for the function.
        r   defaultz0"default" is no longer first argument in Field()r   default_factoryz9"default_factory" is no longer second argument in Field()argsNc              3  <   K   | ]  }t        |t                y wr[   )ro   r@   )rj   args     rV   rl   z:PydanticPlugin._pydantic_field_callback.<locals>.<genexpr>   s     HC:c;7Hs   c              3  "   K   | ]  }  y wr[    )rj   _default_any_types     rV   rl   z:PydanticPlugin._pydantic_field_callback.<locals>.<genexpr>   s     -M1.>-Ms   )default_return_typecallee_arg_namesr   +error_default_and_default_factory_specifiedr   context	arg_typesro   r   r<   itemsr9   ret_typer   alltuple)
rb   r   default_argsdefault_factory_argsdefault_typedefault_argdefault_factory_typer   r   r   s
            @rV   r~   z'PydanticPlugin._pydantic_field_callback   sM    22##A&)3g5gg3##A&*;;x=xx;xx{"xx{07M##==+A.L&q/K k<8##*  ' "#&==#3A#6  .
;';'A'A!'D$.=/88 x6H4HH(--M-M(MrX   rc   r*   returnNone)rg   strr   z(Callable[[ClassDefContext], bool] | None)rg   r   r   z(Callable[[ClassDefContext], None] | None)rg   r   r   z(Callable[[FunctionContext], Type] | None)rg   r   r   z&Callable[[MethodContext], Type] | None)r   r0   r   dict[str, Any])r   r,   r   bool)r   r,   r   r   )r   r-   r   r=   )__name__
__module____qualname____doc__ra   rv   r{   r   r   r   r   rs   ry   r~   __classcell__)rd   s   @rV   rT   rT      s9    #"
!'@. rX   rT   c                  N    e Zd ZU dZdZded<   ded<   ded<   ded<   ddZdd	Zy
)r\   a  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   r   r   r   r   r   c                   |j                   y t        |j                         }|q|j                  di       j                  di       }| j                  D ]?  }|j                  |d      }t	        |t
              st        d|       t        | ||       A y t               }|j                  |j                          | j                  D ]'  }|j                  t        |d      }t        | ||       ) y )NtoolrI   Fz/Configuration value must be a boolean for key: )fallback)config_file
parse_tomlget	__slots__ro   r   
ValueErrorsetattrr   read
getbooleanCONFIGFILE_KEY)rb   rc   toml_configconfigkeysettingr]   s          rV   ra   zPydanticPluginConfig.__init__  s    & !4!45" __VR044_bIF~~ , **S%0!'40$'VWZV[%\]]c7+	, )NMw223~~ ,'22>3QV2Wc7+,rX   c                V    | j                   D ci c]  }|t        | |       c}S c c}w )z/Returns a dict of config names to their values.)r   r   )rb   r   s     rV   r^   zPydanticPluginConfig.to_data   s&    37>>BCWT3''BBBs   &Nr   r   r   )r   r   r   r   r   __annotations__ra   r^   r   rX   rV   r\   r\      s3    I #''##,&CrX   r\   c                   | j                   }t        |t              r|j                  }t        |t              r't        |j
                  t              r|j
                  }n]t        |t              r|}nJd| d|j                  j                   d}t        || j                  | j                         | j                  S |j                   j                  j                  t              }|| j                  S |j                  di       j                  d      }|dur5t!        |j                   j"                  | j                  | j                         | j                  S )z1Raise an error if from_attributes is not enabled.z
ctx.type: 
 (of type )r   from_attributesT)r   ro   r?   itemr9   r   r:   rd   r   error_unexpected_behaviorr   r   r   metadatar   METADATA_KEYerror_from_attributesname)r   ctx_type
model_typedetailpydantic_metadatar   s         rV   r   r   %  s    xxH(H%==(L)j9J9JH.U&&
	Hh	'
hZz(2D2D2M2M1NaP!&#''3;;?&&&"0044\B &&&'++Hb9==>OPOd"joo22CGGS[[I"""rX   c                      e Zd ZdZ	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 d
dZ	 	 	 	 	 	 	 	 	 	 	 	 ddZddZddZddZe	dd       Z
ddZy	)PydanticModelFieldz5Based on mypy.plugins.dataclasses.DataclassAttribute.c	                t    || _         || _        || _        || _        || _        || _        || _        || _        y r[   )r   aliashas_dynamic_aliashas_defaultlinecolumnr   r   )	rb   r   r   r   r   r   r   r   r   s	            rV   ra   zPydanticModelField.__init__?  s>     	
!2&			rX   c                    | j                  |||      }|r| j                  ||      nt        t        j                        }t        ||d|s| j                  rt              S t              S )zABased on mypy.plugins.dataclasses.DataclassAttribute.to_argument.N)variabletype_annotationinitializerkind)	to_varr	   r8   r>   explicitr   r   r   r   )rb   current_infotypedforce_optional	use_aliasr   r   r   s           rV   to_argumentzPydanticModelField.to_argumentS  sl     ;;|S)<AF$**<=GT]TfTfLg+"0D4D4D	
 	
 KT	
 	
rX   c                r   | j                   t        | j                  dd      rt        j                  |j
                  j                        5  t        | j                   | j                  j                  j                  t        |      i      cddd       S | j                   S # 1 sw Y   | j                   S xY w)zABased on mypy.plugins.dataclasses.DataclassAttribute.expand_type.N	self_type)r   r   r   r6   strict_optional_setrc   strict_optionalr	   r   idrC   )rb   r   r   s      rV   r	   zPydanticModelField.expand_typee  s     99 WTYYT%J%V
 **3;;+F+FG e"499tyy/B/B/E/E}UaGb.cde eyyeyys   ?B""B6c                    |r| j                   | j                   }n| j                  }t        || j                  ||            S )z<Based on mypy.plugins.dataclasses.DataclassAttribute.to_var.)r   r   r)   r	   )rb   r   r   r   r   s        rV   r   zPydanticModelField.to_varq  s:    /::D99D4)),<==rX   c                    | j                   s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   	serializerb   s    rV   r   zPydanticModelField.serializez  sV    yyyIIZZ!%!7!7++IIkkII'')
 	
rX   c                n    |j                         }t        |j                  d      |      } | d||d|S )ABased on mypy.plugins.dataclasses.DataclassAttribute.deserialize.r   )r   r   r   )copyr3   pop)r   r   datar   typs        rV   deserializezPydanticModelField.deserialize  s9     yy{(&)93?/$/$//rX   c                    | j                   Yt        j                  |j                  j                        5  t        | j                   || j                        | _         ddd       yy# 1 sw Y   yxY w)zxExpands type vars in the context of a subtype when an attribute is inherited
        from a generic super type.
        N)r   r6   r   rc   r   r7   r   )rb   sub_typer   s      rV   expand_typevar_from_subtypez.PydanticModelField.expand_typevar_from_subtype  sb     99 **3;;+F+FG T3DIIxS	T T !T Ts   'A''A0N)r   r   r   z
str | Noner   r   r   r   r   intr   r  r   Type | Noner   r(   )r   r(   r   r   r   r   r   r   r   r1   r   r   )r   r(   r   r1   r   r  )r   r(   r   r1   r   r   r   r)   r   r   )r   r(   r  r   r   r1   r   r   )r
  r(   r   r1   r   r   )r   r   r   r   ra   r   r	   r   r   classmethodr  r  r   rX   rV   r   r   <  s    ?   	
     (

 
 	

 
 -
 

$
>
 0 0TrX   r   c                  0    e Zd ZdZd Zedd       ZddZy)PydanticModelClassVarzBased on mypy.plugins.dataclasses.DataclassAttribute.

    ClassVars are ignored by subclasses.

    Attributes:
        name: the ClassVar name
    c                    || _         y r[   r   )rb   r   s     rV   ra   zPydanticModelClassVar.__init__  s	    	rX   c                2    |j                         } | di |S )r  r   )r  )r   r  s     rV   r  z!PydanticModelClassVar.deserialize  s     yy{{T{rX   c                    d| j                   iS )r   r   r  r  s    rV   r   zPydanticModelClassVar.serialize  s     DII
 	
rX   N)r  r   r   r  r  )r   r   r   r   ra   r  r  r   r   rX   rV   r  r    s%      

rX   r  c                  @   e Zd ZU dZh dZded<   	 	 	 	 	 	 	 	 	 	 ddZddZddZddZ		 	 	 	 	 	 dd	Z
dd
ZddZ	 	 	 	 	 	 	 	 ddZd dZ	 	 	 	 	 	 	 	 	 	 d!dZ	 	 	 	 	 	 	 	 d"dZd#dZd$d%dZed&d       Zed'd       Z	 	 	 	 	 	 	 	 	 	 	 	 d(dZd)dZed*d       Zy)+r   zTransform 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frozenalias_generatorr   populate_by_namezset[str]tracked_config_fieldsc                <    || _         || _        || _        || _        y r[   )_cls_reason_apir]   )rb   r   r   r   r]   s        rV   ra   z!PydanticModelTransformer.__init__  s!     		*rX   c                   | j                   j                  }t        d |j                  dd D              }| j	                         }| j                  ||      \  }}||y|D ]  }|j                   y t        d |j                  dd D              }| j                  ||||       | j                  |||       | j                  || j                  |j                  du        | j                          |D ci c]  }|j                  |j                          c}|D ci c]  }|j                  |j                          c}|j                         d|j                   t"        <   yc c}w c c}w )	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              3  @   K   | ]  }t         |j                  v   y wr[   )ROOT_MODEL_FULLNAMErg   ri   s     rV   rl   z5PydanticModelTransformer.transform.<locals>.<genexpr>  s     [T/4==@[s   NFc              3  B   K   | ]  }|j                   t        k(    y wr[   )rg   BASESETTINGS_FULLNAMEri   s     rV   rl   z5PydanticModelTransformer.transform.<locals>.<genexpr>  s     [T$--+@@[rm   T)r  )fields
class_varsr   )r  r   rq   rr   collect_configcollect_fields_and_class_varsr   add_initializeradd_model_construct_method
set_frozenr  r  adjust_decorator_signaturesr   r   get_values_dictr   r   )	rb   r   is_root_modelr   r&  r'  fieldis_settings	class_vars	            rV   r   z"PydanticModelTransformer.transform  sV    yy~~[TXXVYWY][[$$&!??V
>Z/ 	Ezz!	 [TXXVYWY][[VV[-H''D		&--42GH((* CIIuzz5??#44IR\]Y9>>9+>+>+@@],,.'
l#  J]s   9"E)!"E.c                ,   | j                   j                  j                  j                         D ]  \  }}t	        |j
                  t              s!|j
                  j                  d   t	        t              sKt	        j                  t              sfj                  j                  t        v sj                  j                  t        k(  r(t        fdt        j                         D              rd|j
                  j"                  _         y)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     K   | ]<  \  }}j                   |   d k(  xr! t        |t              xr |j                  dk(   > yw)modeafterN)	arg_namesro   r$   value)rj   ir   	first_decs      rV   rl   zGPydanticModelTransformer.adjust_decorator_signatures.<locals>.<genexpr>  sO        &3 &//2f<rCQXAYr^a^g^gkr^rr s   AATN)r  r   namesr   ro   rp   r   original_decoratorsr   calleer   rg   DECORATOR_FULLNAMESMODEL_VALIDATOR_FULLNAMErq   	enumerater   funcis_class)rb   r   ru   r:  s      @rV   r-  z4PydanticModelTransformer.adjust_decorator_signatures  s     --335 	2ID##((I.HH88;	y(3"9#3#3X>!((115HH "((115MM  *3INN*C   .2CHHMM*#	2rX   c           	        | j                   }t               }d}d}|j                  j                         D ]-  \  }}| j	                  ||      }|sd}|j                  |       / d}|j                  j                  D ]  }t        |t        t        f      st        |t              r"|j                  d   }	t        |	t              r|	j                  dk7  r[t        |j                  t              rdt!        |j                  j"                  |j                  j$                        D ]+  \  }
}|
	|j                  | j	                  |
|d             - nt        |j                  t&              r|j                  j                  D ]A  \  }}t        |t(              s|j                  | j	                  |j*                  |             C nt        |t              r|j                  dk7  ro|j                  j                  D ]h  }t        |t              s|j                  d   }	t        |	t              s4|j                  | j	                  |	j                  |j                               j |r| j,                  j/                  d|        nd} |s|rF|rD|j0                  r8|j2                  s,| j4                  j6                  rt9        | j,                  |       |j:                  j<                  d	d D ]  }t>        |j@                  vr| j,                  jC                  tE        |jF                               |j@                  t>           d
   j                         D ]  \  }}|jI                  ||         |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 kwargsr   r   )%r  ModelConfigDatakeywordsr   get_config_updateupdatedefsbodyro   r   r   lvaluesr   r   rvaluer   zipr7  r   r   r$   r8  r  failhas_alias_generatorr  r]   r   error_required_dynamic_aliasesr   rr   r   r   add_plugin_dependencyr5   rg   
setdefault)rb   r   r   has_config_kwargshas_config_from_namespacer   exprconfig_datastmtlhsarg_namer   key_expr
value_exprsubstmtr   r8  s                    rV   r(  z'PydanticModelTransformer.collect_config  s   ii "!$)! ,,,,. 	+JD$00t<K$(!k*		+ "&HHMM &	-Dd^X$>?$/ll1o!#x0CHH4Ndkk84),T[[-B-BDKKDTDT)U ]##+$d&<&<XsVZ&<&[\]  X6040A0A Z,*)(G<$d&<&<X^^Z&XYZ
 D(+99(#yy~~ TG%g~> !//!,C%c84 MM$"8"87>>"RST !		o (,%M&	-P  9..//&&DD.tyy$?HHLL$ 	/D4==0 II++,A$--,PQ#}}\:8DJJL /e!!$./	/ rX   c                   | j                   }i }i }t        |j                  j                  dd       D ]v  }t        |j
                  vr| j                  j                  t        |j                               |j
                  t           d   j                         D ]  \  }}t        j                  ||| j                        }	|	j                  |j                  | j                         |	||<   |j                  j                  j                  |      }
|
sz|
j                   st#        |
j                   t$              r| j                  j'                  d|
j                           |j
                  t           d   j                         D ]  \  }}t(        j                  |      ||<    y t+               }t+               }| j-                  |j.                        D ]  }| j1                  |||      }t#        |t              rb|j2                  d   }|r&|j4                  dk7  rt7        | j                  |       ]|j9                  |j4                         |||j4                  <   t#        |t(              s|j2                  d   }|j9                  |j4                         |||j4                  <    t;        |j=                               t;        |j=                               fS )zACollects the fields for the model, accounting for parent classes.r   r#  r&  z7BaseModel field may only be overridden by another fieldr'  r   root)r  reversedr   rr   r   r   r  rS  r5   rg   r   r   r  r  r;  r   rp   ro   r)   rP  r  set%_get_assignment_statements_from_blockrK  $collect_field_or_class_var_from_stmtrM  r    error_extra_fields_on_root_modeladdlistvalues)rb   rD  r/  r   found_fieldsfound_class_varsr   r   r  r0  sym_nodecurrent_field_namescurrent_class_vars_namesrY  maybe_fieldrZ  s                   rV   r)  z6PydanticModelTransformer.collect_fields_and_class_varsW  s]    ii 79=?SXX\\!B/0 	QD 4==0 II++,A$--,PQ"mmL9(CIIK 
d*66tT499M
 11#((DIIF%*T"88>>--d3jPS6TIINNQ   #mmL9,GMMO Q
d)>)J)J4)P &Q5	Q< ),-0U >>sxxH 	9DCCD,XhiK+'9:ll1o SXX%74TYYE'++CHH5-8L*K)>?ll1o(,,SXX6-8 *	9 L'')*D1A1H1H1J,KKKrX   c              #    K   |j                   D ](  }|j                  r| j                  |      E d {    * |j                  ;|j                  j                  s$| j                  |j                        E d {    y y y 7 N7 	wr[   )rL  is_unreachablerc  	else_body)rb   rY  rL  s      rV   ,_get_assignment_statements_from_if_statementzEPydanticModelTransformer._get_assignment_statements_from_if_statement  s{     II 	LD&&EEdKKK	L >>%dnn.K.KAA$..QQQ /L% LQs'   BBBAB:B;BBc              #     K   |j                   D ]A  }t        |t              r| t        |t              s)| j	                  |      E d {    C y 7 wr[   )rL  ro   r   r   rr  )rb   blockrY  s      rV   rc  z>PydanticModelTransformer._get_assignment_statements_from_block  sN     JJ 	SD$/
D&)LLTRRR		S Ss   5AAAAc           
        | j                   }|j                  d   }t        |t              r.t	        j
                  |j                        r|j                  dk(  ry|j                  st        |j                  t              rt        |j                  j                  t              r_t        |j                  j                  j                  t              r1|j                  j                  j                  j                  t        v ry|j                  |v ryt        | j                  |       y|j                  d   }t        |t              syt	        j
                  |j                        r|j                  dk(  ry|j                  j                   j#                  |j                        }|y|j$                  }t        |t&              ryt        |t(              r| j                  j+                  d|       yt        |t,              sy|j.                  rt1        |j                        S t3        |j4                        }t        |t6              r5|j4                  j                  dk(  r| j                  j+                  d|       | j9                  |      }	|j4                  |j:                  rw|j<                  rk| j                  j?                  |j                  d      }
|
r|
|_        n:| j                  j+                  d	|       tA        tB        jD                        |_        | jG                  |      \  }}|r8|jH                  s,| jJ                  jL                  rtO        | j                  |       | jQ                  ||j                  |      }tS        |j                  ||	||jT                  |jV                  ||j                  
      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   rD  NzFType aliases inside BaseModel definitions are not supported at runtimezdataclasses.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  rM  ro   r   rF   is_valid_field_namer   
new_syntaxrN  r   r=  rg   r>  error_untyped_fieldsr  r   r;  r   rp   r!   r'   rP  r)   is_classvarr  rB   r   r:   get_has_defaultrv  is_inferredanalyze_simple_literal_typer8   r>   
from_errorget_alias_infor  r]   r   rR  _infer_dataclass_attr_init_typer   r   r   )rb   rY  rD  r'  r   rZ  ru   rp   	node_typer   r  r   r   	init_types                 rV   rd  z=PydanticModelTransformer.collect_field_or_class_var_from_stmt  s    iill1o#x(0K0KCHH0UY\YaYaesYs4;;1t{{118<t{{1188(CKK&&--66:MM
 xx:% !D1ll1o#x(**3884N8Rhhnn  *; xxdO,
 dI&IINNX $$
  (22 $DII.	i*y~~/F/FJ_/_IINN7
 **4088$2B2B ))77d7SC			] $I$8$89	#'#6#6t#<  \%B%BtGYGYGwGw*499d;88chhM	!/#;;	
 		
rX   c                8   |j                   }|j                  r|S t        |j                         }t        |t              s|S |j                   j                  d      }|r:t        |j                  t              r|j                   j                  d      }|sJ |j                   r+t        t        |j                   |j                   |            }nt        t        j                        S t        |t              r7|j                  t        t        t        gk(  rt!        |j"                  d   |      S | j$                  j'                  d|j                   j(                   d|       |S | j$                  j'                  d|j                   j(                   d|       |S )zvInfer __init__ argument type for an attribute.

        In particular, possibly use the signature of __set__.
        __set__rR   z(Unsupported signature for "__set__" in ""zUnsupported "__set__" in ")r   implicitrB   ro   r:   r   rp   r   get_containing_type_infor7   r8   r>   unannotatedr9   	arg_kindsr   r
   r   r  rP  r   )	rb   ru   r   r   r   tsetter
super_infosetter_types	            rV   r  z8PydanticModelTransformer._infer_dataclass_attr_init_type#  sG   
 ((<<NCHH% !X&NI&&++w/VV<<YG
!!z;;"12I&++WXW]W]_i2j"kK"9#8#899k<8[=R=RW >
 3;3H3H3KQOOIINN%Maffkk]Z[#\^ef  		!;AFFKK=JGTrX   c           	        d| j                   j                  j                  v r.| j                   j                  j                  d   j                  sy| j                  j
                  }|j                  du}t        |j                  xr |j                         }| j                  |||||      }|r3t        dk  r*|d   j                  t        k(  rt        nt        |d   _        |r| j                  j!                  t"              j$                  }	d|	j                  v r|	j                  d   j$                  }
|
|
j&                  |
j&                  }t)        |j*                        D ]}  \  }}|j-                  d      s|j-                  d      s)| j                  j/                  |j0                  |         }t3        ||      }|j5                  t7        ||dt                      | j9                  ||      sCt3        d	      }|j5                  t7        |t;        t<        j>                        dt@                     tC        | j                  | j                   d|tE               
       y)zAdds a fields-aware `__init__` method to the class.

        The added `__init__` will be annotated with types vs. all `Any` depending on the plugin settings.
        ra   NTr   requires_dynamic_aliasesr   r1  )r   r   r   r   __r   kwargs)r   return_type)#r  r   r;  plugin_generatedr]   r   r  r   rQ  get_field_argumentsr   r   r   r   r   r  rn   r%  rp   r   r@  r7  
startswith	anal_typer   r)   appendr   should_init_forbid_extrar8   r>   r   r   
add_methodr;   )rb   r&  r   r1  r/  r   r   r  r   base_settings_nodebase_settings_init_node	func_typearg_idxr[  analyzed_variable_typer   vars                    rV   r*  z(PydanticModelTransformer.add_initializerI  s    ---diinn6J6J:6V6g6g""--++47	#'(B(B(b6KbKbGb#c ''%=# ( 
 /9< '+1glli&?7WDGL!%!A!ABW!X!]!]/555*<*B*B:*N*S*S'*6;R;W;W;c 7 < <I-6y7J7J-K _)#..t4H<O<OPS<T$151D1DYEXEXY`Ea1b.#&x1G#HHX7MtU\$]^_ ,,VV<h-CKKgi.@.@&A4ST499dii$HJWrX   c           	        | j                   j                  t         d| j                   j                  t         d      g      }t        |t	               g      }t        t        d|      |dt              }t        j                  | j                   j                  j                        5  | j                  |ddd|      }ddd       | j                  ||      sCt        d      }j                  t        |t        t         j"                        dt$                     |gz   }t'        | j                   | j(                  d	|t+        | j(                  j,                        d
       y# 1 sw Y   xY w)zAdds 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_setNTFr  r  model_construct)r   r  is_classmethod)r  
named_typeBUILTINS_NAMErA   r;   r   r)   r   r6   r   rc   r   r  r  r  r8   r>   r   r   r  r  rC   r   )	rb   r&  r   r1  set_stroptional_set_strfields_set_argumentr   r  s	            rV   r+  z3PydanticModelTransformer.add_model_construct_methodv  s8    ))&&-'=		@T@TXeWffjUk@l?mn$gxz%:;&s=:J'KM]_celm&&tyy'8'8'H'HI 	++dUeal , D	 ,,VV<h-CKKgi.@.@&A4ST#$t+IIII%diinn5	
	 	s   )E&&E/c                   | j                   j                  }|D ]:  }|j                  j                  |j                        }||j
                  }t        |t              r||_        Ot        |t              r1| j                  j                  s| j                  j                          	 t        |      }d| d|j                    d}	t#        |	| j                  | j                          |j%                  ||d      }||_        ||_        |j&                  dz   |j                  z   |_        t+        t,        |      |j                  |j                  <   = y# t        $ r t        |      }Y w xY w)zMarks 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.
        Nzsym_node.node: r   r   F)r   .)r  r   r;  r   r   rp   ro   r)   is_propertyr!   r  final_iterationdeferr   	TypeErrorreprrd   r   r   rg   	_fullnamer%   r   )
rb   r&  r   r  r   r0  rk  r  var_strr   s
             rV   r,  z#PydanticModelTransformer.set_frozen  s'   
 yy~~ 	BEzz~~ejj1H#mmc3'&,CO_5dii>W>WIIOO%,"%c(  /wiz#--PQRF-fdiiKll4l>"( $ 3chh >'6tS'A

388$/	B % ,"&s),s   *EE/.E/c                   || j                   vry|dk(  rft        |t              r|j                  dk(  }n:t        |t              r|j
                  dk(  }n|st        || j                  |       yt        |      S |dk(  r/d}t        |t              r|j                  dk(  rd}t        |	      S t        |t              r(|j                  d
v rt        di ||j                  dk(  iS t        || j                  |       y)zDetermines 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)rQ  )builtins.Truezbuiltins.Falser  r   )r  ro   r$   r8  r   r   error_invalid_config_valuer  rG  r   rg   )rb   r   r   rE  r  rQ  s         rV   rI  z*PydanticModelTransformer.get_config_update  s    
 t1117?#w'"yyH4C,"xx83  /tTYYD"==$$"&#x(S\\_-L&+#"7JKKc8$9\)\"MdCLLO,K%LMM"4C8rX   c                   | j                   }t        |t              ryt        |t              rt        |j                  t
              r|j                  j                  t        k(  rmt        |j                  |j                        D ]I  \  }}||dk(  r|j                  t        uc S |dk(  s't        |t              xr |j                  dk(   c S  yt        |t               S )zUReturns a boolean indicating whether the field defined in `stmt` is a required field.Fr   r   r  )rN  ro   r&   r   r=  r"   rg   r}   rO  r   r7  rd   r   r   )rY  rW  r   r   s       rV   r{  z(PydanticModelTransformer.get_has_default  s     {{dH%dH%*T[['*Jt{{OcOcguOu
 !DNN; _	T<49#4==<<,, *3 9 ]cllo>]^^_ dL111rX   c                t   | j                   }t        |t              ryt        |t              r7t        |j                  t
              r|j                  j                  t        k(  syt        |j                        D ];  \  }}|dk7  r|j                  |   }t        |t              r|j                  dfc S  y y)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)rN  ro   r&   r   r=  r"   rg   r}   r@  r7  r   r$   r8  )rY  rW  r9  r[  r   s        rV   r  z'PydanticModelTransformer.get_alias_info  s     {{dH% tX&:dkk7+KPTP[P[PdPdhvPv $T^^4 	"KAx7"))A,C#w'yy%''!	" rX   c           
         | j                   j                  }|D cg c]4  }|r|j                  s$|j                  |||xs ||| j                        6 }}|S c c}w )zHelper 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.
        )r   r   r   r   )r  r   r   r   r  )	rb   r&  r   r   r  r1  r   r0  	argumentss	            rV   r  z,PydanticModelTransformer.get_field_arguments  sr     yy~~  

 %"9"9 7F;#II  

	 

 

s   9Ac                    |j                   s&| j                  |t        |j                              ry|j                  ry| j
                  j                  S )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   rQ  r  r]   r   )rb   r&  r   s      rV   r  z1PydanticModelTransformer.should_init_forbid_extra"  sI     &&,,VT&:T:T5UV!!333rX   c                \    | D ]  }|j                   s y |r| D ]  }|j                   y y)zReturns whether any fields on the model have a "dynamic alias", i.e., an alias that cannot be
        determined during static analysis.
        TF)r   r   )r&  rQ  r0  s      rV   r  z1PydanticModelTransformer.is_dynamic_alias_present/  sE    
  	E&&	   ;;&  rX   N)
r   r   r   zExpression | Statementr   r1   r]   r\   r   r   )r   r   )r   r   )r   rG  )rD  rG  r/  r   r   zJtuple[list[PydanticModelField] | None, list[PydanticModelClassVar] | None])rY  r   r   Iterator[AssignmentStmt])rt  r   r   r  )rY  r   rD  rG  r'  z dict[str, PydanticModelClassVar]r   z1PydanticModelField | PydanticModelClassVar | None)ru   r%   r   r   r   r   r   r  )
r&  list[PydanticModelField]r   rG  r1  r   r/  r   r   r   )r&  r  r   rG  r1  r   r   r   )r&  r  r   r1   r  r   r   r   )F)r   r   r   r   rE  r   r   ModelConfigData | None)rY  r   r   r   )rY  r   r   ztuple[str | None, bool])r&  r  r   r   r   r   r  r   r1  r   r   list[Argument])r&  r  r   rG  r   r   )r&  r  rQ  r   r   r   )r   r   r   r   r  r   ra   r   r-  r(  r)  rr  rc  rd  r  r*  r+  r,  rI  staticmethodr{  r  r  r  r  r   rX   rV   r   r     s   '8 ++ '+ -	+
 ,+ 
+"H26JX?L+?L<@?L	S?LBRS{
"{
2A{
Oo{
	:{
z$L+X.+X8G+XVZ+Xko+X	+XZ
.
8G
VZ
	
<B>!F 2 2*  6(  	
 #'  
44  rX   r   c                  N    e Zd ZdZ	 	 	 	 	 d	 	 	 	 	 	 	 	 	 ddZd	dZd
dZddZy)rG  z(Pydantic mypy plugin model config class.Nc                J    || _         || _        || _        || _        || _        y r[   )r  r  r   r  rQ  )rb   r  r  r   r  rQ  s         rV   ra   zModelConfigData.__init__A  s+     ). 0#6 rX   c                p    | j                   j                         D ci c]  \  }}|	|| c}}S c c}}w )zReturns a dict of Pydantic model config names to their values.

        It includes the config if config value is not `None`.
        )__dict__r   )rb   kvs      rV   r.  zModelConfigData.get_values_dictO  s1    
 "&!4!4!6HA!-1HHHs   
22c                p    |y|j                         j                         D ]  \  }}t        | ||        y)z$Update Pydantic model config values.N)r.  r   r   )rb   r   r  r  s       rV   rJ  zModelConfigData.updateV  s;    >**,224 	 DAqD!Q	 rX   c                8    t        | |      t        | ||       yy)zFSet default value for Pydantic model config if config value is `None`.N)r   r   )rb   r   r8  s      rV   rT  zModelConfigData.setdefault]  s     4%D#u% &rX   )NNNNN)
r  bool | Noner  r  r   r  r  r  rQ  r  r   )r   r  r   r   )r   r   r8  r   r   r   )r   r   r   r   ra   r.  rJ  rT  r   rX   rV   rG  rG  >  s[    2 %)"'+(,+/7!7 7 %	7
 &7 )7I &rX   rG  zpydantic-ormzInvalid from_attributes callPydanticzpydantic-configzInvalid config valuezpydantic-aliaszDynamic alias disallowedzpydantic-unexpectedzUnexpected behaviorzpydantic-fieldzUntyped field disallowedzInvalid Field defaultsz!Extra field on RootModel subclassc                <    |j                  d|  d|t               y)zCEmits an error when the model does not have `from_attributes=True`.r  z$" does not have from_attributes=TruecodeN)rP  	ERROR_ORM)
model_namer   r   s      rV   r   r   l  s    HHq@A7QZH[rX   c                <    |j                  d|  d|t               y)z0Emits an error when the config value is invalid.zInvalid value for "Config.r  r  N)rP  ERROR_CONFIG)r   r   r   s      rV   r  r  q  s    HH)$q17HNrX   c                4    | j                  d|t               y)znEmits required dynamic aliases error.

    This will be called when `warn_required_dynamic_aliases=True`.
    z#Required dynamic aliases disallowedr  N)rP  ERROR_ALIASr   r   s     rV   rR  rR  v  s    
 HH2G+HNrX   c                V    d}d|  d}|d| dz  }|j                  ||t               y)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)rP  ERROR_UNEXPECTED)r   r   r   linkfull_messages        rV   r   r   ~  sA    
 DDLVHTVWL<TFB[\\LHH\7)9H:rX   c                4    | j                  d|t               y)z;Emits an error when there is an untyped field in the model.zUntyped fields disallowedr  N)rP  ERROR_UNTYPEDr  s     rV   ry  ry    s    HH('HFrX   c                4    | j                  d|t               y)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)rP  ERROR_EXTRA_FIELD_ROOT_MODELr  s     rV   re  re    s    HHA7QmHnrX   c                4    | j                  d|t               y)zNEmits an error when `Field` has both `default` and `default_factory` together.z>Field default and default_factory cannot be specified togetherr  N)rP  ERROR_FIELD_DEFAULTSr  s     rV   r   r     s    HHMw]qHrrX   c                D   |j                   }||j                  v rd|j                  |   }	|	j                  rIt        |	j                  t
              r/|j                  j                  j                  |	j                         t        | t              r| j                  d      }
n| j                  dg       }
|r6|xs t        t        |            }t        t        d      |dt         d      g}n+|xs t        |      }t        t        d      |dt               g}||z   }g g g }}}|D ]p  }|j"                  sJ d       |j%                  |j"                         |j%                  |j&                  j(                         |j%                  |j*                         r t-        |||||
      }|r|g|_        t        ||t1        t3               g            }||_         t5        ||      |_        ||_        |j:                  dz   |z   |_        |j>                  |_        ||j                  v r2tA        ||j                        }|j                  |   |j                  |<   |rud|_!        t        ||j6                        }||_         |j<                  |_        d|_"        tG        |tI        d      g|      }|j>                  |_        tK        tL        |      }	ntK        tL        |      }	d|	_        |	|j                  |<   |jN                  j                  j                  j%                  |       y)	zhVery closely related to `mypy.plugins.common.add_method_to_class`, with a few pydantic-specific changes.zbuiltins.functionr  NT__pydantic_self__z"All arguments must be fully typed.r  r  )(r   r;  r  ro   rp   r   rK  rL  remover1   r  named_generic_typer?   rC   r   r)   r   r   r  r   r   r   r9   	variablesr   r    r4   r   rB  rg   r  r   rD   is_decoratedr  r   r   r%   r   defn)r   r   r   r   r  r   tvar_defr  r   ru   function_typefirstr   r7  r  r   	signaturerA  r_namer  decs                        rV   r  r    s    88D tzzjjJsxx$AHHMM  *#67':;../BBG>-*=!>	#f+y$FG4t!4	 #12ItWMN4<D&("b)yI #""H$HH",,-**+"	# Y	9k=YI'j	4uhj\23DDI!)T2DI"DM]]S(4/DN		DI tzz-dDJJ?!ZZ-

6
  dii nnx67;99dC(dD)CDJJtIINNt$rX   c                   | j                  d      syt        j                  dk\  rddl}n	 ddl}t        | d      5 }|j                  |      cddd       S # t
        $ r ddl}|j                  d       Y yw xY w# 1 sw Y   yxY w)zReturns 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)
r   sysversion_infotomllibtomliImportErrorwarningswarnopenload)r   toml_r  rfs       rV   r   r     s    
 (
7"	! 
k4	  Bzz"~   	MMfg		 s   A A;A87A8;B)rU   r   r   ztype[Plugin])r   r.   r   r=   )r  r   r   r+   r   r   r   r   )r   r   r   r1   r   r   r   r   )r   r1   r   r   r   r   )r   r   r   z8CheckerPluginInterface | SemanticAnalyzerPluginInterfacer   r   r   r   )r   r+   r   r   r   r   )NNF)r   z8SemanticAnalyzerPluginInterface | CheckerPluginInterfacer   r   r   r   r   r  r  r=   r   r  r  zTypeVarDef | Noner  r   r   r   )r   r   r   zdict[str, Any] | None)r   
__future__r   r  configparserr   typingr   r   r   mypy.errorcodesr   mypy.expandtyper	   r
   
mypy.nodesr   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'   r(   r)   mypy.optionsr*   mypy.pluginr+   r,   r-   r.   r/   r0   r1   mypy.pluginsr2   mypy.plugins.commonr3   mypy.semanalr4   mypy.server.triggerr5   
mypy.stater6   mypy.typeopsr7   
mypy.typesr8   r9   r:   r;   r<   r=   r>   r?   r@   rA   rB   mypy.typevarsrC   	mypy.utilrD   mypy.versionrE   mypy_versionpydantic._internalrF   pydantic.versionrG   rH   r  r   r   rh   r%  r"  rx   r}   r   r?  r>  r   r  rW   rT   r\   r   r   r  r   rG  r  r  r  r  r  r  r  r   r  rR  r   ry  re  r   r  r   r   rX   rV   <module>r     s   d " 
 % * * % @                 B !   % + 5  0    ( 2 4 & /5%
 !'. = 5 R (5 K   (5 0H<
. s V s l+C +C\#.XT XTv
 
2K
 K
\"& "&J n&DjQ	*,BJO(*DjQ24I:V *,F
S !13KZX ()9;^`jk \
O
O;;N;Y`;	;G
o
s ""& U%	AU%	U% U% 	U%
 U% U%  U% U% 
U%py$  5445s   "G G-,G-