a
    !fu                     @   s  d dl Z d dlmZ d dlmZmZmZmZmZm	Z	m
Z
mZmZ d dl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- d dl.m/Z/ d dl0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7 d dl8m9Z9 d d	l:m;Z; d d
l<m=Z= d dl>m?Z?m@Z@mAZAmBZBmCZCmDZDmZmEZEmFZFmGZGmHZHmIZI d dlJmKZK d dlLmMZM d dlNmOZP d dlQmRZR zd dl>mSZS W n  eTy   d dl>mGZS Y n0 dZUdZVdZWdZXdZYdZZdZ[e\e
e]df dddZ^e^ePZ_e_dkr dndZ`d ZOe\d!dd"d#ZaG d$d% d%e5ZbG d&d' d'Zce4ed(d)d*ZdG d+d, d,ZeG d-d. d.ZfG d/d0 d0Zged1d2d3Zhed4d5d3Zied6d7d3Zjed8d9d3Zked:d;d3Zled:d<d3Zme\e1edd=d>d?Zne\e7edd@dAdBZoe7eddCdDdEZpe\ee1e7f eddFdGdHZqe7eddCdIdJZre1eddCdKdLZsdYe2e\ee eee eeS etetddN	dOdPZueee(f e\dQdRdSZveee(f e\dQdTdUZwe\eee\ef  dVdWdXZxdS )Z    N)ConfigParser)	AnyCallableDictListOptionalSetTupleTypeUnion)	ErrorCode)	ARG_NAMEDARG_NAMED_OPTARG_OPTARG_POS	ARG_STAR2MDEFArgumentAssignmentStmtBlockCallExprClassDefContext	DecoratorEllipsisExprFuncBaseFuncDefJsonDict
MemberExprNameExprPassStmtPlaceholderNodeRefExprStrExpr
SymbolNodeSymbolTableNodeTempNodeTypeInfoTypeVarExprVar)Options)CheckerPluginInterfaceClassDefContextFunctionContextMethodContextPluginReportConfigContextSemanticAnalyzerPluginInterface)dataclasses)set_callable_name)make_wildcard_trigger)AnyTypeCallableTypeInstanceNoneType
Overloaded
ProperTyper
   	TypeOfAnyTypeTypeTypeVarType	UnionTypeget_proper_type)fill_typevars)get_unique_redefinition_name)__version__)is_valid_field)
TypeVarDef)r=   pydantic-mypyzpydantic-mypy-metadatazpydantic.main.BaseModelz"pydantic.env_settings.BaseSettingszpydantic.main.ModelMetaclasszpydantic.fields.Fieldzpydantic.dataclasses.dataclass.)versionreturnc                 C   s   t tt| dd dS )N+r   .)tuplemapint	partitionsplitrF    rP   N/var/www/html/python-backend/venv/lib/python3.9/site-packages/pydantic/mypy.pyparse_mypy_versionU   s    rR   )r   i  builtins__builtins__   zTypingType[Plugin]c                 C   s   t S )z
    `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).
    )PydanticPluginrO   rP   rP   rQ   plugin`   s    rW   c                       s   e Zd Zedd fddZeddddZeeee	gdf  dd	d
Z
eddddZeeeegef  dddZeeee	gdf  dddZeeeef dddZe	ddddZe	ddddZeddddZ  ZS )rV   NoptionsrG   c                    s&   t || _| j | _t | d S N)PydanticPluginConfigplugin_configto_data_plugin_datasuper__init__)selfrY   	__class__rP   rQ   r`   k   s    
zPydanticPlugin.__init__z+Optional[Callable[[ClassDefContext], None]])fullnamerG   c                 C   s:   |  |}|r6t|jtr6tdd |jjD r6| jS d S )Nc                 s   s   | ]}t |tkV  qd S rZ   )get_fullnameBASEMODEL_FULLNAME.0baserP   rP   rQ   	<genexpr>t       z5PydanticPlugin.get_base_class_hook.<locals>.<genexpr>)lookup_fully_qualified
isinstancenoder'   anymro$_pydantic_model_class_maker_callbackra   rd   symrP   rP   rQ   get_base_class_hookp   s
    
