o
    7ήc                     @   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l0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6 d dl7m8Z8 d d	l9m:Z: d d
l;m<Z< d dl=m>Z>m?Z?m@Z@mAZAmBZBmZmCZCmDZDmEZEmFZFmGZG d dlHmIZI d dlJmKZK d dlLmMZN d dlOmPZP zd dl=mQZQ W n eRy   d dl=mEZQ Y nw dZSdZTdZUdZVdZWdZXdeYde
eZdf fddZ[e[eNZ\e\dkrdndZ]deYdd fd!d"Z^G d#d$ d$e5Z_G d%d& d&Z`d'e4defd(d)ZaG d*d+ d+ZbG d,d- d-ZcG d.d/ d/Zded0d1d2Zeed3d4d2Zfed5d6d2Zged7d8d2Zhed9d:d2Zied9d;d2Zjd<eYd=e1d>eddfd?d@ZkdAeYd=e6d>eddfdBdCZld=e6d>eddfdDdEZmdFeYd=e1d>eddfdGdHZnd=e6d>eddfdIdJZod=e1d>eddfdKdLZp			M	Md^d'e2dAeYdNee dOedPee dQeeQ dReqdSeqddfdTdUZrdVeee(f deYfdWdXZsdVeee(f deYfdYdZZtd[eYdeeeYef  fd\d]ZudS )_    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SemanticAnalyzerPluginInterface)dataclasses)set_callable_name)make_wildcard_trigger)AnyTypeCallableTypeInstanceNoneType
Overloadedr
   	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.fields.Fieldzpydantic.dataclasses.dataclassversionreturn.c                 C   s$   t dd | ddd dD S )Nc                 s   s    | ]}t |V  qd S N)int).0part rJ   4/tmp/pip-target-vg8gfxp4/lib/python/pydantic/mypy.py	<genexpr>S   s    z%parse_mypy_version.<locals>.<genexpr>+   r   .)tuplesplitrD   rJ   rJ   rK   parse_mypy_versionR   s   $rS   )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).
    )PydanticPluginrR   rJ   rJ   rK   pluginZ   s   rW   c                       s   e Zd Zdeddf fddZdeddfdd	Zdedd
fddZdedee	e
gef  fddZdedee	egdf  fddZdeddfddZdeddfddZ  ZS )rV   optionsrE   Nc                    s   t || _t | d S rF   )PydanticPluginConfigplugin_configsuper__init__)selfrX   	__class__rJ   rK   r\   e   s   
zPydanticPlugin.__init__fullnamez+Optional[Callable[[ClassDefContext], None]]c                 C   s:   |  |}|rt|jtrtdd |jjD r| jS d S )Nc                 s       | ]	}t |tkV  qd S rF   )get_fullnameBASEMODEL_FULLNAMErH   baserJ   rJ   rK   rL   m       z5PydanticPlugin.get_base_class_hook.<locals>.<genexpr>)lookup_fully_qualified
isinstancenoder'   anymro$_pydantic_model_class_maker_callbackr]   r`   symrJ   rJ   rK   get_base_class_hooki   s
   
z"PydanticPlugin.get_base_class_hookz+Optional[Callable[[FunctionContext], Type]]c                 C   s"   |  |}|r|jtkr| jS d S rF   )rg   r`   FIELD_FULLNAME_pydantic_field_callbackrm   rJ   rJ   rK   get_function_hookq   s   
z PydanticPlugin.get_function_hookc                 C   s   | drtS d S )Nz	.from_orm)endswithfrom_orm_callbackr]   r`   rJ   rJ   rK   get_method_hookw   s   
zPydanticPlugin.get_method_hookc                 C   s   |t krtjS d S rF   )DATACLASS_FULLNAMEr1   dataclass_class_maker_callbackru   rJ   rJ   rK   get_class_decorator_hook|   s   z'PydanticPlugin.get_class_decorator_hookctxc                 C   s   t || j}|  d S rF   )PydanticModelTransformerrZ   	transform)r]   rz   transformerrJ   rJ   rK   rl      s   z3PydanticPlugin._pydantic_model_class_maker_callbackr
   c           	         s  |j  |jd dksJ d|jd dksJ d|jd }|jd }|r0|r0t|j|j  S |rF|jd d }|d }t|tsD|S  S |r|jd d }t|t	rdt
