
    4g.                       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.m/Z/ dd	l0m1Z1 dd
l2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8 ddl9m:Z: ddl;m<Z< ddl=m>Z> ddl?m@Z@ ddlAmBZB ddlCmDZDmEZEmFZFmGZGmHZHmIZImJZJmKZKmLZLmMZM ddlNmOZO ddlPmQZQ ddlRmSZT ddlUmVZV ddlWmXZX dZYdZZdZ[dZ\dZ]dZ^dZ_dZ`dZah dZb eXeT      Zcd Zdd!ZSdEd"Ze G d# d$e6      Zf G d% d&      ZgdFd'Zh G d( d)      Zi G d* d+      Zj G d, d-      Zk G d. d/      Zl ed0d1d2      Zm ed3d4d2      Zn ed5d6d2      Zo ed7d8d2      Zp ed9d:d2      Zq ed9d;d2      Zr ed9d<d2      ZsdGd=ZtdHd>ZudId?Zv	 	 	 	 	 	 	 	 dJd@ZwdIdAZxdKdBZy	 	 	 dL	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 dMdCZzdNdDZ{y)Oz^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-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_validatorrG   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    c/var/www/python.lazyprojects.co.uk/rdoDiscordBot/venv/lib/python3.12/site-packages/pydantic/mypy.pypluginrS   j   s
         c                  T     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
 xZS )rP   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     rR   r]   zPydanticPlugin.__init__|   s4    1': ..668!rT   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rW   fullnameBASEMODEL_FULLNAME.0bases     rR   	<genexpr>z5PydanticPlugin.get_base_class_hook.<locals>.<genexpr>   s     P<44==$66<   N)lookup_fully_qualified
isinstancenoder)   anymro$_pydantic_model_class_maker_callback)r^   rd   syms      rR   get_base_class_hookz"PydanticPlugin.get_base_class_hook   sG    ))(3:chh1P388<<PP@@@rT   c                .    |t         k(  r| j                  S y)z,Update Pydantic `ModelMetaclass` definition.N)MODEL_METACLASS_FULLNAME)_pydantic_model_metaclass_marker_callbackr^   rd   s     rR   get_metaclass_hookz!PydanticPlugin.get_metaclass_hook   s    //AAArT   c                2    |j                  d      rt        S y)z-Adjust return type of `from_orm` method call.z	.from_ormN)endswithfrom_attributes_callbackrv   s     rR   get_method_hookzPydanticPlugin.get_method_hook   s    [)++rT   c                    | j                   S )zjReturn all plugin config data.

        Used by mypy to determine if cache needs to be discarded.
        )r[   )r^   ctxs     rR   report_config_dataz!PydanticPlugin.report_config_data   s    
    rT   c                    t        |j                  |j                  |j                  | j                        }|j                          y rW   )PydanticModelTransformerclsreasonapirY   	transform)r^   r}   transformers      rR   rp   z3PydanticPlugin._pydantic_model_class_maker_callback   s0    .sww

CGGTM_M_`rT   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)rY   debug_dataclass_transformr   infodeclared_metaclassgetattrtyper   )r^   r}   info_metaclasss      rR   ru   z8PydanticPlugin._pydantic_model_metaclass_marker_callback   s^     7788NNN~>&&(BDI;?N8 JrT   r_   r+   returnNone)rd   strr   z(Callable[[ClassDefContext], None] | None)rd   r   r   z&Callable[[MethodContext], Type] | None)r}   r0   r   dict[str, Any])r}   r-   r   r   )__name__
__module____qualname____doc__r]   rr   rw   r{   r~   rp   ru   __classcell__)r`   s   @rR   rP   rP   y   s*    #"
! @rT   rP   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
)rX   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   boolr   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toolrF   Fz/Configuration value must be a boolean for key: )fallback)config_file
parse_tomlget	__slots__rl   r   
ValueErrorsetattrr   read
getbooleanCONFIGFILE_KEY)r^   r_   toml_configconfigkeysettingrY   s          rR   r]   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+ &rT   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   )r^   r   s     rR   rZ   zPydanticPluginConfig.to_data   s*    37>>B>CWT3''>BBBs   &Nr   r   r   )r   r   r   r   r   __annotations__r]   rZ    rT   rR   rX   rX      s3    I #''##,&CrT   rX   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 t!        d |j                   j"                  D              s| 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 )c              3  B   K   | ]  }|j                   t        k(    y wrW   rc   rf   s     rR   ri   z+from_attributes_callback.<locals>.<genexpr>   s     S?Rtt}} 22?Rrj   r   from_attributesT)r   rl   r=   itemr8   ret_typer9   r`   r   error_unexpected_behaviorr   contextdefault_return_typemetadatar   METADATA_KEYrn   ro   error_from_attributesname)r}   ctx_type
model_typedetailpydantic_metadatar   s         rR   rz   rz      s5    xxH(H%==(L)j9J9JH.U&&
	Hh	'
hZz(2D2D2M2M1NaP!&#''3;;?&&&"0044\B &&&Sz?R?RSS&&&'++Hb9==>OPOd"joo22CGGS[[I"""rT   c                      e Zd ZdZ	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 d
dZ	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 ddZ	 d	 	 	 	 	 	 	 ddZ	 d	 	 	 	 	 	 	 	 	 ddZddZe	dd       Z
ddZy	)PydanticModelFieldz5Based on mypy.plugins.dataclasses.DataclassAttribute.c                    || _         || _        || _        || _        || _        || _        || _        || _        |	| _        |
| _	        y rW   )
r   alias	is_frozenhas_dynamic_aliashas_defaultstrictlinecolumnr   r   )r^   r   r   r   r   r   r   r   r   r   r   s              rR   r]   zPydanticModelField.__init__   sL     	
"!2&			rT   c	                0   | j                  ||||      }	| j                  |n| j                  }
|s|
r| j                  ||      }nt        t        j
                        }t        |	|d|rt              S |s| j                  rt              S t              S )zABased on mypy.plugins.dataclasses.DataclassAttribute.to_argument.N)variabletype_annotationinitializerkind)to_varr   r	   r7   r<   explicitr   r   r   r   r   )r^   current_infotypedmodel_strictforce_optional	use_aliasr   force_typevars_invariantis_root_model_rootr   r   r   s               rR   to_argumentzPydanticModelField.to_argument  s     ;;|S)=UV!%!4$++F"..|SAO%i&8&89O+! 	
 	
 $2T5E5E-
 	
 LU
 	