z"PydanticPlugin.get_base_class_hookc                 C   s   |t kr| jS d S rZ   )MODEL_METACLASS_FULLNAME)_pydantic_model_metaclass_marker_callbackra   rd   rP   rP   rQ   get_metaclass_hookx   s    z!PydanticPlugin.get_metaclass_hookz+Optional[Callable[[FunctionContext], Type]]c                 C   s"   |  |}|r|jtkr| jS d S rZ   )rl   rd   FIELD_FULLNAME_pydantic_field_callbackrr   rP   rP   rQ   get_function_hook}   s    
z PydanticPlugin.get_function_hookc                 C   s   | drtS d S )Nz	.from_orm)endswithfrom_orm_callbackrw   rP   rP   rQ   get_method_hook   s    
zPydanticPlugin.get_method_hookc                 C   s   |t krtdk rtjS dS )zMark pydantic.dataclasses as dataclass.

        Mypy version 1.1.1 added support for `@dataclass_transform` decorator.
        )   r   N)DATACLASS_FULLNAMEMYPY_VERSION_TUPLEr2   Zdataclass_class_maker_callbackrw   rP   rP   rQ   get_class_decorator_hook   s    z'PydanticPlugin.get_class_decorator_hookctxrG   c                 C   s   | j S )zjReturn all plugin config data.

        Used by mypy to determine if cache needs to be discarded.
        )r^   )ra   r   rP   rP   rQ   report_config_data   s    z!PydanticPlugin.report_config_datac                 C   s   t || j}|  d S rZ   )PydanticModelTransformerr\   	transform)ra   r   ZtransformerrP   rP   rQ   rq      s    z3PydanticPlugin._pydantic_model_class_maker_callbackc                 C   s<   | j jrdS |jjj}|s"J dt|jddr8d|j_dS )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_transformclsinfoZdeclared_metaclassgetattrtyper   )ra   r   Zinfo_metaclassrP   rP   rQ   rv      s    
z8PydanticPlugin._pydantic_model_metaclass_marker_callbackr
   c           	         s   |j  |jd dksJ d|jd dks2J d|jd }|jd }|r`|r`t|j|j  S |r|jd d }|d }t|ts|S n|r|jd d }t|t	rt
dkr|jd }n| d }t|tr|j}t|dd	}|rtd
d |D rt fdd|D |_|S  S )ax  
        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())r   i  argsNc                 s   s   | ]}t |tV  qd S rZ   )rm   r=   )rh   argrP   rP   rQ   rj      rk   z:PydanticPlugin._pydantic_field_callback.<locals>.<genexpr>c                 3   s   | ]
} V  qd S rZ   rP   )rh   _Zdefault_any_typerP   rQ   rj      rk   )default_return_typeZcallee_arg_namesr   +error_default_and_default_factory_specifiedapicontext	arg_typesrm   r   r9   r   itemsr6   ret_typer   allrJ   )	ra   r   Zdefault_argsZdefault_factory_argsZdefault_typeZdefault_argZdefault_factory_typer   r   rP   r   rQ   rz      s6    	



z'PydanticPlugin._pydantic_field_callback)__name__
__module____qualname__r*   r`   strrt   r   r   r,   rx   r{   r.   r
   r~   r   r0   r   r   r   rq   rv   r-   rz   __classcell__rP   rP   rb   rQ   rV   j   s   	rV   c                   @   s`   e Zd ZU dZeed< eed< eed< eed< eed< eddd	d
