o pf!!@sddZddlmZddlZddlZddlZddlmZm Z ddlm Z m Z m Z ddl mZmZmZmZddlmZdd lmZdd lmZdd lmZdd lmZd dlmZmZmZd dlm Z d dl!m"Z"d dl#m$Z$d dl%m&Z&d dl'm(Z(d dl)m*Z*ej+rddl,m-Z-Gdddej.Z/Gddde/ej.Z0neZ1d/d0d d!Z2d"d#d1d)d*Z3d2d-d.Z4dS)3z0Private logic for creating pydantic dataclasses.) annotationsN)partialwraps)AnyCallableClassVar) ArgsKwargsSchemaSerializerSchemaValidator core_schema) TypeGuard)PydanticUndefinedAnnotation) FieldInfo)create_schema_validator)PydanticDeprecatedSince20)_config _decorators _typing_extra)collect_dataclass_fields)GenerateSchema)get_standard_typevars_map)set_dataclass_mocks)CallbackGetCoreSchemaHandler)generate_pydantic_signature) ConfigDictc@s0eZdZUded<ded<ded<dd d ZdS)StandardDataclasszClassVar[dict[str, Any]]__dataclass_fields__z ClassVar[Any]__dataclass_params__zClassVar[Callable[..., None]] __post_init__argsobjectkwargsreturnNonecOsdSN)selfr!r#r'r'X/home/ertert/spirit/venv/lib/python3.10/site-packages/pydantic/_internal/_dataclasses.py__init__&szStandardDataclass.__init__N)r!r"r#r"r$r%)__name__ __module__ __qualname____annotations__r*r'r'r'r)r!s rc@sJeZdZUdZded<ded<ded<ded <d ed <d ed <ded<dS)PydanticDataclassaiA protocol containing attributes only available once a class has been decorated as a Pydantic dataclass. Attributes: __pydantic_config__: Pydantic-specific configuration settings for the dataclass. __pydantic_complete__: Whether dataclass building is completed, or if there are still undefined fields. __pydantic_core_schema__: The pydantic-core schema used to build the SchemaValidator and SchemaSerializer. __pydantic_decorators__: Metadata containing the decorators defined on the dataclass. __pydantic_fields__: Metadata about the fields defined on the dataclass. __pydantic_serializer__: The pydantic-core SchemaSerializer used to dump instances of the dataclass. __pydantic_validator__: The pydantic-core SchemaValidator used to validate instances of the dataclass. zClassVar[ConfigDict]__pydantic_config__zClassVar[bool]__pydantic_complete__z ClassVar[core_schema.CoreSchema]__pydantic_core_schema__z$ClassVar[_decorators.DecoratorInfos]__pydantic_decorators__zClassVar[dict[str, FieldInfo]]__pydantic_fields__zClassVar[SchemaSerializer]__pydantic_serializer__zClassVar[SchemaValidator]__pydantic_validator__N)r+r,r-__doc__r.r'r'r'r)r/)s   r/clstype[StandardDataclass]types_namespacedict[str, Any] | Noner$r%cCs t|}t|||d}||_dS)zCollect and set `cls.__pydantic_fields__`. Args: cls: The class. types_namespace: The types namespace, defaults to `None`. ) typevars_mapN)rrr4)r8r:r<fieldsr'r'r)set_dataclass_fieldsDs r>T) raise_errors type[Any]config_wrapper_config.ConfigWrapperr?boolc st|dr tdt|durt|}t||t|}t|||}t |j |j |dd}dd d}|j d|_ ||_ |j |_||_t|dd}z|r\||tt|jdd|dd} n|j|dd} Wn"ty} z|rot||jd| jdWYd} ~ dSd} ~ ww||} z|| } Wn|jyt||jdYdSwtd|}| |_t| ||j|j d| |j |_!t"| | |_#|j$rt%|j&d fdd } | 'd||_&dS)!aFinish building a pydantic dataclass. This logic is called on a class which has already been wrapped in `dataclasses.dataclass()`. This is somewhat analogous to `pydantic._internal._model_construction.complete_model_class`. Args: cls: The class. config_wrapper: The config wrapper instance. raise_errors: Whether to raise errors, defaults to `True`. types_namespace: The types namespace. Returns: `True` if building a pydantic dataclass is successfully completed, `False` otherwise. Raises: PydanticUndefinedAnnotation: If `raise_error` is `True` and there is an undefined annotations. __post_init_post_parse__zVSupport for `__post_init_post_parse__` has been dropped, the method will not be calledNT)initr=rA is_dataclass__dataclass_self__r/r!rr#r$r%c_s"d}|}|jjt|||ddS)NT) self_instance)r6validate_pythonr)rGr!r#__tracebackhide__sr'r'r)r*sz$complete_dataclass..__init__z .__init____get_pydantic_core_schema__F)from_dunder_get_core_schemaunpack)ref_mode`zall referenced typesztype[PydanticDataclass] dataclassinstance__fieldstr__valuecs|||dSr&)validate_assignment)rRrSrU validatorr'r)validated_setattrsz-complete_dataclass..validated_setattr)rGr/r!rr#rr$r%)rRrrSrTrUrTr$r%)(hasattrwarningswarnDeprecationWarningrget_cls_types_namespacer>rrrr*r4r- config_dictr0 __signature__getattrrrgenerate_schemarrr+name core_config clean_schemaCollectedInvalidtypingcastr2rr,plugin_settingsr6r r5rVr __setattr____get__) r8rAr?r:r< gen_schemasigr*get_core_schemaschemaerdrYr'rWr)complete_dataclassQs|          rq_cls"TypeGuard[type[StandardDataclass]]cCs2t|ot|d ot|jtt|diS)a>Returns True if a class is a stdlib dataclass and *not* a pydantic dataclass. We check that - `_cls` is a dataclass - `_cls` does not inherit from a processed pydantic dataclass (and thus have a `__pydantic_validator__`) - `_cls` does not have any annotations that are not dataclass fields e.g. ```py import dataclasses import pydantic.dataclasses @dataclasses.dataclass class A: x: int @pydantic.dataclasses.dataclass class B(A): y: int ``` In this case, when we first check `B`, we make an extra check and look at the annotations ('y'), which won't be a superset of all the dataclass fields (only the stdlib fields i.e. 'x') Args: cls: The class. Returns: `True` if the class is a stdlib dataclass, `False` otherwise. r6r.) dataclassesrFrZsetr issupersetra)rrr'r'r)is_builtin_dataclasss  rwr&)r8r9r:r;r$r%) r8r@rArBr?rCr:r;r$rC)rrr@r$rs)5r7 __future__r _annotationsrtrgr[ functoolsrrrrr pydantic_corerr r r typing_extensionsr errorsrr=rplugin._schema_validatorrrrrr_fieldsr_generate_schemar _genericsr _mock_val_serr_schema_generation_sharedr _signaturer TYPE_CHECKINGconfigrProtocolrr/r]r>rqrwr'r'r'r)s:              n