dkr^|jd }n| d }t|tr|j}t|dd	}|rtd
d |D rt 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()rN   default_factoryz9"default_factory" is no longer second argument in Field())r   i  argsNc                 s       | ]}t |tV  qd S rF   )rh   r;   )rH   argrJ   rJ   rK   rL          z:PydanticPlugin._pydantic_field_callback.<locals>.<genexpr>c                 3   s    | ]} V  qd S rF   rJ   )rH   _default_any_typerJ   rK   rL      s    )default_return_typecallee_arg_namesr   +error_default_and_default_factory_specifiedapicontext	arg_typesrh   r   r8   MYPY_VERSION_TUPLEitemsr5   ret_typegetattrallrP   )	r]   rz   default_argsdefault_factory_argsdefault_typedefault_argdefault_factory_typer   r   rJ   r   rK   rq      s8   	




z'PydanticPlugin._pydantic_field_callback)__name__
__module____qualname__r*   r\   strro   rr   r   r   r.   r
   rv   r,   ry   rl   r-   rq   __classcell__rJ   rJ   r^   rK   rV   d   s      rV   c                   @   sD   e Zd ZU dZeed< eed< eed< eed< deddfd	d
ZdS )rY   )init_forbid_extra
init_typedwarn_required_dynamic_aliaseswarn_untyped_fieldsr   r   r   r   rX   rE   Nc                 C   s   |j d u rd S t|j }|d ur:|di di }| jD ]}||d}t|ts1td| t| || qd S t }|	|j  | jD ]}|j
t|dd}t| || qFd S )NtoolrC   Fz/Configuration value must be a boolean for key: )fallback)config_file
parse_tomlget	__slots__rh   bool
ValueErrorsetattrr   read
getbooleanCONFIGFILE_KEY)r]   rX   toml_configconfigkeysettingrZ   rJ   rJ   rK   r\      s"   




zPydanticPluginConfig.__init__)r   r   r   r   r   __annotations__r*   r\   rJ   rJ   rJ   rK   rY      s   
 rY   rz   c                 C   s   t | jtrt | jjtr| jj}n"t | jtr| j}nd| j d| jjj d}t|| j| j	 | j
S |jjt}|du rB| j
S |di d}|durZtt|j| j| j	 | j
S )z3
    Raise an error if orm_mode is not enabled
    z
ctx.type: z
 (of type )Nr   orm_modeT)rh   typer5   r   r6   r_   r   error_unexpected_behaviorr   r   r   metadatar   METADATA_KEYerror_from_ormget_name)rz   
model_typedetailpydantic_metadatar   rJ   rJ   rK   rt      s   
rt   c                   @   sn  e Zd ZU h dZee ed< dededdfddZ	d2d	d
Z
d2ddZd3ddZddded fddZded dddeddfddZded ddfddZded deddfddZdeded fdd Zed!ed"ed#edefd$d%Zed"edeee ef fd&d'Zded d(ed)ed*edee f
d+d,Zded dddefd-d.Zeded d/edefd0d1ZdS )4r{   >   extrafrozenr   allow_mutationalias_generatorallow_population_by_field_nametracked_config_fieldsrz   rZ   rE   Nc                 C   s   || _ || _d S rF   )_ctxrZ   )r]   rz   rZ   rJ   rJ   rK   r\      s   
z!PydanticModelTransformer.__init__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 pQ|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   ra   rF   )rb   BASESETTINGS_FULLNAMErd   rJ   rJ   rK   rL     rf   z5PydanticModelTransformer.transform.<locals>.<genexpr>FT)r   c                 S   s   i | ]}|j | qS rJ   )name	serializerH   fieldrJ   rJ   rK   
<dictcomp>  s    z6PydanticModelTransformer.transform.<locals>.<dictcomp>)fieldsr   )r   clsinfoadjust_validator_signaturescollect_configcollect_fieldsr   r   r   final_iterationdeferrj   rk   add_initializeradd_construct_method
set_frozenr   r   set_values_dictr   r   )r]   rz   r   r   r   r   is_settingsrJ   rJ   rK   r|      s"   




z"PydanticModelTransformer.transformc                 C   sb   | j jjj D ]&\}}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   rh   ri   r   original_decoratorsr   calleer   r`   funcis_class)r]   r   rn   	first_decrJ   rJ   rK   r     s   


