o h0 @s0dZddlmZddlZddlZddlZddlmZm Z m Z m Z m Z m Z mZddlmZmZmZddlmZmZmZddlmZdd lmZdd lmZdd lmZmZer\dd l m!Z!d Z"e dZ#ej$dkreej%efdeddddddddddd dId%d&Z&eej%efdeddddddddddd dJd*d&Z&n2eej%efdeddddddddd+dKd,d&Z&eej%efdeddddddddd+dLd-d&Z&eej%efd dMddddddddddd dNd0d&Z&ee'Z(d1ej$krd2krnn dOd7d8Z)e)ej*_+ddd9dd:dPdBdCZ,dQdGdHZ-dS)Rz7Provide an enhanced dataclass that performs validation.) annotationsN) TYPE_CHECKINGAnyCallableGenericNoReturnTypeVaroverload)Literal TypeGuarddataclass_transform)_config _decorators _typing_extra) _dataclasses)getattr_migration) ConfigDict)Field FieldInfo)PydanticDataclass) dataclassrebuild_dataclass_T )field_specifiersFT. initrepreqorder unsafe_hashfrozenconfigvalidate_on_initkw_onlyslotsrLiteral[False]r boolr!r"r#r$r% ConfigDict | type[object] | Noner& bool | Noner'r(return-Callable[[type[_T]], type[PydanticDataclass]]c CdSNrr1r1J/var/www/html/ai/venv/lib/python3.10/site-packages/pydantic/dataclasses.pyrsr_clstype[_T]type[PydanticDataclass]c Cr/r0r1) r3rr r!r"r#r$r%r&r'r(r1r1r2r+srr r!r"r#r$r%r&cCr/r0r1r6r1r1r2r?s c Cr/r0r1) r3rr r!r"r#r$r%r&r1r1r2rNstype[_T] | NoneGCallable[[type[_T]], type[PydanticDataclass]] | type[PydanticDataclass]c s|dusJd|dusJdtjdkr!t| | ddd d nidd d dfdd } |dur<| S| |S)a2Usage docs: https://docs.pydantic.dev/2.5/concepts/dataclasses/ 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 or not to include the field in the `__repr__` output. eq: Determines if a `__eq__` 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 an unsafe hashing function should be included in the class. frozen: Determines if the generated class should be a 'frozen' `dataclass`, which does not allow its attributes to be modified from its constructor. config: A configuration for the `dataclass` generation. 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(cls type[Any]r-Nonec SsX|jD]&}zt||}Wn tyYqwt|tr)|jr)t||tj|ddqdS)a3Make 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)` T)defaultr'N) __annotations__getattrAttributeError isinstancerr'setattr dataclassesfield)r9 field_name field_valuer1r1r2make_pydantic_fields_compatibles  z2dataclass..make_pydantic_fields_compatiblecSr/r0r1)_r1r1r2rFsr5c s|}}|durt|dd}|dur|}t|}tj|}|j}t|rBd}|f}t |t r;t |j }||f}t |j|}|tj|fdd}||_||_|j|_|j|_tj||ddd} | |_|S)zCreate a Pydantic dataclass from a regular dataclass. Args: cls: The class to create the Pydantic dataclass from. Returns: A Pydantic dataclass. N__pydantic_config__T)rr r!r"r#r$F raise_errorstypes_namespace)r>r ConfigWrapperrDecoratorInfosbuild__doc___pydantic_dataclassesis_builtin_dataclass issubclassr__parameters__types new_class__name__rBr__pydantic_decorators__ __module__ __qualname__complete_dataclass__pydantic_complete__) r9 original_cls config_dict cls_configconfig_wrapper decorators original_docbases generic_basepydantic_completer%r!r$kwargsrFr"r r#r1r2create_dataclasssL          z#dataclass..create_dataclassN)r9r:r-r;)r-r;)r9r:r-r5)sys version_infodict) r3rr r!r"r#r$r%r&r'r(rgr1rer2r_s.    ?)r)r argsrrfrcOstd)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 callable) TypeError)rmrfr1r1r2 _call_initvarsro)forcerJ_parent_namespace_depth_types_namespacer9rqrJrrintrsdict[str, Any] | NonecCsl|s|jrdS|dur|}n|dkrtj|dpi}|}ni}t||}tj|tj|j dd||dS)a}Try 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 build the dataclass core schema for. force: Whether to force the rebuilding of the model 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`. Nr) parent_depthF)checkrI) r[copyrparent_frame_namespaceget_cls_types_namespacerPrZrrLrH)r9rqrJrrrsrKframe_parent_nsr1r1r2rs   r__clsr:"TypeGuard[type[PydanticDataclass]]cCst|o d|jvS)zWhether a class is a pydantic dataclass. Args: __cls: The class. Returns: `True` if the class is a pydantic dataclass, `False` otherwise. __pydantic_validator__)rB is_dataclass__dict__)r|r1r1r2is_pydantic_dataclass/s r)rr)r r*r!r*r"r*r#r*r$r*r%r+r&r,r'r*r(r*r-r.)r3r4rr)r r*r!r*r"r*r#r*r$r*r%r+r&r,r'r*r(r*r-r5)rr)r r*r!r*r"r*r#r*r$r*r%r+r&r,r-r.)r3r4rr)r r*r!r*r"r*r#r*r$r*r%r+r&r,r-r5r0)r3r7rr)r r*r!r*r"r*r#r*r$r*r%r+r&r,r'r*r(r*r-r8)rmrrfrr-r) r9r5rqr*rJr*rrrtrsrur-r,)r|r:r-r}).rO __future__r _annotationsrBrhrTtypingrrrrrrr typing_extensionsr r r _internalrrrrrP _migrationrr%rfieldsrr_internal._dataclassesr__all__rrirCrrV __getattr__roInitVar__call__rrr1r1r1r2s $       1