rT   c                   |rFt        | j                  t              r,| j                  j                         }t        |_        || _        | j                  | j                  j                  t        j                  |j                  j                        5  t        |      }t        |t              sJ |r-|j                  D ]  }t        |t              st        |_          t        | j                  | j                  j                  j                   |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)rl   r   r>   copy_modifiedr   variancer   	self_typer5   strict_optional_setr_   strict_optionalrA   r9   argsr	   id)r^   r   r   r   modified_typefilled_with_typevarsargs          rR   r	   zPydanticModelField.expand_type/  s     $
 $))[1 $		 7 7 9)2&)	99 TYY%8%8%D
 **3;;+F+FG'4\'B$!"6AAA+388%c;7+4CL  9 #499tyy/B/B/E/EG[.\] HG yy H yys   ?D,AD,,E c                    |r| j                   | j                   }n| j                  }t        || j                  |||            S )z<Based on mypy.plugins.dataclasses.DataclassAttribute.to_var.)r   r   r*   r	   )r^   r   r   r   r   r   s         rR   r   zPydanticModelField.to_varM  s?     /::D99D4)),=UVWWrT   c           
        | j                   sJ | j                  | 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   	serializer^   s    rR   r   zPydanticModelField.serialize\  sd    yyyIIZZ!%!7!7++kkIIkkII'')

 
	