Zee	e
f dddZdS )r[   )init_forbid_extra
init_typedwarn_required_dynamic_aliaseswarn_untyped_fieldsr   r   r   r   r   r   NrX   c                 C   s   |j d u rd S t|j }|d urr|di di }| jD ]4}||d}t|tsbtd| t| || q:n:t }|	|j  | jD ] }|j
t|dd}t| || qd S )NZtoolrE   Fz/Configuration value must be a boolean for key: )fallback)config_file
parse_tomlget	__slots__rm   bool
ValueErrorsetattrr   read
getbooleanCONFIGFILE_KEY)ra   rY   Ztoml_configconfigkeyZsettingr\   rP   rP   rQ   r`      s    




zPydanticPluginConfig.__init__rG   c                    s    fdd j D S )Nc                    s   i | ]}|t  |qS rP   )r   )rh   r   ra   rP   rQ   
<dictcomp>  rk   z0PydanticPluginConfig.to_data.<locals>.<dictcomp>)r   r   rP   r   rQ   r]      s    zPydanticPluginConfig.to_data)r   r   r   r   r   __annotations__r*   r`   r   r   r   r]   rP   rP   rP   rQ   r[      s   
r[   r   c                 C   s   | j }t|tr|j}t|tr4t|jtr4|j}n<t|trD|}n,d| d|jj d}t	|| j
| j | jS |j jt}|du r| jS |di d}|durtt|j | j
| j | jS )z3
    Raise an error if orm_mode is not enabled
    z
ctx.type: 
 (of type )Nr   orm_modeT)r   rm   r<   itemr6   r   r7   rc   r   error_unexpected_behaviorr   r   r   metadatar   METADATA_KEYerror_from_ormget_name)r   Zctx_typeZ
model_typedetailZpydantic_metadatar   rP   rP   rQ   r}     s"    

r}   c                   @   sf  e Zd ZU h dZee ed< eeddddZ	dddd	Z
ddd
dZddddZded dddZed deddddZed ddddZed eddddZeed dddZeeeeedd d!Zeee ed"d#d$Zeeeee ef d%d&d'Zed eeeee d(d)d*Zed ded+d,d-Z eed eed.d/d0Z!dS )1r   >   frozenalias_generatorextraallow_population_by_field_namer   allow_mutationtracked_config_fieldsN)r   r\   rG   c                 C   s   || _ || _d S rZ   )_ctxr\   )ra   r   r\   rP   rP   rQ   r`   '  s    z!PydanticModelTransformer.__init__r   c                 C   s   | j }| j jj}|   |  }| |}|D ]&}||j jdu r.|jj	s.|j
  q.tdd |jdd D }| ||| | | | j||jdu p|jdu d dd	 |D | d
|jt< dS )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 allow_mutation = False or frozen = True
        * stores the fields, config, and if the class is settings in the mypy metadata for access by subclasses
        Nc                 s   s   | ]}t |tkV  qd S rZ   )re   BASESETTINGS_FULLNAMErg   rP   rP   rQ   rj   ?  rk   z5PydanticModelTransformer.transform.<locals>.<genexpr>FT)r   c                 S   s   i | ]}|j | qS rP   )name	serializerh   fieldrP   rP   rQ   r   D  rk   z6PydanticModelTransformer.transform.<locals>.<dictcomp>)fieldsr   )r   r   r   adjust_validator_signaturescollect_configcollect_fieldsr   r   r   final_iterationdeferro   rp   add_initializeradd_construct_method
set_frozenr   r   set_values_dictr   r   )ra   r   r   r   r   r   is_settingsrP   rP   rQ   r   +  s     



z"PydanticModelTransformer.transformc                 C   sb   | j jjj D ]L\}}t|jtr|jjd }t|t	rt|j
tr|j
jdkrd|jj_qdS )a[  When we decorate a function `f` with `pydantic.validator(...), 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()` call as a classmethod.
        r   z#pydantic.class_validators.validatorTN)r   r   r   namesr   rm   rn   r   Zoriginal_decoratorsr   calleer   rd   funcis_class)ra   r   rs   Z	first_decrP   rP   rQ   r   H  s    

z4PydanticModelTransformer.adjust_validator_signaturesModelConfigDatac           	      C   s   | j }|j}t }|jjD ]d}t|ts*q|jdkr|jjD ] }t|tsLq<|	| 
| q<|jr|js| jjrt|j| q|jjdd D ]L}t|jvrq|jtt| |jt d  D ]\}}||| qq|S )zz
        Collects the values of the config attributes that are used by the plugin, accounting for parent classes.
        ZConfigr   Nr   )r   r   r   defsbodyrm   r   r   r   updateget_config_updatehas_alias_generatorr   r\   r   error_required_dynamic_aliasesr   r   rp   r   r   add_plugin_dependencyr4   re   r   
