B hnd*@sdZddlmZddlZddlZddlZddlmZm Z m Z m Z m Z m Z mZddlmZmZddlmZmZmZddlmZdd lmZdd lmZdd lmZerdd lmZd Z e dZ!ej"dkrheej#efdeddddddddddd dddddddddddd ddZ$eej#efdeddddddddddd ddddddddddddd d dZ$neej#efdeddddddddd!dddddddddd" d#dZ$eej#efdeddddddddd!ddddddddddd$ d%dZ$eej#efdd7ddddddddddd d&ddddddddddd'd d(dZ$ee%Z&d)ej"krfd*krnnd+d+d,d-d.d/Z'e'ej(_)ddd0dd1dddd2d3dd4d5d6Z*dS)8z7Provide an enhanced dataclass that performs validation.) annotationsN) TYPE_CHECKINGAnyCallableGenericNoReturnTypeVaroverload)Literaldataclass_transform)_config _decorators _typing_extra) _dataclasses)getattr_migration) ConfigDict)Field)PydanticDataclass) dataclassrebuild_dataclass_T) )Zfield_specifiersFT.) initrepreqorder unsafe_hashfrozenconfigvalidate_on_initkw_onlyslotszLiteral[False]boolz ConfigDict | type[object] | Nonez bool | Nonez-Callable[[type[_T]], type[PydanticDataclass]]) rrrrrrr r!r"r#returnc CsdS)N) rrrrrrr r!r"r#r&r&8/tmp/pip-unpacked-wheel-y3jxpnai/pydantic/dataclasses.pyrsrztype[_T]ztype[PydanticDataclass]) _clsrrrrrrr r!r"r#r%c CsdS)Nr&) r(rrrrrrr r!r"r#r&r&r'r+s)rrrrrrr r!) rrrrrrr r!r%cCsdS)Nr&)rrrrrrr r!r&r&r'r?s ) r(rrrrrrr r!r%c CsdS)Nr&) r(rrrrrrr r!r&r&r'rNsztype[_T] | NonezGCallable[[type[_T]], type[PydanticDataclass]] | type[PydanticDataclass]c srdkstd|dk s tdtjdkr8t| | dnidddfd d } |d krj| S| |S) aA decorator used to create a Pydantic-enhanced dataclass, similar to the standard Python `dataclasses`, 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 supported)rr)r"r#z type[Any]ztype[PydanticDataclass])clsr%c s|}}|dkr(t|dd}|dk r(|}t|}tj|}|j}t|rd}|f}t |t rvt |j }||f}t |j|}tj|fd}||_||_|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__)rrrrrrF) raise_errorstypes_namespace)getattrr ConfigWrapperrZDecoratorInfosbuild__doc___pydantic_dataclassesZis_builtin_dataclass issubclassr__parameters__types new_class__name__ dataclassesrZ__pydantic_decorators__ __module__ __qualname__complete_dataclass__pydantic_complete__) r)Z original_clsZ config_dictZ cls_configZconfig_wrapperZ decoratorsZ original_docbasesZ generic_baseZpydantic_complete)r rrrkwargsrrrr&r'create_dataclasssB        z#dataclass..create_dataclassN)AssertionErrorsys version_infodict) r(rrrrrrr r!r"r#r>r&)r rrrr=rrrr'r_s, "<)r)r rr)argsr=r%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)rEr=r&r&r' _call_initvarsrG)forcer+_parent_namespace_depth_types_namespaceintzdict[str, Any] | None)r)rIr+rJrKr%cCsp|s|jrdS|dk r |}n.|dkr>tj|dp6i}|}ni}t||}tj|tj|j dd||dSdS)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)Z parent_depthF)check)r+r,) r;copyrZparent_frame_namespaceZget_cls_types_namespacer1r:r r.r*)r)rIr+rJrKr,Zframe_parent_nsr&r&r'rs   r)N)+r0 __future__rZ _annotationsr7r@r4typingrrrrrrr typing_extensionsr r _internalr rrrr1Z _migrationrr rfieldsrZ_internal._dataclassesr__all__rrAfieldrr6 __getattr__rGZInitVar__call__rr&r&r&r's $      ,0(*,i