z4PydanticModelTransformer.adjust_validator_signaturesModelConfigDatac           	      C   s   | j }|j}t }|jjD ]2}t|tsq|jdkr?|jjD ]}t|ts&q|	| 
| q|jr?|js?| jjr?t|j| q|jjdd D ]&}t|jvrPqH|jtt| |jt d  D ]
\}}||| qcqH|S )zz
        Collects the values of the config attributes that are used by the plugin, accounting for parent classes.
        ConfigrN   Nr   )r   r   r   defsbodyrh   r   r   r   updateget_config_updatehas_alias_generatorr   rZ   r   error_required_dynamic_aliasesr   r   rk   r   r   add_plugin_dependencyr3   rb   r   
setdefault)	r]   rz   r   r   stmtsubstmtr   r   valuerJ   rJ   rK   r   )  s6   



z'PydanticModelTransformer.collect_configmodel_configPydanticModelFieldc                    s  | j }| j j}g }t }|jjD ]x}t|tsq|jd }t|tr't	|j
s(q|js5| jjr5t|j| |jj|j
}|du rBq|j}	t|	trKqt|	tsQq|	jrUq| |||}
| |\}}|rr|jsr| jjrrt|j| |t|j
|
|||j|jd | |j
 q|! }|jj"dd D ]O}t#|j$vrqg }|j%t&t'| |j$t# d ( D ]-\ } |vrt)||}|   || q fdd|D \}|*| || q|| }q|S )zR
        Collects the fields for the model, accounting for parent classes
        r   Nr   is_requiredaliashas_dynamic_aliaslinecolumnrN   r   c                 3   s    | ]
}|j  kr|V  qd S rF   r   )rH   ar   rJ   rK   rL     s    z:PydanticModelTransformer.collect_fields.<locals>.<genexpr>)+r   r   setr   r   rh   r   lvaluesr   rA   r   
new_syntaxrZ   r   error_untyped_fieldsr   r   r   r   ri   r!   r)   is_classvarget_is_requiredget_alias_infor   r   r   appendr   r   r   addcopyrk   r   r   r   r3   rb   r   deserializeremove)r]   r   rz   r   r   known_fieldsr   lhsrn   ri   r   r   r   
all_fieldsr   superclass_fieldsdatar   rJ   r   rK   r   H  st   








z'PydanticModelTransformer.collect_fieldsr   r   r   c           
      C   s   | j }| jj}|jdu}|pt|jo|j }| j||||d}| ||s7td}	|	t
|	ttjdt d|jjjvrHt|d|t  dS 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   rZ   r   r   r   r   get_field_argumentsshould_init_forbid_extrar)   r   r   r4   r9   explicitr   r   r   r   
add_methodr7   )
r]   r   r   r   rz   r  r  r  init_argumentsvarrJ   rJ   rK   r     s   
z(PydanticModelTransformer.add_initializerc              	   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rh|
}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.str_fields_setNTFr  z.object_PydanticBaseModelrO   r   	construct)return_type	self_typetvar_defis_classmethod)r   r   
named_typeBUILTINS_NAMEr<   r7   r   r)   r   r
  r   r`   rB   r(   r%   r   r   r   rh   r;   r  )r]   r   rz   set_stroptional_set_strfields_set_argumentconstruct_argumentsobj_typeself_tvar_nametvar_fullnametvdself_tvar_exprr  rJ   rJ   rK   r     s0   $


z-PydanticModelTransformer.add_construct_methodr   c                 C   s   | j jj}|D ]=}|j|j}|dur"|j}t|ts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.
        NF)r  rO   )r   r   r   r   r   r   ri   rh   r)   is_propertyto_varrb   r   	_fullnamer%   r   )r]   r   r   r   r   sym_noder  rJ   rJ   rK   r     s   
z#PydanticModelTransformer.set_frozenr   c                 C   s   |j d }t|tr|j| jv sdS |jdkrAt|jtr$|jjdk}nt|jtr1|jjdk}nt	|j| j
j| dS t|dS |jdkr[d}t|jtrV|jjdkrVd	}t|d
S t|jtru|jjdv rutd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   forbid)forbid_extrar   Tzbuiltins.NoneF)r   )builtins.Truezbuiltins.Falser(  rJ   )r   rh   r   r   r   rvaluer#   r   r   error_invalid_config_valuer   r   r   r`   )r]   r   r  r'  r   rJ   rJ   rK   r     s(   




