U e!!@sdZddlmZddlZddlZddlZddlmZm 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+r8ddl,m-Z-Gdddej.Z/Gddde/ej.Z0neZ1d,ddddddZ2d d!d"d#d$dd$d%d&d'Z3d"d(d)d*d+Z4dS)-z0Private 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@s8eZdZUded<ded<ded<dddd d d Zd S) StandardDataclasszClassVar[dict[str, Any]]__dataclass_fields__z ClassVar[Any]Z__dataclass_params__z$ClassVar[Callable[(Ellipsis, None)]]Z __post_init__objectNone)argskwargsreturncOsdSN)selfr!r"r%r%C/tmp/pip-unpacked-wheel-6q_yhcy8/pydantic/_internal/_dataclasses.py__init__&szStandardDataclass.__init__N)__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]Z__pydantic_complete__z ClassVar[core_schema.CoreSchema]__pydantic_core_schema__z$ClassVar[_decorators.DecoratorInfos]Z__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-ztype[StandardDataclass]zdict[str, Any] | Noner )clstypes_namespacer#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)rrr0)r4r5r6fieldsr%r%r'set_dataclass_fieldsDsr8T) raise_errorsz type[Any]z_config.ConfigWrapperbool)r4config_wrapperr9r5r#c st|drtdt|dkr(t|}t||t|}t|||}t |j |j |dd}ddddd d d }|j d |_ ||_ |j |_||_t|d d}z6|r||tt|jdd|dd} n|j|dd} WnHtk r} z(|rt||jd| jdWYdSd} ~ XYnX||} z|| } Wn(|jk r\t||jdYdSXtd|}| |_t| ||j|j d| |j |_!t"| | |_#|j$rt%|j&dddddfdd } | '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. Z__post_init_post_parse__zVSupport for `__post_init_post_parse__` has been dropped, the method will not be calledNT)initr7r; is_dataclassr-rr )__dataclass_self__r!r"r#c_s"d}|}|jjt|||ddS)NT)Z self_instance)r2Zvalidate_pythonr)r>r!r"Z__tracebackhide__sr%r%r'r(sz$complete_dataclass..__init__z .__init__Z__get_pydantic_core_schema__F)Zfrom_dunder_get_core_schemaunpack)Zref_mode`zall referenced typesztype[PydanticDataclass]Z dataclassstr)instance__field__valuer#cs|||dSr$)validate_assignment)rCrDrEZ validatorr%r'validated_setattrsz-complete_dataclass..validated_setattr)(hasattrwarningswarnDeprecationWarningrZget_cls_types_namespacer8rrrr(r0r+Z config_dictr. __signature__getattrrrZgenerate_schemarrr)name core_configZ clean_schemaZCollectedInvalidtypingcastr/rr*Zplugin_settingsr2r r1rFr __setattr____get__) r4r;r9r5r6Z gen_schemasigr(Zget_core_schemaZschemaerPrHr%rGr'complete_dataclassQs          rWz"TypeGuard[type[StandardDataclass]])_clsr#cCs2t|o0t|d o0t|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. r2r,) dataclassesr=rIsetr issupersetrN)rXr%r%r'is_builtin_dataclasss   r\)N)5r3 __future__rZ _annotationsrYrQrJ functoolsrrrrrZ pydantic_corerr r r Ztyping_extensionsr errorsrr7rZplugin._schema_validatorrrrrr_fieldsrZ_generate_schemarZ _genericsrZ _mock_val_serrZ_schema_generation_sharedr _signaturer TYPE_CHECKINGconfigrProtocolrr-rLr8rWr\r%r%r%r's8             n