setdefault)	ra   r   r   r   stmtsubstmtr   r   valuerP   rP   rQ   r   Z  s2    



z'PydanticModelTransformer.collect_configPydanticModelField)model_configrG   c                    s  | j }| j j}g }t }|jjD ]}t|ts0q |jd }t|tr t	|j
sPq |jsj| jjrjt|j| |jj|j
}|du rq |j}	t|	trq t|	tsq |	jrq | |||}
| |\}}|r|js| jjrt|j| |t|j
|
|||j|jd | |j
 q |! }|jj"dd D ]}t#|j$vr>q*g }|j%t&t'| |j$t# d ( D ]^\ } |vrt)||}|   || n( fdd|D \}|*| || qh|| }q*|S )zR
        Collects the fields for the model, accounting for parent classes
        r   Nr   is_requiredaliashas_dynamic_aliaslinecolumnr   r   c                 3   s   | ]}|j  kr|V  qd S rZ   r   )rh   ar   rP   rQ   rj     rk   z:PydanticModelTransformer.collect_fields.<locals>.<genexpr>)+r   r   setr   r   rm   r   lvaluesr   rC   r   Z
new_syntaxr\   r   error_untyped_fieldsr   r   r   r   rn   r!   r)   Zis_classvarget_is_requiredget_alias_infor   r   r   appendr   r   r   addcopyrp   r   r   r   r4   re   r   deserializeremove)ra   r   r   r   r   Zknown_fieldsr   lhsrs   rn   r   r   r   Z
all_fieldsr   Zsuperclass_fieldsdatar   rP   r   rQ   r   y  st    







z'PydanticModelTransformer.collect_fields)r   r   r   rG   c           
      C   s   | j }| jj}|jdu}|p,t|jo*|j }| j||||d}| ||sntd}	|	t
|	ttjdt d|jjjvrt|d|t  dS )z
        Adds a fields-aware `__init__` method to the class.

        The added `__init__` will be annotated with types vs. all `Any` depending on the plugin settings.
        Ttypedforce_all_optional	use_aliaskwargsNr`   )r   r\   r   r   r   r   get_field_argumentsshould_init_forbid_extrar)   r   r   r5   r;   explicitr   r   r   r   
add_methodr8   )
ra   r   r   r   r   r   r  r   Zinit_argumentsvarrP   rP   rQ   r     s    
z(PydanticModelTransformer.add_initializer)r   rG   c              	   C   s   | j }|jt d|jt dg}t|t g}ttd||dt}| j	|dddd}|g| }|jt d}d	}|j
jd
 | }	t||	dg |}
t||	g |}tt||j
jj|< t|
tr|
}nt|
}t|d||||
dd dS )z
        Adds a fully typed `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.strZ_fields_setNTFr   z.objectZ_PydanticBaseModelrI   r   	construct)return_type	self_typetvar_defis_classmethod)r   r   
named_typeBUILTINS_NAMEr>   r8   r   r)   r   r  r   rd   rD   r(   r%   r   r   r   rm   r=   r  )ra   r   r   Zset_strZoptional_set_strZfields_set_argumentZconstruct_argumentsobj_typeZself_tvar_nameZtvar_fullnameZtvdZself_tvar_exprr
  rP   rP   rQ   r     s0    $

z-PydanticModelTransformer.add_construct_method)r   r   rG   c           
   	   C   s   | j }|jj}|D ]}|j|j}|dur|j}t|trD||_	qt|t
