a hA@sdZddlmZddlZddlZddlZddlmZm Z m Z m Z m Z m Z mZmZddlmZddlmZmZddlmZmZmZmZdd lmZdd lmZdd lmZdd l m!Z!dd l"m#Z#m$Z$m%Z%erddl&m'Z'ddl(m)Z)dZ*edZ+ej,dkreej-e#e%fdeddddddddddd dddddddddddd ddZ.eej-e#e%fdeddddddddddd d ddddddddddd!d" d#dZ.neej-e#e%fdeddddddddd$dddddddddd% d&dZ.eej-e#e%fdeddddddddd$d ddddddddd!d' d(dZ.eej-e#e%fddAddddddddddd d)ddddddddddd*d" d+dZ.d!dd,d-d.Z/ee0Z1ej,d/krd0d0d1d2d3d4Z2e2ej3_4ddd5dd6d!ddd7d8dd9d:d;Z5dd?d@Z6dS)Bz7Provide an enhanced dataclass that performs validation.) annotationsN) TYPE_CHECKINGAnyCallableGenericLiteralNoReturnTypeVaroverload)warn) TypeGuarddataclass_transform)_config _decorators_namespace_utils _typing_extra) _dataclasses)getattr_migration) ConfigDict)PydanticUserError)Field FieldInfo PrivateAttr)PydanticDataclass)MappingNamespace) dataclassrebuild_dataclass_T )Zfield_specifiersFT. initrepreqorder unsafe_hashfrozenconfigvalidate_on_initkw_onlyslotszLiteral[False]boolz ConfigDict | type[object] | Nonez bool | Nonez-Callable[[type[_T]], type[PydanticDataclass]]) r#r$r%r&r'r(r)r*r+r,returnc CsdSNr"r0r0P/var/www/html/assistant/venv/lib/python3.9/site-packages/pydantic/dataclasses.pyrsrztype[_T]type[PydanticDataclass]) _clsr#r$r%r&r'r(r)r*r+r,r.c CsdSr/r0) r3r#r$r%r&r'r(r)r*r+r,r0r0r1r/sr#r$r%r&r'r(r)r*) r#r$r%r&r'r(r)r*r.cCsdSr/r0r4r0r0r1rBs ) r3r#r$r%r&r'r(r)r*r.c CsdSr/r0) r3r#r$r%r&r'r(r)r*r0r0r1rQs ztype[_T] | NonezGCallable[[type[_T]], type[PydanticDataclass]] | type[PydanticDataclass]c s|dusJd|dus Jdtjdkr6| | dnidddd d dd dfd d } |durx| S| |S)a!!! abstract "Usage Documentation" [`dataclasses`](../concepts/dataclasses.md) A decorator used to create a Pydantic-enhanced dataclass, similar to the standard Python `dataclass`, but with added validation. This function should be used similarly to `dataclasses.dataclass`. Args: _cls: The target `dataclass`. init: Included for signature compatibility with `dataclasses.dataclass`, and is passed through to `dataclasses.dataclass` when appropriate. If specified, must be set to `False`, as pydantic inserts its own `__init__` function. repr: A boolean indicating whether to include the field in the `__repr__` output. eq: Determines if a `__eq__` method should be generated for the class. order: Determines if comparison magic methods should be generated, such as `__lt__`, but not `__eq__`. unsafe_hash: Determines if a `__hash__` method should be included in the class, as in `dataclasses.dataclass`. frozen: Determines if the generated class should be a 'frozen' `dataclass`, which does not allow its attributes to be modified after it has been initialized. If not set, the value from the provided `config` argument will be used (and will default to `False` otherwise). config: The Pydantic config to use for the `dataclass`. validate_on_init: A deprecated parameter included for backwards compatibility; in V2, all Pydantic dataclasses are validated on init. kw_only: Determines if `__init__` method parameters must be specified by keyword only. Defaults to `False`. slots: Determines if the generated class should be a 'slots' `dataclass`, which does not allow the addition of new attributes after instantiation. Returns: A decorator that accepts a class as its argument and returns a Pydantic `dataclass`. Raises: AssertionError: Raised if `init` is not `False` or `validate_on_init` is `False`. Fz7pydantic.dataclasses.dataclass only supports init=Falsez-validate_on_init=False is no longer supportedr)r+r, type[Any]Noneclsr.c Ss|jD]}t|di}|D]}t||d}t|ts6qd|i}tjdkrV|jrVd|d<|jdurj|j|d<t||t j fi||j dduri|_ |||j |<qqdS)aMake sure that stdlib `dataclasses` understands `Field` kwargs like `kw_only` To do that, we simply change `x: int = pydantic.Field(..., kw_only=True)` into `x: int = dataclasses.field(default=pydantic.Field(..., kw_only=True), kw_only=True)` __annotations__NdefaultrTr+r$)__mro__getattr isinstancersys version_infor+r$setattr dataclassesfield__dict__getr9)r8Zannotation_clsr field_nameZ field_valueZ field_argsr0r0r1make_pydantic_fields_compatibles      z2dataclass..make_pydantic_fields_compatibler2c sddlm}||r*td|jddd|}tdd|jD}|spd urpt|d rptd |jd td dd ur|n t |d d }t |}t j |}|j}t|rd }|f}t|trt|j} || f}t|j|}|d ur} |jr(td|jdtd dn |jp&d} tj|fd| d}d|_||_||_|j|_|j|_tt|_ d|_!tj"||dd|S)zCreate a Pydantic dataclass from a regular dataclass. Args: cls: The class to create the Pydantic dataclass from. Returns: A Pydantic dataclass. r)is_model_classz(Cannot create a Pydantic dataclass from z" as it is already a Pydantic modelzdataclass-on-model)codecss|]}t|VqdSr/)rA is_dataclass).0baser0r0r1 z6dataclass..create_dataclass..N__pydantic_config__z[`config` is set via both the `dataclass` decorator and `__pydantic_config__` for dataclass zK. The `config` specification from `dataclass` decorator will take priority.)category stacklevelzN`frozen` is set via both the `dataclass` decorator and `config` for dataclass zW.This is not recommended. The `frozen` specification on `dataclass` will take priority.FT)r#r$r%r&r'r() raise_errors)#Z_internal._utilsrGr__name__any __bases__hasattrr UserWarningr<r ConfigWrapperrZDecoratorInfosbuild__doc___pydantic_dataclassesZis_builtin_dataclass issubclassr__parameters__types new_classr(rAr__is_pydantic_dataclass__Z__pydantic_decorators__ __module__ __qualname__ classmethod_pydantic_fields_completeZ__pydantic_fields_complete____pydantic_complete__complete_dataclass) r8rGZ original_clsZhas_dataclass_baseZ config_dictZconfig_wrapperZ decoratorsZ original_docbasesZ generic_baseZfrozen_r)r%r(kwargsrFr&r$r'r0r1create_dataclasssp              z#dataclass..create_dataclassN)r>r?) r3r#r$r%r&r'r(r)r*r+r,rjr0rhr1ras/  !"br7cCstdd|jDS)zReturn whether the fields where successfully collected (i.e. type hints were successfully resolves). This is a private property, not meant to be used outside Pydantic. css|] }|jVqdSr/)Z _complete)rJZ field_infor0r0r1rL#rMz,_pydantic_fields_complete..)allZ__pydantic_fields__values)r8r0r0r1rdsrd)r rr)argsrir.cOs tddS)a9This function does nothing but raise an error that is as similar as possible to what you'd get if you were to try calling `InitVar[int]()` without this monkeypatch. The whole purpose is just to ensure typing._type_check does not error if the type hint evaluates to `InitVar[]`. z 'InitVar' object is not callableN) TypeError)rnrir0r0r1 _call_initvar,srprO)forcerR_parent_namespace_depth_types_namespaceintzMappingNamespace | None)r8rqrRrrrsr.cCs|s|jrdSdD]}||jvrt||qd|_|dur@|}n |dkr\tj|ddpXi}ni}tj|d}tj|t j |j dd||dd S) axTry to rebuild the pydantic-core schema for the dataclass. This may be necessary when one of the annotations is a ForwardRef which could not be resolved during the initial attempt to build the schema, and automatic rebuilding fails. This is analogous to `BaseModel.model_rebuild`. Args: cls: The class to rebuild the pydantic-core schema for. force: Whether to force the rebuilding of the schema, defaults to `False`. raise_errors: Whether to raise errors, defaults to `True`. _parent_namespace_depth: The depth level of the parent namespace, defaults to 2. _types_namespace: The types namespace, defaults to `None`. Returns: Returns `None` if the schema is already "complete" and rebuilding was not required. If rebuilding _was_ required, returns `True` if rebuilding was successful, otherwise `False`. N)Z__pydantic_core_schema__Z__pydantic_validator__Z__pydantic_serializer__FrT)Z parent_depthrq)Zparent_namespace)check)rR ns_resolverZ _force_build) rerCdelattrrZparent_frame_namespacerZ NsResolverr[rfrrXrN)r8rqrRrrrsattrZ rebuild_nsrvr0r0r1r6s*   rr5z"TypeGuard[type[PydanticDataclass]])class_r.cCs0zd|jvot|WSty*YdS0dS)zWhether a class is a pydantic dataclass. Args: class_: The class. Returns: `True` if the class is a pydantic dataclass, `False` otherwise. r`FN)rCrArIAttributeError)ryr0r0r1is_pydantic_dataclassss  r{)N)7rZ __future__rZ _annotationsrAr>r^typingrrrrrrr r warningsr Ztyping_extensionsr r _internalrrrrrr[Z _migrationrr)rerrorsrfieldsrrrZ_internal._dataclassesrZ_internal._namespace_utilsr__all__rr?rBrrdrS __getattr__rpZInitVar__call__rr{r0r0r0r1s (        *.& (*= =