z*PydanticModelTransformer.get_config_updater   r   r  c                 C   s   |j }t|tr$t| j|j j}t|tr"tdd |j	D r"dS dS t|t
rYt|jtrY|jjtkrYt|j|jD ]\}}|du sH|dkrO|jtu   S |dkrV dS q<dS t|tS )zg
        Returns a boolean indicating whether the field defined in `stmt` is a required field.
        c                 s   r   rF   )rh   r7   )rH   itemrJ   rJ   rK   rL     r   z;PydanticModelTransformer.get_is_required.<locals>.<genexpr>FTNr~   r   )r)  rh   r&   r=   r   r   r   r<   rj   r   r   r   r"   r`   rp   zipr   	arg_namesr_   r   )r   r   r  expr
value_typer   r   rJ   rJ   rK   r     s   
"
z(PydanticModelTransformer.get_is_requiredc                 C   s   | j }t|tr
dS t|trt|jtr|jjtksdS t|j	D ]\}}|dkr+q"|j
| }t|tr<|jdf  S  dS d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)  rh   r&   r   r   r"   r`   rp   	enumerater-  r   r#   r   )r   r.  iarg_namer   rJ   rJ   rK   r   "  s$   




z'PydanticModelTransformer.get_alias_infor  r  r  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  rJ   rK   
<listcomp>H  s    z@PydanticModelTransformer.get_field_arguments.<locals>.<listcomp>)r   r   r   )r]   r   r  r  r  	argumentsrJ   r5  rK   r
  ?  s
   
z,PydanticModelTransformer.get_field_argumentsc                 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'  rZ   r   )r]   r   r   rJ   rJ   rK   r  O  s   z1PydanticModelTransformer.should_init_forbid_extrar   c                 C   s8   | D ]}|j r
 dS q|r| 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   rJ   rJ   rK   r8  ]  s   
z1PydanticModelTransformer.is_dynamic_alias_present)rE   N)rE   r   ) r   r   r   r   r   r   r   r,   rY   r\   r|   r   r   r   r   r   r   r   r   r   r   r   staticmethodr   r   r   r	   r   r   r
  r  r8  rJ   rJ   rJ   rK   r{      s:   
 	


T% 
 r{   c                   @   s   e Zd Zdededee dededefddZd	ed
ede	fddZ
d	ededed
edef
ddZdefddZed	ededd fddZdS )r   r   r   r   r   r   r   c                 C   (   || _ || _|| _|| _|| _|| _d S rF   r   )r]   r   r   r   r   r   r   rJ   rJ   rK   r\   n  s   
zPydanticModelField.__init__r   r  rE   c                 C   s,   | j }|r| jd ur| j}t||| j  jS rF   )r   r   r)   r   )r]   r   r  r   rJ   rJ   rK   r#  x  s   zPydanticModelField.to_varr  r3  c                 C   sV   |r|| j  jd ur|| j  j}nttj}t| |||d |s#| js'tdS t	dS )N)variabletype_annotationinitializerkind)
r   r   r4   r9   r  r   r#  r   r   r   )r]   r   r  r3  r  r<  rJ   rJ   rK   r4  ~  s   

zPydanticModelField.to_argumentc                 C   s   | j S rF   )__dict__r]   rJ   rJ   rK   r     s   zPydanticModelField.serializer  c                 C   s   | di |S )NrJ   rJ   )r   r   r  rJ   rJ   rK   r     s   zPydanticModelField.deserializeN)r   r   r   r   r   r   rG   r\   r'   r)   r#  r   r4  r   r   classmethodr   rJ   rJ   rJ   rK   r   m  s&    

