o g%@szdZddlmZddlZddlZddlZddlmZm Z ddlm Z m Z ddl m Z mZmZmZddlmZdd lmZdd lmZmZdd lmZd d lmZmZd dlmZd dlmZd dl m!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/ddl0m1Z1ddl2m3Z3Gddde/ej4Z5neZ6  d3d4d#d$Z7d%dd&d'd5d-d.Z8d6d1d2Z9dS)7z0Private logic for creating pydantic dataclasses.) annotationsN)partialwraps)AnyClassVar) ArgsKwargsSchemaSerializerSchemaValidator core_schema) TypeGuard)PydanticUndefinedAnnotation)PluggableSchemaValidatorcreate_schema_validator)PydanticDeprecatedSince20)_config _decorators)collect_dataclass_fields)GenerateSchema)get_standard_typevars_map)set_dataclass_mocks) NsResolver)CallbackGetCoreSchemaHandler)generate_pydantic_signature)LazyClassAttribute)DataclassInstance) ConfigDict) FieldInfoc@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__z4ClassVar[SchemaValidator | PluggableSchemaValidator]__pydantic_validator__N)__name__ __module__ __qualname____doc____annotations__r,r,V/var/www/visachat/venv/lib/python3.10/site-packages/pydantic/_internal/_dataclasses.pyr&s   rclstype[StandardDataclass] ns_resolverNsResolver | Noneconfig_wrapper_config.ConfigWrapper | NonereturnNonecCs"t|}t||||d}||_dS)zCollect and set `cls.__pydantic_fields__`. Args: cls: The class. ns_resolver: Namespace resolver to use when getting dataclass annotations. config_wrapper: The config wrapper instance, defaults to `None`. )r0 typevars_mapr2N)rrr$)r.r0r2r6fieldsr,r,r-set_dataclass_fieldsAs   r8TF) raise_errorsr0 _force_build type[Any]_config.ConfigWrapperr9boolr:c s|j}d#dd }|jd |_||_|j|_t|||d |s*|jr*t||jd St|d r5t dt t |}t |||d}tdtt||j|j|jdd|_t|dd} z| rj| |tt|jd d|dd} n|j|d d} Wn"ty} z|r}t||jd| jdWYd} ~ d Sd} ~ ww|j|jd} z|| } Wn|jyt||jdYd Swtd|}| |_t | ||j!|jd| |j"|_#t$| | |_%|j&rt'|j(d$fd!d" } | )d||_(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`. ns_resolver: The namespace resolver instance to use when collecting dataclass fields and during schema building. _force_build: Whether to force building the dataclass, no matter if [`defer_build`][pydantic.config.ConfigDict.defer_build] is set. 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. __dataclass_self__rargsrkwargsr4r5c_s"d}|}|jjt|||ddS)NT) self_instance)r&validate_pythonr)r>r?r@__tracebackhide__sr,r,r-__init__vsz$complete_dataclass..__init__z .__init__)r2F__post_init_post_parse__zVSupport for `__post_init_post_parse__` has been dropped, the method will not be called)r0r6 __signature__T)initr7populate_by_nameextra is_dataclass__get_pydantic_core_schema__N)from_dunder_get_core_schemaunpack)ref_mode`)titlezall referenced typesztype[PydanticDataclass] dataclassinstancefieldstrvaluecs|||dS)N)validate_assignment)rSrTrV validatorr,r-validated_setattrsz-complete_dataclass..validated_setattr)r>rr?rr@rr4r5)rSrrTrUrVrUr4r5)+rEr) config_dictr r8 defer_buildrr'hasattrwarningswarnDeprecationWarningrrrrrr$rIrJrGgetattrrgenerate_schemar name core_config clean_schemaCollectedInvalidtypingcastr"rr(plugin_settingsr&rr%rWr __setattr____get__r!)r.r2r9r0r: original_initrEr6 gen_schemaget_core_schemaschemaerdrZr,rXr-complete_dataclassUs          rq_cls"TypeGuard[type[StandardDataclass]]cCs2t|ot|d ot|jtt|diS)aBReturns 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. ```python 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. r&r+) dataclassesrKr]set__dataclass_fields__ issupersetra)rrr,r,r-is_builtin_dataclasss  rx)NN)r.r/r0r1r2r3r4r5) r.r;r2r<r9r=r0r1r:r=r4r=)rrr;r4rs):r* __future__r _annotationsrtrgr^ functoolsrrrr pydantic_corerrr r typing_extensionsr errorsr plugin._schema_validatorrrrrr_fieldsr_generate_schemar _genericsr _mock_val_serr_namespace_utilsr_schema_generation_sharedr _signaturer_utilsr TYPE_CHECKING _typeshedrStandardDataclassconfigrr7rProtocolrr`r8rqrxr,r,r,r-sF