rb|jjsb|j  qzt|}W n ty   t|}Y n0 d| d|j d}	t|	|j|j q|j|dd}||_||_	t|d t| |_tt||jt|< qdS )z
        Marks all fields as properties so that attempts to set them trigger mypy errors.

        This is the same approach used by the attrs and dataclasses plugins.
        Nzsym_node.node: r   r   F)r  rI   )r   r   r   r   r   r   rn   rm   r)   Zis_propertyr!   r   r   r   r   	TypeErrorreprrc   r   to_varre   r   	_fullnamer%   r   )
ra   r   r   r   r   r   Zsym_noder  Zvar_strr   rP   rP   rQ   r     s*    
z#PydanticModelTransformer.set_frozen)r   rG   c                 C   s   |j d }t|tr |j| jv s$dS |jdkrt|jtrH|jjdk}n0t|jtrb|jjdk}nt	|j| j
j| dS t|dS |jdkrd}t|jtr|jjdkrd	}t|d
S t|jtr|jjdv rtf i |j|jjdkiS t	|j| j
j| dS )z
        Determines the config update due to a single statement in the Config class definition.

        Warns if a tracked config attribute is set to a value the plugin doesn't know how to interpret (e.g., an int)
        r   Nr   Zforbid)forbid_extrar   Tzbuiltins.NoneF)r   )builtins.Truezbuiltins.Falser  )r   rm   r   r   r   rvaluer#   r   r   error_invalid_config_valuer   r   r   rd   )ra   r   r   r  r   rP   rP   rQ   r   )  s(    




z*PydanticModelTransformer.get_config_update)r   r   r   rG   c                 C   s   |j }t|tr.t| j|j j}t| S t|t	rt|j
tr|j
jtkrt|j|jD ]4\}}|du sv|dkr|jtu   S |dkr^ dS q^t| j|j j}t| S t|tS )zg
        Returns a boolean indicating whether the field defined in `stmt` is a required field.
        Nr   r   F)r  rm   r&   r?   r   r   r   r   type_has_implicit_defaultr   r   r"   rd   ry   zipr   	arg_namesrc   r   )r   r   r   exprZ
value_typer   r   rP   rP   rQ   r   E  s    
"z(PydanticModelTransformer.get_is_required)type_rG   c                 C   s4   t | trdS t | tr0tdd | jD r0dS dS )z
        Returns True if the passed type will be given an implicit default value.

        In pydantic v1, this is the case for Optional types and Any (with default value None).
        Tc                 s   s"   | ]}t |tpt |tV  qd S rZ   )rm   r8   r5   )rh   r   rP   rP   rQ   rj   i  s   zEPydanticModelTransformer.type_has_implicit_default.<locals>.<genexpr>F)rm   r5   r>   ro   r   )r  rP   rP   rQ   r  _  s    
z2PydanticModelTransformer.type_has_implicit_default)r   rG   c                 C   s   | j }t|trdS t|tr6t|jtr6|jjtks:dS t|j	D ]:\}}|dkrVqD|j
| }t|trx|jdf  S  dS qDdS )a   
        Returns a pair (alias, has_dynamic_alias), extracted from the declaration of the field defined in `stmt`.

        `has_dynamic_alias` is True if and only if an alias is provided, but not as a string literal.
        If `has_dynamic_alias` is True, `alias` will be None.
        )NFr   F)NT)r  rm   r&   r   r   r"   rd   ry   	enumerater  r   r#   r   )r   r  iZarg_namer   rP   rP   rQ   r   p  s$    




z'PydanticModelTransformer.get_alias_info)r   r   r   r  rG   c                    s&   | j jj fdd|D }|S )z
        Helper function used during the construction of the `__init__` and `construct` method signatures.

        Returns a list of mypy Argument instances for use in the generated signatures.
        c                    s(   g | ] }r|j s|j d qS ))r   force_optionalr  )r   to_argumentr   r   r   r   r  rP   rQ   
<listcomp>  s   
z@PydanticModelTransformer.get_field_arguments.<locals>.<listcomp>)r   r   r   )ra   r   r   r   r  	argumentsrP   r!  rQ   r    s
    