rT   c                n    |j                         }t        |j                  d      |      } | d||d|S )ABased on mypy.plugins.dataclasses.DataclassAttribute.deserialize.r   )r   r   r   )copyr2   pop)r   r   datar   typs        rR   deserializezPydanticModelField.deserializek  s9     yy{(&)93?/$/$//rT   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   r5   r   r_   r   r6   r   )r^   sub_typer   s      rR   expand_typevar_from_subtypez.PydanticModelField.expand_typevar_from_subtyper  sS     99 **3;;+F+FG3DIIxS	 HG !GGs   'A''A0N)r   r   r   z
str | Noner   r   r   r   r   r   r   bool | Noner   intr   r   r   Type | Noner   r)   )r   r)   r   r   r   r   r   r   r   r   r   r1   r   r   r   r   r   r   F)r   r)   r   r1   r   r   r   r   )
r   r)   r   r1   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]   r   r	   r   r   classmethodr   r   r   rT   rR   r   r      sU   ?  	
        0

 
 	

 
 
 -
 #'
 !
 

< ns$+Jfj	F */XX -X 	X
 #'X 
X
 0 0TrT   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 rW   r   )r^   r   s     rR   r]   zPydanticModelClassVar.__init__  s	    	rT   c                2    |j                         } | di |S )r   r   )r   )r   r   s     rR   r   z!PydanticModelClassVar.deserialize  s     yy{{T{rT   c                    d| j                   iS )r   r   r   r   s    rR   r   zPydanticModelClassVar.serialize  s     DII
 	
rT   N)r   r   r   r   r   )r   r   r   r   r]   r   r   r   r   rT   rR   r   r   {  s%      

rT   r   c                  v   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ed)d       Zed'd       Z	 d%	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 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frozenr   alias_generatorr   populate_by_namezset[str]tracked_config_fieldsc                <    || _         || _        || _        || _        y rW   )_cls_reason_apirY   )r^   r   r   r   rY   s        rR   r]   z!PydanticModelTransformer.__init__  s!     		*rT   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rW   )ROOT_MODEL_FULLNAMErd   rf   s     rR   ri   z5PydanticModelTransformer.transform.<locals>.<genexpr>  s     []T/4==@]s   NFc              3  B   K   | ]  }|j                   t        k(    y wrW   )rd   BASESETTINGS_FULLNAMErf   s     rR   ri   z5PydanticModelTransformer.transform.<locals>.<genexpr>  s     []T$--+@@]rj   T)r  )fields
class_varsr   )r	  r   rn   ro   collect_configcollect_fields_and_class_varsr   add_initializeradd_model_construct_method
set_frozenr  r  adjust_decorator_signaturesr   r   get_values_dictr   r   )	r^   r   is_root_modelr   r  r  fieldis_settings	class_vars	            rR   r   z"PydanticModelTransformer.transform  s]    yy~~[TXXVYWY][[$$&!??V
>Z/Ezz!  [TXXVYWY][[VV[-H'']S		&--42GH((* CII&uzz5??#44&IR\]R\Y9>>9+>+>+@@R\],,.'
l#  J]s   :"E*""E/c                &   | j                   j                  j                  j                         D ]  }t	        |j
                  t              s|j
                  j                  d   t	        t              sHt	        j                  t              sc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_namesrl   r%   value)rg   ir   	first_decs      rR   ri   zGPydanticModelTransformer.adjust_decorator_signatures.<locals>.<genexpr>  sO       *C3 &//2f<rCQXAYr^a^g^gkr^rr*Cs   AATN)r	  r   namesvaluesrl   rm   r   original_decoratorsr   calleer    rd   DECORATOR_FULLNAMESMODEL_VALIDATOR_FULLNAMErn   	enumerater   funcis_class)r^   rq   r&  s     @rR   r  z4PydanticModelTransformer.adjust_decorator_signatures  s     99>>''..0C#((I.HH88;	y(3"9#3#3X>!((115HH "((115MM  *3INN*C   .2CHHMM*# 1rT   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 kwargs   r   )%r	  ModelConfigDatakeywordsitemsget_config_updateupdatedefsbodyrl   r   r   lvaluesr    r   rvaluer   zipr#  r   r   r%   r$  r  failhas_alias_generatorr  rY   r   error_required_dynamic_aliasesr   ro   r   r   add_plugin_dependencyr4   rd   
setdefault)r^   r   r   has_config_kwargshas_config_from_namespacer   exprconfig_datastmtlhsarg_namer   key_expr
value_exprsubstmtr   r$  s                    rR   r  z'PydanticModelTransformer.collect_config  s   ii "!$)! ,,,,.JD$00t<K$(!k*	 / "&HHMMDd^X$>?$/ll1o!#x0CHH4Ndkk84),T[[-B-BDKKDTDT)U##+$d&<&<XsVZ&<&[\ *V  X6040A0A,*)(G<$d&<&<X^^Z&XY 1B
 D(+99(#yy~~G%g~> !//!,C%c84 MM$"8"87>>"RS  . !		o (,%M "P  9..//&&DD.tyy$?HHLL$D4==0 II++,A$--,PQ#}}\:8DJJLe!!$.  M % rT   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                  |||      }||j2                  d   }t#        |t4              sJ t#        |t              rS|r&|j6                  dk7  rt9        | j                  |       r|j;                  |j6                         |||j6                  <   t#        |t(              s|j;                  |j6                         |||j6                  <    t=        |j?                               t=        |j?                               fS )zACollects the fields for the model, accounting for parent classes.r4  r  r  z7BaseModel field may only be overridden by another fieldr  r   root) r	  reversedr   ro   r   r   r  rB  r4   rd   r7  r   r   r   r'  r   rm   rl   r*   r?  r   set%_get_assignment_statements_from_blockr:  $collect_field_or_class_var_from_stmtr<  r    r    error_extra_fields_on_root_modeladdlistr(  )r^   r1  r  r   found_fieldsfound_class_varsr   r   r   r  sym_nodecurrent_field_namescurrent_class_vars_namesrH  maybe_fieldrI  s                   rR   r  z6PydanticModelTransformer.collect_fields_and_class_vars<  sT    ii 79=?SXX\\!B/0D 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  L  #mmL9,GMMO
d)>)J)J4)P & P5 1< ),-0U >>sxxHDCCD,XhiK",,q/Cc8,,,+'9: SXX%74TYYE'++CHH5-8L*K)>?(,,SXX6-8 * I" L'')*D1A1H1H1J,KKKrT   c              #    K   |j                   D ](  }|j                  r| j                  |      E d {    * |j                  ;|j                  j                  s$| j                  |j                        E d {    y y y 7 N7 	wrW   )r;  is_unreachablerR  	else_body)r^   rH  r;  s      rR   ,_get_assignment_statements_from_if_statementzEPydanticModelTransformer._get_assignment_statements_from_if_statement  st     IID&&EEdKKK  >>%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rW   )r;  rl   r   r   r`  )r^   blockrH  s      rR   rR  z>PydanticModelTransformer._get_assignment_statements_from_block  sG     JJD$/
D&)LLTRRR	  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                  |      }	| j;                  |      }
|j4                  |j<                  rw|j>                  rk| j                  jA                  |j                  d      }|r||_        n:| j                  j+                  d	|       tC        tD        jF                        |_        | jI                  |      \  }}|r8|jJ                  s,| jL                  jN                  rtQ        | j                  |       | jS                  |      }| jU                  ||j                  |      }tW        |j                  ||	|
|||jX                  |jZ                  ||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   r1  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   r   ).r	  r<  rl   r    rD   is_valid_field_namer   
new_syntaxr=  r   r*  rd   r+  error_untyped_fieldsr  r   r'  r   rm   r"   r(   r?  r*   is_classvarr   r@   r   r9   get_has_default
get_strictrd  is_inferredanalyze_simple_literal_typer7   r<   
from_errorget_alias_infor  rY   r   rA  is_field_frozen_infer_dataclass_attr_init_typer   r   r   )r^   rH  r1  r  r   rI  rq   rm   	node_typer   r   r   r   r   r   	init_types                   rR   rS  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	!/#;;
 	
