관리-도구
편집 파일: mypy.cpython-38.pyc
U /?�f�� � @ s� d Z ddlmZ ddlZddlmZ ddlmZmZm Z ddl m Z ddl m Z mZ ddlmZmZmZmZmZmZmZmZmZmZmZmZm 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/ dd l0m1Z1 dd l2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9 dd l:m;Z; dd l<m=Z= dd l>m?Z? ddl@mAZA ddlBmCZC ddlDmEZE ddlFmGZGmHZHmIZImJZJmKZKmLZLmMZMmNZNmOZOmPZPmQZQ ddlRmSZS ddlTmUZU ddlVmWZX ddlYmZZZ ddl[m\Z\ zddlFm]Z] W n" e^k �r� ddlFmOZ] Y nX dZ_dZ`dZad Zbd Zcd Zdd Zed Zfd!Zgd"d!d#d$d%d&hZhe\eX�Zieid'k�rBd(nd)Zjd*ZWd+d,d-�d.d/�ZkG d0d1� d1e7�ZlG d2d3� d3�Zmd4d5d6�d7d8�ZnG d9d:� d:�ZoG d;d<� d<�ZpG d=d>� d>�ZqG d?d@� d@�Zre dAdBdC�Zse dDdEdC�Zte dFdGdC�Zue dHdIdC�Zve dJdKdC�Zwe dJdLdC�Zxe dJdMdC�Zyd+dNdOdPdQ�dRdS�Zzd+dTdOdPdU�dVdW�Z{dTdOdPdX�dYdZ�Z|d+d[dOdPd\�d]d^�Z}dTdOdPdX�d_d`�Z~dNdOdPdX�dadb�ZdNdOdPdX�dcdd�Z�dsdfdgd+dhd5didjdkdPdl� dmdn�Z�d+dodp�dqdr�Z�dS )tz^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�FunctionContext� MethodContext�Plugin�ReportConfigContext� SemanticAnalyzerPluginInterface)� dataclasses)�deserialize_and_fixup_type)�set_callable_name)�make_wildcard_trigger)�state)�map_type_from_supertype) �AnyType� CallableType�Instance�NoneType� Overloaded�Type� TypeOfAny�TypeType� TypeVarType� UnionType�get_proper_type)� fill_typevars)� get_unique_redefinition_name)� __version__)�_fields)�parse_mypy_version)� TypeVarDef)r@ � pydantic-mypyzpydantic-mypy-metadatazpydantic.main.BaseModelz#pydantic_settings.main.BaseSettingsz pydantic.root_model.RootModelz5pydantic._internal._model_construction.ModelMetaclasszpydantic.fields.Fieldz pydantic.dataclasses.dataclassz.pydantic.functional_validators.model_validatorz.pydantic.functional_validators.field_validatorz*pydantic.functional_serializers.serializerz0pydantic.functional_serializers.model_serializerz.pydantic.deprecated.class_validators.validatorz3pydantic.deprecated.class_validators.root_validator)r i� �builtins� __builtins__� �strz type[Plugin])�version�returnc C s 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)rN � rQ �;/opt/hc_python/lib/python3.8/site-packages/pydantic/mypy.py�plugins s rS c s� e Zd ZdZddd�� fdd� Zddd �d d �Zdd d �d d�Zddd �dd�Zddd �dd�Zdd d �dd�Z ddd�dd�Z d d d�d d �Z d dd�d d!�Z d"d#d�d$d%�Z � ZS )&rP zThe Pydantic mypy plugin.r* �None��optionsrO c s& t |�| _| j�� | _t� �|� d S �N)�PydanticPluginConfig� plugin_config�to_data� _plugin_data�super�__init__)�selfrV �� __class__rQ rR r] � s zPydanticPlugin.__init__rM z(Callable[[ClassDefContext], bool] | None)�fullnamerO c C s: | � |�}|r6t|jt�r6tdd� |jjD ��r6| jS dS )z Update Pydantic model class.c s s | ]}|j tkV qd S rW )ra �BASEMODEL_FULLNAME��.0�baserQ rQ rR � <genexpr>� s z5PydanticPlugin.get_base_class_hook.<locals>.<genexpr>N)�lookup_fully_qualified� isinstance�noder( �any�mro�$_pydantic_model_class_maker_callback�r^ ra �symrQ rQ rR �get_base_class_hook� s z"PydanticPlugin.get_base_class_hookz(Callable[[ClassDefContext], None] | Nonec C s |t kr| jS dS )z,Update Pydantic `ModelMetaclass` definition.N)�MODEL_METACLASS_FULLNAME�)_pydantic_model_metaclass_marker_callback�r^ ra rQ rQ rR �get_metaclass_hook� s z!PydanticPlugin.get_metaclass_hookz(Callable[[FunctionContext], Type] | Nonec C s" | � |�}|r |jtkr | jS dS )z/Adjust the return type of the `Field` function.N)rg ra �FIELD_FULLNAME�_pydantic_field_callbackrm rQ rQ rR �get_function_hook� s z PydanticPlugin.get_function_hookz&Callable[[MethodContext], Type] | Nonec C s |� d�rtS dS )z-Adjust return type of `from_orm` method call.z .from_ormN)�endswith�from_attributes_callbackrr rQ rQ rR �get_method_hook� s z PydanticPlugin.get_method_hookc C s |t krtdk rtjS dS )z�Mark pydantic.dataclasses as dataclass. Mypy version 1.1.1 added support for `@dataclass_transform` decorator. )� rz N)�DATACLASS_FULLNAME�MYPY_VERSION_TUPLEr2 Z dataclass_class_maker_callbackrr rQ rQ rR �get_class_decorator_hook� s z'PydanticPlugin.get_class_decorator_hookr0 �dict[str, Any]��ctxrO c C s | j S )zjReturn all plugin config data. Used by mypy to determine if cache needs to be discarded. )r[ )r^ r� rQ rQ rR �report_config_data� s z!PydanticPlugin.report_config_datar, �boolc C s t |j|j|j| j�}|�� S rW )�PydanticModelTransformer�cls�reason�apirY � transform)r^ r� Z transformerrQ rQ rR rl � s z3PydanticPlugin._pydantic_model_class_maker_callbackc C s< | j jr dS |jjj}|s"td��t|jdd�r8d|j_dS )z�Reset dataclass_transform_spec attribute of ModelMetaclass. Let the plugin handle it. This behavior can be disabled if 'debug_dataclass_transform' is set to True', for testing purposes. Nz-callback not passed from 'get_metaclass_hook'�dataclass_transform_spec) rY �debug_dataclass_transformr� �infoZdeclared_metaclass�AssertionError�getattr�typer� )r^ r� Zinfo_metaclassrQ rQ rR rq � s z8PydanticPlugin._pydantic_model_metaclass_marker_callbackr- r= c s |j � |jd dks td��|jd dks2td��|jd }|jd }|r`|r`t|j|j� � S |r�|jd d }|d }t|t �s�|S nt|r�|jd d }t|t �r�|j d }t|t �r�|j }t|dd�}|r�td d � |D ��r�t� fd d �|D ��|_|S � S ) ao Extract the type of the `default` argument from the Field function, and use it as the return type. In particular: * Check whether the default and default_factory argument is specified. * Output an error if both are specified. * Retrieve the type of the argument which is specified, and use it as return type for the function. r �defaultz0"default" is no longer first argument in Field()rz �default_factoryz9"default_factory" is no longer second argument in Field()�argsNc s s | ]}t |t�V qd S rW )rh r@ )rd �argrQ rQ rR rf � s z:PydanticPlugin._pydantic_field_callback.<locals>.<genexpr>c 3 s | ] }� V qd S rW rQ )rd �_�Zdefault_any_typerQ rR rf � s )�default_return_typeZcallee_arg_namesr� r� �+error_default_and_default_factory_specifiedr� �context� arg_typesrh r r< �itemsr9 �ret_typer� �all�tuple) r^ r� Z default_argsZdefault_factory_argsZ default_typeZ default_argZdefault_factory_typer� r� rQ r� rR ru � s2 z'PydanticPlugin._pydantic_field_callback)�__name__� __module__� __qualname__�__doc__r] ro rs rv ry r} r� rl rq ru � __classcell__rQ rQ r_ rR rP � s rP c @ sT e Zd ZU dZdZded< ded< ded< ded< dd d �d d �Zd d�dd�ZdS )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� r� r� r� r� r� r* rT rU c C s� |j d krd S t|j �}|d k rr|�di ��di �}| jD ]4}|�|d�}t|t�sbtd|� ���t| ||� q:n:t� }|� |j � | jD ] }|j t |dd�}t| ||� q�d S )NZtoolrI Fz/Configuration value must be a boolean for key: )�fallback) � config_file� parse_toml�get� __slots__rh r� � ValueError�setattrr �read� getboolean�CONFIGFILE_KEY)r^ rV Z toml_config�config�keyZsettingrY rQ rQ rR r] s z PydanticPluginConfig.__init__r~ �rO c s � fdd�� j D �S )z/Returns a dict of config names to their values.c s i | ]}|t � |��qS rQ )r� )rd r� �r^ rQ rR � <dictcomp># s z0PydanticPluginConfig.to_data.<locals>.<dictcomp>)r� r� rQ r� rR rZ ! s z PydanticPluginConfig.to_dataN)r� r� r� r� r� �__annotations__r] rZ rQ rQ rQ rR rX � s rX r. r= r c C s� | j }t|t�r|j}t|t�r4t|jt�r4|j}n<t|t�rD|}n,d|� d|jj� d�}t || j | j � | j S |j j �t�}|dkr�| j S |�di ��d�}|dk r�t|j j| j | j � | j S )z1Raise an error if from_attributes is not enabled.z ctx.type: � (of type �)Nr� �from_attributesT)r� rh r? �itemr9 r� r: r` r� �error_unexpected_behaviorr� r� r� �metadatar� � METADATA_KEY�error_from_attributes�name)r� Zctx_typeZ model_type�detailZpydantic_metadatar� rQ rQ rR rx & s" rx c @ s� e Zd ZdZddddddddd�d d �Zddddd dd d �dd�Zd$dd ddd�dd�Zd%dd dddd�dd�Zdd�dd�Ze ddd d d �d d ��Z dd d d �d!d"�Z d#S )&�PydanticModelFieldz5Based on mypy.plugins.dataclasses.DataclassAttribute.rM z str | Noner� �int� Type | Noner( �r� �alias�has_dynamic_alias� has_default�line�columnr� r� c C s4 || _ || _|| _|| _|| _|| _|| _|| _d S rW r� ) r^ r� r� r� r� r� r� r� r� rQ rQ rR r] @ s zPydanticModelField.__init__r1 r )� current_info�typed�force_optional� use_aliasr� �force_typevars_invariantrO c C sH | � ||||�}|r | �||�nttj�}t||d|s<| jr@tntd�S )zABased on mypy.plugins.dataclasses.DataclassAttribute.to_argument.N)�variable�type_annotationZ initializer�kind) �to_varr r8 r> �explicitr r� r r ) r^ r� r� r� r� r� r� r� r� rQ rQ rR � to_argumentT s �z PydanticModelField.to_argumentF)r� r� r� rO c C s� |r&t | jt�r&| j�� }t|_|| _| jdk r�t| jdd�dk r�t� |j j ��L t |�}|rz|j D ]}t |t�rdt|_qdt| j| jjj|i�W 5 Q R � S Q R X | jS )zABased on mypy.plugins.dataclasses.DataclassAttribute.expand_type.N� self_type)rh r� r@ Z copy_modifiedr Zvariancer� r� r6 �strict_optional_setrV �strict_optionalrC r� r r� �id)r^ r� r� r� Z modified_typeZfilled_with_typevarsr� rQ rQ rR r g s (z PydanticModelField.expand_typer) )r� r� r� r� rO c C s0 |r| j dk r| j }n| j}t|| �|||��S )z<Based on mypy.plugins.dataclasses.DataclassAttribute.to_var.N)r� r� r) r )r^ r� r� r� r� r� rQ rQ rR r� � s zPydanticModelField.to_varr r� c C s0 | j s t�| 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� � serializer� rQ rQ rR r� � s �z PydanticModelField.serialize)r� �datar� rO c C s, |� � }t|�d�|�}| f ||d�|��S )�ABased on mypy.plugins.dataclasses.DataclassAttribute.deserialize.r� )r� r� )�copyr3 �pop)r� r� r� r� �typrQ rQ rR � deserialize� s z PydanticModelField.deserializerT )�sub_typer� rO c C s: | j dk r6t�|jj�� t| j || j�| _ W 5 Q R X dS )zxExpands type vars in the context of a subtype when an attribute is inherited from a generic super type. N)r� r6 r� rV r� r7 r� )r^ r� r� rQ rQ rR �expand_typevar_from_subtype� s z.PydanticModelField.expand_typevar_from_subtypeN)F)F) r� r� r� r� r] r� r r� r� � classmethodr� r� rQ rQ rQ rR r� = s �" � r� c @ s: e Zd ZdZdd� Zedd d�dd��Zdd�d d �Zd S ) �PydanticModelClassVarz�Based on mypy.plugins.dataclasses.DataclassAttribute. ClassVars are ignored by subclasses. Attributes: name: the ClassVar name c C s || _ d S rW �r� )r^ r� rQ rQ rR r] � s z PydanticModelClassVar.__init__r )r� rO c C s |� � }| f |�S )r� )r� )r� r� rQ rQ rR r� � s z!PydanticModelClassVar.deserializer� c C s d| j iS )r� r� r� r� rQ rQ rR r� � s �z PydanticModelClassVar.serializeN)r� r� r� r� r] r� r� r� rQ rQ rQ rR r� � s r� c @ s� e Zd ZU dZdddddhZded< d d d d d d�dd�Zdd�dd�Zd d�dd�Zdd�dd�Z dddd�d d �Z d d d �d!d"�Z d#d d$�d%d&�Z d'dd(d)d*�d+d,�Z d-d.d/d0d1�d2d3�Zd4dddd d5�d6d7�Zd4ddd d8�d9d:�Zd4d dd d;�d<d=�ZdTd.d?dd@dA�dBdC�Zed'dd �dDdE��Zed'dFd �dGdH��ZdUd4ddddddIdJ�dKdL�Zd4dddM�dNdO�Zed4dddP�dQdR��ZdSS )Vr� z�Transform the BaseModel subclass according to the plugin settings. Attributes: tracked_config_fields: A set of field configs that the plugin has to track their value. �extra�frozenr� �populate_by_name�alias_generatorzset[str]�tracked_config_fieldsr zExpression | Statementr1 rX rT )r� r� r� rY rO c C s || _ || _|| _|| _d S rW )�_clsZ_reason�_apirY )r^ r� r� r� rY rQ rQ rR r] � s z!PydanticModelTransformer.__init__r� r� c C s� | j j}tdd� |jdd� D ��}| �� }| �||�\}}|dksL|dkrPdS |D ]}|jdkrT dS qTtdd� |jdd� D ��}| �||||� |s�| �|||� | j || j |j dkd� | � � d d � |D �d d � |D �|� � d �|jt< dS ) a� Configures the BaseModel subclass according to the plugin settings. In particular: * determines the model config and fields, * adds a fields-aware signature for the initializer and construct methods * freezes the class if frozen = True * stores the fields, config, and if the class is settings in the mypy metadata for access by subclasses c s s | ]}t |jkV qd S rW )�ROOT_MODEL_FULLNAMEra rc rQ rQ rR rf � s z5PydanticModelTransformer.transform.<locals>.<genexpr>N���Fc s s | ]}|j tkV qd S rW )ra �BASESETTINGS_FULLNAMErc rQ rQ rR rf � s T)r� c S s i | ]}|j |�� �qS rQ �r� r� �rd �fieldrQ rQ rR r� s z6PydanticModelTransformer.transform.<locals>.<dictcomp>c S s i | ]}|j |�� �qS rQ r� )rd Z class_varrQ rQ rR r� s )�fields� class_varsr� )r� r� rj rk �collect_config� collect_fields_and_class_varsr� �add_initializer�add_model_construct_method� set_frozenr� r� �adjust_decorator_signatures�get_values_dictr� r� )r^ r� � is_root_modelr� r� r� r� � is_settingsrQ rQ rR r� � s( � z"PydanticModelTransformer.transformc s� | j jj�� D ]t\}}t|jt�r|jjd � t� t�rt� j t �r� j j t kr� j j t krxt� fdd�t� j�D ��sd|jj_qdS )a� When we decorate a function `f` with `pydantic.validator(...)`, `pydantic.field_validator` or `pydantic.serializer(...)`, mypy sees `f` as a regular method taking a `self` instance, even though pydantic internally wraps `f` with `classmethod` if necessary. Teach mypy this by marking any function whose outermost decorator is a `validator()`, `field_validator()` or `serializer()` call as a `classmethod`. r c 3 s4 | ],\}}� j | d ko*t|t�o*|jdkV qdS )�mode�afterN)� arg_namesrh r$ �value)rd �ir� �Z first_decrQ rR rf s �zGPydanticModelTransformer.adjust_decorator_signatures.<locals>.<genexpr>TN)r� r� �namesr� rh ri r Zoriginal_decoratorsr �calleer ra �DECORATOR_FULLNAMES�MODEL_VALIDATOR_FULLNAMErj � enumerater� �func�is_class)r^ r� rn rQ r rR r� s � � � � ��z4PydanticModelTransformer.adjust_decorator_signatures�ModelConfigDatac C sL | j }t� }d}d}|j�� D ]&\}}| �||�}|r d}|�|� q d}|jjD �]V}t|t t f�shqRt|t ��r"|j d } t| t �rR| j dkr�qRt|jt�r�t|jj|jj�D ](\} } | dkr�q�|�| j| | dd�� q�nBt|jt��r�|jjD ]*\} } t| t��s q�|�| �| j| �� q�njt|t ��r�|j dk�r<qR|jjD ]F}t|t ��sX�qD|j d } t| t ��sr�qD|�| �| j |j�� �qD|�r�| j�d|� �q�d}qR|�s�|�r�|�r�|j�r�|j�s�| jj�r�t | j|� |j j d d� D ]R}t |j k�r�q�| j�!t"|j#�� |j t d �� D ]\}}|�$||� �q,�q�|S ) zhCollects the values of the config attributes that are used by the plugin, accounting for parent classes.FTNr � model_config)� lax_extraZConfigzYSpecifying config in two places is ambiguous, use either Config attribute or class kwargsrz r� )%r� r �keywordsr� �get_config_update�update�defs�bodyrh r r �lvaluesr r� �rvaluer �zipr� r� r r$ r� r� �fail�has_alias_generatorr� rY r� � error_required_dynamic_aliasesr� rk r� r� �add_plugin_dependencyr5 ra � setdefault)r^ r� r� Zhas_config_kwargsZhas_config_from_namespacer� �exprZ config_data�stmt�lhs�arg_namer� Zkey_exprZ value_exprZsubstmtr� r� rQ rQ rR r� % sz � ���� z'PydanticModelTransformer.collect_configzJtuple[list[PydanticModelField] | None, list[PydanticModelClassVar] | None])r r� rO c C s� | j }i }i }t|jjdd� �D ]�}t|jkr2q"| j�t|j �� |jt d � � D ]d\}}t � ||| j�} | � |j| j� | ||< |jj�|�} | rV| jrVt| jt�sV| j�d| j� qV|jt d � � D ]\}}t� |�||< q�q"t� } t� } | �|j�D ]�} | �| ||�}t|t ��r`| jd }|�rH|jdk�rHt| j| � n| � |j� |||j< n,t|t��r | jd }| � |j� |||j< �q t |� � �t |� � �fS )zACollects the fields for the model, accounting for parent classes.rz r� r� z7BaseModel field may only be overridden by another fieldr� r �root) r� �reversedr� rk r� r� r� r r5 ra r� r� r� r� r r� ri rh r) r r� �set�%_get_assignment_statements_from_blockr �$collect_field_or_class_var_from_stmtr r� � error_extra_fields_on_root_model�add�list�values)r^ r r� r� Z found_fieldsZfound_class_varsr� r� r� r� �sym_nodeZcurrent_field_namesZcurrent_class_vars_namesr Z maybe_fieldr rQ rQ rR r� q sD � z6PydanticModelTransformer.collect_fields_and_class_varsr zIterator[AssignmentStmt])r rO c c sJ |j D ]}|js| �|�E d H q|jd k rF|jjsF| �|j�E d H d S rW )r Zis_unreachabler Z else_body)r^ r r rQ rQ rR �,_get_assignment_statements_from_if_statement� s zEPydanticModelTransformer._get_assignment_statements_from_if_statementr )�blockrO c c s< |j D ]0}t|t�r |V qt|t�r| �|�E d H qd S rW )r rh r r r& )r^ r' r rQ rQ rR r � s z>PydanticModelTransformer._get_assignment_statements_from_blockr z dict[str, PydanticModelClassVar]z1PydanticModelField | PydanticModelClassVar | None)r r r� rO c C s, | j }|jd }t|t�r0t�|j�r0|jdkr4dS |js�t|jt �rxt|jj t �rxt|jj j t�rx|jj j j t krxdS |j|kr�dS t | j|� dS |jd }t|t�s�dS t�|j�r�|jdkr�dS |jj�|j�}|dkr�dS |j}t|t�r�dS t|t��r| j�d|� dS t|t��s&dS |j�r8t|j�S t|j�}t|t��rj|jj dk�rj| j�d|� | � |�} |jdk�r�|j �r�|j �r�| jj |jdd�} | �r�| |_n| j�d |� t t!j"�|_| �#|�\} } | �r�|j$�s�| j%j&�r�t'| j|� | �(||j|�} t)|j| | | |j*|j+| |jd �S ) a^ Get pydantic model field from statement. Args: stmt: The statement. model_config: Configuration settings for the model. class_vars: ClassVars already known to be defined on the model. Returns: A pydantic model field if it could find the field in statement. Otherwise, `None`. r r NzFType aliases inside BaseModel definitions are not supported at runtimezdataclasses.InitVarz%InitVar is not supported in BaseModelT)�is_finalzGNeed type argument for Final[...] with non-literal default in BaseModel)r� r� r� r� r� r� r� r� ),r� r rh r rF Zis_valid_field_namer� Z new_syntaxr r r ra r �error_untyped_fieldsr� r� r r� ri r! r' r r) Z is_classvarr� rB r� r: �get_has_defaultr( Z is_inferredZanalyze_simple_literal_typer8 r> Z from_error�get_alias_infor� rY r� r � _infer_dataclass_attr_init_typer� r� r� )r^ r r r� r� r rn ri Z node_typer� r� r� r� Z init_typerQ rQ rR r � s� � ��� � � � �z=PydanticModelTransformer.collect_field_or_class_var_from_stmtr% rM r r� )rn r� r� rO c C s� |j }|jr|S t|j �}t|t�s(|S |j �d�}|r�t|jt�r�|j �d�}|sXt �|j rttt |j |j |��}n t t j �S t|t�r�|jtttgkr�t|jd |�S | j�d|j j� d�|� n| j�d|j j� d�|� |S )zvInfer __init__ argument type for an attribute. In particular, possibly use the signature of __set__. �__set__rL z(Unsupported signature for "__set__" in "�"zUnsupported "__set__" in ")r� ZimplicitrB rh r: r� ri r Zget_containing_type_infor� r7 r8 r> Z unannotatedr9 � arg_kindsr r r� r� r r� ) r^ rn r� r� r� �t�setterZ super_infoZ setter_typerQ rQ rR r, = s. � z8PydanticModelTransformer._infer_dataclass_attr_init_typezlist[PydanticModelField])r� r� r� r� rO c C sr d| j jjkr"| j jjd js"dS | jj}|jdk }t|joB|j �}| j |||||dd�}|r�t dkr�|d j t krzt nt|d _ |�r&| j�t�j} d| jk�r&| jd j} | dk �r&| jdk �r&| j} t| j�D ]N\} } | �d�s�| �d�s�q�| j�| j| �}t| |�}|�t||dt�� q�| � ||��sVtd �}|�t|t t j �dt �� t!| j| j d|t"� d � dS ) z�Adds a fields-aware `__init__` method to the class. The added `__init__` will be annotated with types vs. all `Any` depending on the plugin settings. r] NT)r� �requires_dynamic_aliasesr� r� r� )rz r rz r �__r� �kwargs)r� � return_type)#r� r� r �plugin_generatedrY r� r� r� r �get_field_argumentsr| r� r r r r� rg r� ri r� r r� � startswithZ anal_typer� r) �appendr �should_init_forbid_extrar8 r> r� r � add_methodr; )r^ r� r� r� r� r� r� r2 r� Zbase_settings_nodeZbase_settings_init_nodeZ func_typeZarg_idxr Zanalyzed_variable_typer� �varrQ rQ rR r� c s> � z(PydanticModelTransformer.add_initializer)r� r� r� rO c C s� | j �t� d�| j �t� d��g�}t|t� g�}ttd|�|dt�}t� | j j j �� | j |ddd|d�}W 5 Q R X | � ||�s�td�}|�t|ttj�dt�� |g| }t| j | jd |t| jj�dd � dS ) z�Adds a fully typed `model_construct` classmethod to the class. Similar to the fields-aware __init__ method, but always uses the field names (not aliases), and does not treat settings fields as optional. z.setz.strZ _fields_setNTF)r� r2 r� r� r4 Zmodel_construct)r� r5 �is_classmethod)r� � named_type� BUILTINS_NAMErA r; r r) r r6 r� rV r� r7 r: r9 r8 r> r� r r; r� rC r� ) r^ r� r� r� Zset_strZoptional_set_strZfields_set_argumentr� r< rQ rQ rR r� � s. $ � �z3PydanticModelTransformer.add_model_construct_method)r� r� r� rO c C s� | j j}|D ]�}|j�|j�}|dk r�|j}t|t�r>||_q�t|t �r\| j j s\| j � � q�z t |�}W n tk r� t|�}Y nX d|� d|j� d�} t| | j | j � q |j||dd�}||_||_|jd |j |_tt|�|j|j< q dS )z�Marks all fields as properties so that attempts to set them trigger mypy errors. This is the same approach used by the attrs and dataclasses plugins. Nzsym_node.node: r� r� F)r� �.)r� r� r r� r� ri rh r) Z is_propertyr! r� Zfinal_iterationZdeferrM � TypeError�reprr` r� r� ra � _fullnamer% r ) r^ r� r� r� r� r� r% r< Zvar_strr� rQ rQ rR r� � s( z#PydanticModelTransformer.set_frozenFr �ModelConfigData | None)r� r� r rO c C s� || j krdS |dkrbt|t�r,|jdk}n,t|t�rB|jdk}n|sTt|| j|� dS t|d�S |dkr�d}t|t �r�|j dkr�d}t|d �S t|t �r�|j d kr�tf ||j d ki�S t|| j|� dS ) z�Determines the config update due to a single kwarg in the ConfigDict definition. Warns if a tracked config attribute is set to a value the plugin doesn't know how to interpret (e.g., an int) Nr� Zforbid)� forbid_extrar� T� builtins.NoneF)r )� builtins.Truezbuiltins.FalserG ) r� rh r$ r� r r� �error_invalid_config_valuer� r r ra )r^ r� r� r rE r rQ rQ rR r � s( z*PydanticModelTransformer.get_config_updatec C s� | j }t|t�rdS t|t�r�t|jt�r�|jjtkr�t|j |j �D ]H\}}|dks\|dkrj|j t k S |dkrDt|t �o�|jdk S qDdS t|t � S )zUReturns a boolean indicating whether the field defined in `stmt` is a required field.FNr� r� rF )r rh r& r r r" ra rt r r� r� r` r r )r r r� r� rQ rQ rR r* � s " z(PydanticModelTransformer.get_has_defaultztuple[str | None, bool]c C s� | j }t|t�rdS t|t�r6t|jt�r6|jjtks:dS t|j �D ]:\}}|dkrVqD|j | }t|t �rx|j df S dS qDdS )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" ra rt r r� r� r$ r� )r r r� r r� rQ rQ rR r+ s$ � � � z'PydanticModelTransformer.get_alias_info�list[Argument])r� r� r� r2 r� r� rO c s* �j j�� ������fdd�|D �}|S )z�Helper function used during the construction of the `__init__` and `model_construct` method signatures. Returns a list of mypy Argument instances for use in the generated signatures. c s2 g | ]*}�r|j s|j���p ���j� d ��qS ))r� r� r� r� r� )r� r� r� r� �r� r� r� r2 r^ r� r� rQ rR � <listcomp>1 s ��z@PydanticModelTransformer.get_field_arguments.<locals>.<listcomp>)r� r� )r^ r� r� r� r2 r� r� � argumentsrQ rJ rR r7 # s � z,PydanticModelTransformer.get_field_arguments)r� r� rO c C s. |j s | �|t|j��r dS |jr&dS | jjS )a@ Indicates whether the generated `__init__` should get a `**kwargs` at the end of its signature. We disallow arbitrary kwargs if the extra config setting is "forbid", or if the plugin config says to, *unless* a required dynamic alias is present (since then we can't determine a valid signature). FT)r� �is_dynamic_alias_presentr� r rE rY r� )r^ r� r� rQ rQ rR r: ? s z1PydanticModelTransformer.should_init_forbid_extra)r� r rO c C s8 | D ]}|j r dS q|r4| D ]}|jdkr dS q dS )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� rQ rQ rR rM L s z1PydanticModelTransformer.is_dynamic_alias_presentN)F)F)r� r� r� r� r� r� r] r� r� r� r� r& r r r, r� r� r� r � staticmethodr* r+ r7 r: rM rQ rQ rQ rR r� � s: � %LA}&. #" � r� c @ sX e Zd ZdZddddddd�dd�Zdd�d d �Zd d d �dd�Zddd d�dd�ZdS )r z(Pydantic mypy plugin model config class.Nz bool | None�rE r� r� r� r c C s" || _ || _|| _|| _|| _d S rW rO )r^ rE r� r� r� r rQ rQ rR r] ^ s zModelConfigData.__init__r~ r� c C s dd� | j �� D �S )z�Returns a dict of Pydantic model config names to their values. It includes the config if config value is not `None`. c S s i | ]\}}|d k r||�qS rW rQ )rd �k�vrQ rQ rR r� q s z3ModelConfigData.get_values_dict.<locals>.<dictcomp>)�__dict__r� r� rQ rQ rR r� l s z ModelConfigData.get_values_dictrD rT )r� rO c C s2 |dkr dS |� � �� D ]\}}t| ||� qdS )z$Update Pydantic model config values.N)r� r� r� )r^ r� rP rQ rQ rQ rR r s s zModelConfigData.updaterM r )r� r� rO c C s t | |�dkrt| ||� dS )zFSet default value for Pydantic model config if config value is `None`.N)r� r� )r^ r� r� rQ rQ rR r z s zModelConfigData.setdefault)NNNNN)r� r� r� r� r] r� r r rQ rQ rQ rR r [ s �r z pydantic-ormz Invalid from_attributes callZPydanticzpydantic-configzInvalid config valuezpydantic-aliaszDynamic alias disallowedzpydantic-unexpectedzUnexpected behaviorzpydantic-fieldzUntyped field disallowedzInvalid Field defaultsz!Extra field on RootModel subclassr+ r rT )� model_namer� r� rO c C s |j d| � d�|td� dS )zCEmits an error when the model does not have `from_attributes=True`.r. z$" does not have from_attributes=True��codeN)r � ERROR_ORM)rS r� r� rQ rQ rR r� � s r� r1 )r� r� r� rO c C s |j d| � d�|td� dS )z0Emits an error when the config value is invalid.zInvalid value for "Config.r. rT N)r � ERROR_CONFIG)r� r� r� rQ rQ rR rH � s rH )r� r� rO c C s | j d|td� dS )znEmits required dynamic aliases error. This will be called when `warn_required_dynamic_aliases=True`. z#Required dynamic aliases disallowedrT N)r � ERROR_ALIAS�r� r� rQ rQ rR r � s r z8CheckerPluginInterface | SemanticAnalyzerPluginInterface)r� r� r� rO c C s4 d}d| � d�}|d|� d�7 }|j ||td� dS )z Emits unexpected behavior error.z6https://github.com/pydantic/pydantic/issues/new/choosez7The pydantic mypy plugin ran into unexpected behavior: � z&Please consider reporting this bug at z so we can try to fix it!rT N)r �ERROR_UNEXPECTED)r� r� r� �linkZ full_messagerQ rQ rR r� � s r� c C s | j d|td� dS )z;Emits an error when there is an untyped field in the model.zUntyped fields disallowedrT N)r � ERROR_UNTYPEDrY rQ rQ rR r) � s r) c C s | j d|td� dS )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`rT N)r � ERROR_EXTRA_FIELD_ROOT_MODELrY rQ rQ rR r! � s r! c C s | j d|td� dS )zNEmits an error when `Field` has both `default` and `default_factory` together.z>Field default and default_factory cannot be specified togetherrT N)r �ERROR_FIELD_DEFAULTSrY rQ rQ rR r� � s r� Fz8SemanticAnalyzerPluginInterface | CheckerPluginInterfacer rI r� zTypeVarDef | Noner� ) r� r� r� r� r5 r� �tvar_defr= rO c C s |j }||jkr<|j| } | jr<t| jt�r<|jj�| j� t| t �rR| � d�} n | � dg �} |r�|ppt t |��}ttd�|dtd�g} n |p�t |�}ttd�|dt�g} | | }g g g } } }|D ]8}|js�td��| �|j� | �|jj� |�|j� q�t| || || �}|�r |g|_t||tt� g��}||_ t||�|_ ||_ |j d | |_ |j |_ ||jk�r�t!||j�}|j| |j|< |�r�d|_"t||j �}||_ |j |_ d|_#t$|t%d�g|�}|j |_ t&t'|�} n t&t'|�} d| _| |j|< |j(jj�|� dS ) zhVery closely related to `mypy.plugins.common.add_method_to_class`, with a few pydantic-specific changes.zbuiltins.functionr� NTZ__pydantic_self__z"All arguments must be fully typed.r@ r� ))r� r r6 rh ri r r r �remover1 r> Znamed_generic_typer? rC r r) r r� r� r9 r� r� r� r9 � variablesr r r4 r� r ra rC r� rD Z is_decoratedr= r r r% r Zdefn)r� r� r� r� r5 r� r` r= r� rn Z function_type�firstr� r� r/ r� � signaturer Zr_namerQ �decrQ rQ rR r; � sZ r; zdict[str, Any] | None)r� rO c C s� | � d�sdS tjdkr"ddl}n6z ddl}W n( tk rV ddl}|�d� Y dS X t| d��}|� |�W 5 Q R � S Q R X dS )z�Returns a dict of config keys to values. It reads configs from toml file and returns `None` if the file is not a toml file. z.tomlN)� � r zJNo TOML parser installed, cannot read configuration from `pyproject.toml`.�rb) rw �sys� version_info�tomllib�tomli� ImportError�warnings�warn�open�load)r� Ztoml_rn �rfrQ rQ rR r� s r� )NNF)�r� � __future__r ri � configparserr �typingr r r Zmypy.errorcodesr Zmypy.expandtyper r 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& r' r( r) Z mypy.optionsr* Z mypy.pluginr+ r, r- r. r/ r0 r1 Z mypy.pluginsr2 Zmypy.plugins.commonr3 Z mypy.semanalr4 Zmypy.server.triggerr5 Z mypy.stater6 Z mypy.typeopsr7 Z mypy.typesr8 r9 r: r; r< r= r> r? r@ rA rB Z mypy.typevarsrC Z mypy.utilrD Z mypy.versionrE Z mypy_versionZpydantic._internalrF Zpydantic.versionrG rH rm r� r� rb r� r� rp rt r{ r r r| r? rS rP rX rx r� r� r� r rV rW rX r[ r] r_ r^ r� rH r r� r) r! r� r; r� rQ rQ rQ rR �<module> s� �" $ 4 � v.s % � X