z,PydanticModelTransformer.get_field_arguments)r   r   rG   c                 C   s.   |j s| |t|jrdS |jr&dS | jjS )aH  
        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  r\   r   )ra   r   r   rP   rP   rQ   r    s    z1PydanticModelTransformer.should_init_forbid_extra)r   r   rG   c                 C   s8   | D ]}|j r dS q|r4| D ]}|jdu r dS qdS )z
        Returns whether any fields on the model have a "dynamic alias", i.e., an alias that cannot be
        determined during static analysis.
        TNF)r   r   )r   r   r   rP   rP   rQ   r$    s    
z1PydanticModelTransformer.is_dynamic_alias_present)"r   r   r   r   r   r   r   r,   r[   r`   r   r   r   r   r   r   r   r   r   r   r   r   staticmethodr   r   r   r:   r  r	   r   r   r  r  r$  rP   rP   rP   rQ   r     s,   
	T%!r   c                   @   st   e Zd Zeeee eeedddZeee	dddZ
eeeeeddd	Zed
ddZeeed dddZdS )r   r   c                 C   s(   || _ || _|| _|| _|| _|| _d S rZ   r   )ra   r   r   r   r   r   r   rP   rP   rQ   r`     s    zPydanticModelField.__init__)r   r  rG   c                 C   s,   | j }|r| jd ur| j}t||| j  jS rZ   )r   r   r)   r   )ra   r   r  r   rP   rP   rQ   r    s    zPydanticModelField.to_var)r   r   r  r  rG   c                 C   sR   |r"|| j  jd ur"|| j  j}n
ttj}t| |||d |sF| jsJtnt	dS )N)variabletype_annotationZinitializerkind)
r   r   r5   r;   r  r   r  r   r   r   )ra   r   r   r  r  r'  rP   rP   rQ   r     s    

zPydanticModelField.to_argumentr   c                 C   s   | j S rZ   )__dict__r   rP   rP   rQ   r     s    zPydanticModelField.serialize)r   r   rG   c                 C   s   | f i |S rZ   rP   )r   r   r   rP   rP   rQ   r     s    zPydanticModelField.deserializeN)r   r   r   r   r   r   rL   r`   r'   r)   r  r   r   r   r   classmethodr   rP   rP   rP   rQ   r     s   
r   c                   @   sz   e Zd Zdee ee ee ee ee ee dddZeeef dddZ	ed  ddd	d
Z
eeddddZdS )r   Nr  r   r   r   r   r   c                 C   s(   || _ || _|| _|| _|| _|| _d S rZ   r+  )ra   r  r   r   r   r   r   rP   rP   rQ   r`     s    	zModelConfigData.__init__r   c                 C   s   dd | j  D S )Nc                 S   s   i | ]\}}|d ur||qS rZ   rP   )rh   kvrP   rP   rQ   r     rk   z3ModelConfigData.set_values_dict.<locals>.<dictcomp>)r)  r   r   rP   rP   rQ   r     s    zModelConfigData.set_values_dict)r   rG   c                 C   s2   |d u rd S |   D ]\}}t| || qd S rZ   )r   r   r   )ra   r   r,  r-  rP   rP   rQ   r     s    zModelConfigData.update)r   r   rG   c                 C   s   t | |d u rt| || d S rZ   )r   r   )ra   r   r   rP   rP   rQ   r     s    zModelConfigData.setdefault)NNNNNN)r   r   r   r   r   r`   r   r   r   r   r   r   rP   rP   rP   rQ   r     s"         r   zpydantic-ormzInvalid from_orm callZPydanticzpydantic-configzInvalid config valuezpydantic-aliaszDynamic alias disallowedzpydantic-unexpectedzUnexpected behaviorzpydantic-fieldzUntyped field disallowedzInvalid Field defaults)
model_namer   r   rG   c                 C   s   |j d|  d|td d S )N"z" does not have orm_mode=Truecode)fail	ERROR_ORM)r.  r   r   rP   rP   rQ   r     s    r   )r   r   r   rG   c                 C   s   |j d|  d|td d S )NzInvalid value for "Config.r/  r0  )r2  ERROR_CONFIG)r   r   r   rP   rP   rQ   r    s    r  )r   r   rG   c                 C   s   | j d|td d S )Nz#Required dynamic aliases disallowedr0  )r2  ERROR_ALIASr   r   rP   rP   rQ   r     s    r   )r   r   r   rG   c                 C   s4   d}d|  d}|d| d7 }|j ||td d S )Nz6https://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!r0  )r2  ERROR_UNEXPECTED)r   r   r   linkZfull_messagerP   rP   rQ   r     s    r   c                 C   s   | j d|td d S )NzUntyped fields disallowedr0  )r2  ERROR_UNTYPEDr6  rP   rP   rQ   r     s    r   c                 C   s   | j d|td d S )Nz>Field default and default_factory cannot be specified togetherr0  )r2  ERROR_FIELD_DEFAULTSr6  rP   rP   rQ   r   !  s    r   F)	r   r   r   r	  r
  r  r  is_newrG   c                 C   s  | j j}||jv r@|j| }	|	jr@t|	jtr@| j jj	|	j |pJt