rT   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__rN   z(Unsupported signature for "__set__" in ""zUnsupported "__set__" in ")r   implicitr@   rl   r9   r   rm   r   get_containing_type_infor6   r7   r<   unannotatedr8   	arg_kindsr   r
   	arg_typesr  r?  r   )	r^   rq   r   r   defaulttsetter
super_infosetter_types	            rR   rp  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rT   c           
        d| j                   j                  j                  v r.| j                   j                  j                  d   j                  sy| j                  j
                  }t        |j                        }|j                  du}t        |j                  xr |j                         }| j                  |||||||d      }	|r7| j                  j                  t              j                  }
t        |
t               sJ d|
j                  v r|
j                  d   j                  }t        |t"              sJ ||j$                  |j$                  }t        |t&              sJ t)        |j*                        D ]  \  }}|"|j-                  d      s|j-                  d      s+| j                  j/                  |j0                  |         }t3        ||      }|	j5                  t7        ||dt8                      | j;                  ||      sCt3        d      }|	j5                  t7        |t=        t>        j@                        dtB                     tE        | j                  | j                   d|	tG                      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.
        r]   NT)r   r   requires_dynamic_aliasesr   r  r  r   ___kwargs)r   return_type)$r	  r   r'  plugin_generatedrY   r   r   r   r  r@  get_field_argumentsr  rk   r  rm   rl   r)   r   r   r8   r-  r#  