r   c                   @   s   e Zd Z						ddee dee dee dee dee dee fdd	Zd
eeef fddZ	ded  d
dfddZ
deded
dfddZdS )r   Nr'  r   r   r   r   r   c                 C   r:  rF   )r'  r   r   r   r   r   )r]   r'  r   r   r   r   r   rJ   rJ   rK   r\     s   	
zModelConfigData.__init__rE   c                 C   s   dd | j  D S )Nc                 S   s   i | ]\}}|d ur||qS rF   rJ   )rH   kvrJ   rJ   rK   r     s    z3ModelConfigData.set_values_dict.<locals>.<dictcomp>)r?  r   r@  rJ   rJ   rK   r        zModelConfigData.set_values_dictr   c                 C   s2   |d u rd S |   D ]
\}}t| || qd S rF   )r   r   r   )r]   r   rB  rC  rJ   rJ   rK   r     s
   zModelConfigData.updater   r   c                 C   s"   t | |d u rt| || d S d S rF   )r   r   )r]   r   r   rJ   rJ   rK   r     s   zModelConfigData.setdefault)NNNNNN)r   r   r   r   r   r\   r   r   r   r   r   r   rJ   rJ   rJ   rK   r     s.    
r   zpydantic-ormzInvalid from_orm call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   c                 C      |j d|  d|td d S )N"z" does not have orm_mode=Truecode)fail	ERROR_ORM)rF  r   r   rJ   rJ   rK   r        r   r   c                 C   rG  )NzInvalid value for "Config.rH  rI  )rK  ERROR_CONFIG)r   r   r   rJ   rJ   rK   r*    rM  r*  c                 C      | j d|td d S )Nz#Required dynamic aliases disallowedrI  )rK  ERROR_ALIASr   r   rJ   rJ   rK   r     rD  r   r   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!rI  )rK  ERROR_UNEXPECTED)r   r   r   linkfull_messagerJ   rJ   rK   r     s   r   c                 C   rO  )NzUntyped fields disallowedrI  )rK  ERROR_UNTYPEDrQ  rJ   rJ   rK   r     rD  r   c                 C   rO  )Nz>Field default and default_factory cannot be specified togetherrI  )rK  ERROR_FIELD_DEFAULTSrQ  rJ   rJ   rK   r     rD  r   Fr   r  r  r  r  is_newc                 C   s   | j j}||jv r |j| }	|	jr t|	jtr | j jj	|	j |p%t
|}|s*|r8ttdt|dtg}
n|p=t
|}ttd|dtg}
|
| }g g g }}}|D ]}|js_J d||j |t|j ||j qV| 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
    _clsN__pydantic_self__z"All arguments must be fully typed.z	.functionrO   TrA  )*r   r   r   plugin_generatedrh   ri   r   r   r   r   r>   r   r)   r:   make_normalizedr   r<  r   r   r;  r>  r   r  r  r5   	variablesr   r    r2   r   r   rb   r$  r   r?   is_decoratedr  r   r   r%   r   defn)rz   r   r   r  r  r  r  rX  r   rn   firstr   r-  	arg_kindsr   function_type	signaturer   r_namerC  decrJ   rJ   rK   r    sV   




r  xc                 C      | j }t|r
| S |S zc
    Used for compatibility with mypy 0.740; can be dropped once support for 0.740 is dropped.
    )r`   callablerf  fnrJ   rJ   rK   rb   (     rb   c                 C   rg  rh  )r   ri  rj  rJ   rJ   rK   r   2  rl  r   r   c                 C   s   |  dsd S d}tjdkrdd l}n-zdd l}W n& ty?   d}zdd l}W n ty<   dd l}|d Y Y d S w Y nw t	| |}|
|W  d    S 1 sUw   Y  d S )Nz.tomlrb)      r   rzJNo TOML parser installed, cannot read configuration from `pyproject.toml`.)rs   sysversion_infotomllibtomliImportErrortomlwarningswarnopenload)r   	read_modetoml_rw  rfrJ   rJ   rK   r   <  s*   



$r   )NNFF)vrq  configparserr   typingr   r   r   r   r   r   r	   r
   
TypingTyper   mypy.errorcodesr   
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)   mypy.optionsr*   mypy.pluginr+   r,   r-   r.   r/   r0   mypy.pluginsr1   mypy.semanalr2   mypy.server.triggerr3   
mypy.typesr4   r5   r6   r7   r8   r9   r:   r;   r<   r=   mypy.typevarsr>   	mypy.utilr?   mypy.versionr@   mypy_versionpydantic.utilsrA   rB   ru  r   r   rc   r   rp   rw   r   rG   rS   r   r  rW   rV   rY   rt   r{   r   r   rL  rN  rP  rS  rV  rW  r   r*  r   r   r   r   r   r  rb   r   r   rJ   rJ   rJ   rK   <module>   s    ,| 4
W   %	

S
"
