a h#@sdZddlmZddlZddlZddlZddlmZm Z ddlm Z m Z ddl m Z mZmZmZddlmZdd lmZdd lmZmZdd lmZd d lmZmZd dlmZd dlmZm Z d dl!m"Z"d dl#m$Z$d dl%m&Z&d dl'm(Z(d dl)m*Z*ej+r>ddl,m-Z.ddl/m0Z0ddl1m2Z2Gddde.ej3Z4neZ5d.ddddddd Z6d!dd"d#d$d%d&dd&d&d'd(d)Z7d$d*d+d,d-Z8dS)/z0Private logic for creating pydantic dataclasses.) annotationsN)partialwraps)AnyClassVar) ArgsKwargsSchemaSerializerSchemaValidator core_schema) TypeGuard)PydanticUndefinedAnnotation)PluggableSchemaValidatorcreate_schema_validator)PydanticDeprecatedSince20)_config _decorators)collect_dataclass_fields)GenerateSchemaInvalidSchemaError)get_standard_typevars_map)set_dataclass_mocks) NsResolver)generate_pydantic_signature)LazyClassAttribute)DataclassInstance) ConfigDict) FieldInfoc@s\eZdZUdZded<ded<ded<ded <d ed <d ed <ded<eddddZdS)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]Z__pydantic_decorators__zClassVar[dict[str, FieldInfo]]__pydantic_fields__zClassVar[SchemaSerializer]__pydantic_serializer__z4ClassVar[SchemaValidator | PluggableSchemaValidator]__pydantic_validator__bool)returncCsdSN)clsr)r)[/var/www/html/assistant/venv/lib/python3.9/site-packages/pydantic/_internal/_dataclasses.py__pydantic_fields_complete__:sz.PydanticDataclass.__pydantic_fields_complete__N)__name__ __module__ __qualname____doc____annotations__ classmethodr,r)r)r)r+r%s  rztype[StandardDataclass]zNsResolver | Nonez_config.ConfigWrapper | NoneNone)r* ns_resolverconfig_wrapperr'cCs"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`. )r4 typevars_mapr5N)rrr#)r*r4r5r6fieldsr)r)r+set_dataclass_fieldsCs r8TF) raise_errorsr4 _force_buildz type[Any]z_config.ConfigWrapperr&)r*r5r9r4r:r'c s|j}ddddddd}|jd|_||_|j|_t|||d|sZ|jrZt|d St|d rpt d t t |}t |||d }t d tt||j|j|jdd|_z||} WnBty} z(|r΂t|d| jdWYd} ~ d Sd} ~ 00|j|jd} z|| } Wnty4t|Yd S0td|}| |_t| ||j|jd| |j |_!t"| | |_#|j$rt%|j&dddddfdd } | '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. rrr3)__dataclass_self__argskwargsr'c_s"d}|}|jjt|||ddS)NT)Z self_instance)r%Zvalidate_pythonr)r;r<r=Z__tracebackhide__sr)r)r+__init__xsz$complete_dataclass..__init__z .__init__)r5FZ__post_init_post_parse__zVSupport for `__post_init_post_parse__` has been dropped, the method will not be called)r4r6 __signature__T)initr7validate_by_nameextra is_dataclass`N)titleztype[PydanticDataclass]Z dataclassstr)instancefieldvaluer'cs|||dSr()validate_assignment)rHrIrJZ validatorr)r+validated_setattrsz-complete_dataclass..validated_setattr))r?r/Z config_dictr r8Z defer_buildrhasattrwarningswarnDeprecationWarningrrrrrr#rBrCr@Zgenerate_schemar name core_configr-Z clean_schemartypingcastr"rr.Zplugin_settingsr%rr$rKr __setattr____get__r!) r*r5r9r4r:Z original_initr?r6Z gen_schemaZschemaerSrMr)rLr+complete_dataclassWsn      rYz"TypeGuard[type[StandardDataclass]])_clsr'cCs2t|o0t|d o0t|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%r1) dataclassesrDrNsetZ__dataclass_fields__ issupersetgetattr)rZr)r)r+is_builtin_dataclasss   r_)NN)9r0 __future__rZ _annotationsr[rTrO functoolsrrrrZ pydantic_corerrr r Ztyping_extensionsr errorsr Zplugin._schema_validatorrrrrr_fieldsrZ_generate_schemarrZ _genericsrZ _mock_val_serrZ_namespace_utilsr _signaturer_utilsr TYPE_CHECKINGZ _typeshedrZStandardDataclassconfigrr7rProtocolrrQr8rYr_r)r)r)r+sB             u