startswith	anal_typerz  r*   appendr   r   should_init_forbid_extrar7   r<   r   r   
add_methodr:   )r^   r  r   r  r  r   r   r   r  r   base_settings_nodebase_settings_init_node	func_typearg_idxrJ  analyzed_variable_typer   vars                     rR   r  z(PydanticModelTransformer.add_initializer5  s'    ---diinn6J6J:6V6g6g""--FMM*++47	#'(B(B(b6KbKbGb#c ''%%=#'%) ( 	
 !%!A!ABW!X!]!]0(;;;/555*<*B*B:*N*S*S'!"97CCC*6;R;W;W;c 7 < <I%i>>>-6y7J7J-K)#+x/B/B4/HPXPcPcdgPh$151D1DYEXEXY`Ea1b.#&x1G#HHX7MtU\$]^ .L ,,VV<h-CKKgi.@.@&A4ST499dii$HJWrT   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t        |j                        dd||      }ddd       | j                  ||      sCt        d      }	j!                  t        |	t#        t$        j&                        dt(                     |r|gz   n|gz   }t+        | j                   | j,                  d	|t/        | j,                  j0                        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_setNTF)r   r   r  r   r  r  r  model_construct)r   r  is_classmethod)r  
named_typeBUILTINS_NAMEr?   r:   r   r*   r   r5   r   r_   r   r  r   r   r  r  r7   r<   r   r   r  r	  rA   r   )
r^   r  r   r  r  set_stroptional_set_strfields_set_argumentr   r  s
             rR   r  z3PydanticModelTransformer.add_model_construct_methodd  sT    ))&&-'=		@T@TXeWffjUk@l?mn$gxz%:;&s=:J'KM]_celm&&tyy'8'8'H'HI++!&--0).'+ , D J ,,VV<h-CKKgi.@.@&A4ST/<t*++CVBWZ^B^IIII%diinn5	
! JIs   ),FFc                    | j                   j                  }|D ]H  }|j                  j                  |j                        }||j
                  }t        |t              r|xs |j                  |_	        ]t        |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                  <   K 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   rm   rl   r*   r   is_propertyr"   r  final_iterationdeferr   	TypeErrorreprr`   r   r   rd   	_fullnamer&   r   )
r^   r  r   r  r   r  rY  r  var_strr   s
             rR   r  z#PydanticModelTransformer.set_frozen  s)   
 yy~~Ezz~~ejj1H#mmc3'&,&?CO_5dii>W>WIIOO%,"%c(  /wiz#--PQRF-fdiiKll4l>"( $ 3chh >'6tS'A