|}|sT|rpttdt|dtg}
n |pzt
|}ttd|dtg}
|
| }g g g   }}}|D ]:}|jsJ d||j |t|j ||j q| jt d}t|||||}|r|g|_t||tt g}||_t|||_||_t|d | |_ |j!|_!||jv rt"||j}|j| |j|< |rd|_#t||j}||_|j |_ d|_$t%|t&dg|}|j!|_!t't(|}	n
t't(|}	d|	_|	|j|< |j)jj| dS )	z}
    Adds a new method to a class.

    This can be dropped if/when https://github.com/python/mypy/issues/7301 is merged
    _clsNZ__pydantic_self__z"All arguments must be fully typed.z	.functionrI   Tr*  )*r   r   r   Zplugin_generatedrm   rn   r   r   r   r   r@   r   r)   r<   Zmake_normalizedr   r'  r   r   r&  r(  r   r  r  r6   	variablesr   r    r3   r   r   re   r  r   rA   Zis_decoratedr  r   r   r%   r   Zdefn)r   r   r   r	  r
  r  r  r<  r   rs   firstr   r  Z	arg_kindsr   Zfunction_type	signaturer   Zr_namer-  decrP   rP   rQ   r  %  sV    



r  )xrG   c                 C   s   | j }t|r| S |S zc
    Used for compatibility with mypy 0.740; can be dropped once support for 0.740 is dropped.
    )rd   callablerB  fnrP   rP   rQ   re   x  s    re   c                 C   s   | j }t|r| S |S rC  )r   rD  rE  rP   rP   rQ   r     s    r   )r   rG   c                 C   s   |  dsd S d}tjdkr&dd l}nZzdd l}W nL ty~   d}zdd l}W n( tyx   dd l}|d Y Y d S 0 Y n0 t	| |}|
|W  d    S 1 s0    Y  d S )Nz.tomlrb)      r   rzJNo TOML parser installed, cannot read configuration from `pyproject.toml`.)r|   sysversion_infoZtomllibtomliImportErrortomlwarningswarnopenload)r   Z	read_modeZtoml_rP  rfrP   rP   rQ   r     s"    



r   )NNFF)yrK  configparserr   typingr   r   r   r   r   r   r	   r
   Z
TypingTyper   Zmypy.errorcodesr   Z
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)   Zmypy.optionsr*   Zmypy.pluginr+   r,   r-   r.   r/   r0   r1   Zmypy.pluginsr2   Zmypy.semanalr3   Zmypy.server.triggerr4   Z
mypy.typesr5   r6   r7   r8   r9   r:   r;   r<   r=   r>   r?   Zmypy.typevarsr@   Z	mypy.utilrA   Zmypy.versionrB   Zmypy_versionZpydantic.utilsrC   rD   rN  r   r   rf   r   ru   ry   r   r   rL   rR   r   r  rW   rV   r[   r}   r   r   r   r3  r4  r5  r8  r:  r;  r   r  r   r   r   r   r   r  re   r   r   rP   rP   rP   rQ   <module>   s   ,|$	8
u%   !%
	    S