388$/  % ,"&s),s   8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)r@  )builtins.Truebuiltins.Falser  r   )r  rl   r%   r$  r   r   error_invalid_config_valuer  r5  r    rd   )r^   r   r   r2  r  r@  s         rR   r8  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rT   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{  default_factoryr  )r=  rl   r'   r   r*  r#   rd   FIELD_FULLNAMEr>  r   r#  r`   r   r    rH  rF  r   r   s       rR   ri  z(PydanticModelTransformer.get_has_default  s     {{dH%dH%*T[['*Jt{{OcOcguOu
 !DNN;	T<49#4==<<,, *3 9 ]cllo>]^^ < dL111rT   c                l   | j                   }t        |t              rt        |j                  t              r~|j                  j
                  t        k(  rat        |j                  |j                        D ]>  \  }}|dk7  rt        |t              r"|j
                  dk(  r y|j
                  dk(  r y y y)zEReturns a the `strict` value of a field if defined, otherwise `None`.r   r  Tr  FN)r=  rl   r   r*  r#   rd   r  r>  r   r#  r    r  s       rR   rj  z#PydanticModelTransformer.get_strict  s     {{dH%*T[['*Jt{{OcOcguOu DNN;	T8#c8,||6#)99$ < rT   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)r=  rl   r'   r   r*  r#   rd   r  r-  r#  r   r%   r$  rH  rF  r%  rJ  r   s        rR   rn  z'PydanticModelTransformer.get_alias_info  s     {{dH% tX&:dkk7+KPTP[P[PdPdhvPv $T^^4KAx7"))A,C#w'yy%''! 5 rT   c                v   | j                   }t        |t              ryt        |t              r7t        |j                  t
              r|j                  j                  t        k(  syt        |j                        D ]<  \  }}|dk(  s|j                  |   }t        |t              xr |j                  dk(  c S  y)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=  rl   r'   r   r*  r#   rd   r  r-  r#  r   r    r  s        rR   ro  z(PydanticModelTransformer.is_field_frozen  s     {{dH% tX&:dkk7+KPTP[P[PdPdhvPv $T^^4KAx8#iil!#x0TS\\_5TT 5 rT   c	                    | j                   j                  }	|D 
cg c]H  }
|r|
j                  s8|
j                  |	|||xs ||| j                  ||xr |
j
                  dk(        J }}
|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.
        rO  )r   r   r   r   r   r   r   )r	  r   r   r   r  r   )r^   r  r   r   r   r  r  r  r   r   r  	argumentss               rR   r  z,PydanticModelTransformer.get_field_arguments)  s     yy~~  
  %"9"9 )7F;#II)A#0#IUZZ65I  	   	 
 
s   AA,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   r@  r  rY   r   )r^   r  r   s      rR   r  z1PydanticModelTransformer.should_init_forbid_extraI  sI     &&,,VT&:T:T5UV!!333rT   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  r@  r  s      rR   r  z1PydanticModelTransformer.is_dynamic_alias_presentV  s;    
 E&&  ;;&   rT   N)
r   r   r   zExpression | Statementr   r1   rY   rX   r   r   )r   r   )r   r   )r   r5  )r1  r5  r  r   r   zJtuple[list[PydanticModelField] | None, list[PydanticModelClassVar] | None])rH  r   r   Iterator[AssignmentStmt])rb  r   r   r  )rH  r   r1  r5  r  z dict[str, PydanticModelClassVar]r   z1PydanticModelField | PydanticModelClassVar | None)rq   r&   r   r   r   r   r   r   )
r  list[PydanticModelField]r   r5  r  r   r  r   r   r   )r  r  r   r1   r  r   r   r   r   )r   r   r   r   r2  r   r   ModelConfigData | None)rH  r   r   r   )rH  r   r   r   )rH  r   r   ztuple[str | None, bool])r  r  r   r   r   r   r   r   r  r   r  r   r  r   r   r   r   list[Argument])r  r  r   r5  r   r   )r  r  r@  r   r   r   )r   r   r   r   r  r   r]   r   r  r  r  r`  rR  rS  rp  r  r  r  r8  staticmethodri  rj  rn  ro  r  r  r  r   rT   rR   r   r     s   '8 ++ '+ -	+
 ,+ 
+"H26JXBL+BL<@BL	SBLHRS
"
2A
Oo
	:
B$L-X.-X8G-XVZ-Xko-X	-X^&
(&
  &
 	&

 &
 
&
PB>!F 2 2*    6  B */(  	
  #'   #' 
@4  rT   r   c                  T    e Zd ZdZ	 	 	 	 	 	 d	 	 	 	 	 	 	 	 	 	 	 ddZd	dZd
dZddZy)r5  z(Pydantic mypy plugin model config class.Nc                X    || _         || _        || _        || _        || _        || _        y rW   )r  r  r   r  r@  r   )r^   r  r  r   r  r@  r   s          rR   r]   zModelConfigData.__init__h  s2     ). 0#6 rT   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__r7  )r^   kvs      rR   r  zModelConfigData.get_values_dictx  s7    
 "&!4!4!6H!6A!-1!6HHHs   
22c                p    |y|j                         j                         D ]  \  }}t        | ||        y)z$Update Pydantic model config values.N)r  r7  r   )r^   r   r  r  s       rR   r9  zModelConfigData.update  s6    >**,224DAqD!Q 5rT   c                8    t        | |      t        | ||       yy)zFSet default value for Pydantic model config if config value is `None`.N)r   r   )r^   r   r$  s      rR   rC  zModelConfigData.setdefault  s     4%D#u% &rT   )NNNNNN)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  r9  rC  r   rT   rR   r5  r5  e  sh    2 %)"'+(,+/"!  %	
 & )  I &rT   r5  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`.ru  z$" does not have from_attributes=TruecodeN)r?  	ERROR_ORM)
model_namer   r   s      rR   r   r     s    HHq@A7QZH[rT   c                <    |j                  d|  d|t               y)z0Emits an error when the config value is invalid.zInvalid value for "Config.ru  r  N)r?  ERROR_CONFIG)r   r   r   s      rR   r  r    s    HH)$q17HNrT   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)r?  ERROR_ALIASr   r   s     rR   rA  rA    s    
 HH2G+HNrT   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)r?  ERROR_UNEXPECTED)r   r   r   linkfull_messages        rR   r   r     sA    
 DDLVHTVWL<TFB[\\LHH\7)9H:rT   c                4    | j                  d|t               y)z;Emits an error when there is an untyped field in the model.zUntyped fields disallowedr  N)r?  ERROR_UNTYPEDr  s     rR   rg  rg    s    HH('HFrT   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)r?  ERROR_EXTRA_FIELD_ROOT_MODELr  s     rR   rT  rT    s    HHA7QmHnrT   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  rl   rm   r   r:  r;  remover1   r  named_generic_typer=   rA   r   r*   r   r   r  r   r   r   r8   	variablesr   r!   r3   r   r/  rd   r  r   rB   is_decoratedr  r   r    r&   r   defn)r   r   r   r   r  r   tvar_defr  r   rq   function_typefirstrz  r#  ry  r   	signaturer.  r_namer  decs                        rR   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$rT   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)
ry   sysversion_infotomllibtomliImportErrorwarningswarnopenload)r   toml_r  rfs       rR   r   r     s|    
 (
7"	! 
k4	 Bzz"~ 
!	   	MMfg		 
!	 s   A A;A87A8;B)rQ   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TypeVarType | 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)   r*   mypy.optionsr+   mypy.pluginr,   r-   r.   r/   r0   r1   mypy.plugins.commonr2   mypy.semanalr3   mypy.server.triggerr4   
mypy.stater5   mypy.typeopsr6   
mypy.typesr7   r8   r9   r:   r;   r<   r=   r>   r?   r@   mypy.typevarsrA   	mypy.utilrB   mypy.versionrC   mypy_versionpydantic._internalrD   pydantic.versionrE   r   r   re   r  r  rt   r  DATACLASS_FULLNAMEr,  r+  MYPY_VERSION_TUPLEr  rS   rP   rX   rz   r   r   r   r5  r  r  r  r  r  ERROR_FIELD_DEFAULTSr  r   r  rA  r   rg  rT  r  r   r   rT   rR   <module>r     sf   d " 
 % * * % @! ! ! ! ! ! ! ! !D !  + 5  0   ( 2 4 & / '. = 5 R (5 K   (5  3@V 3@l+C +C\#4AT ATH
 
2N Nb$& $&N n&DjQ	*,BJO(*DjQ24I:V *,F
S !13KZX ()9;^`jk \
O
O;;N;Y`;	;G
o "#' U%	AU%	U% U% 	U%
 U% U% !U% U% 
U%prT   