h-i0 dZddlmZddlZddlZddlZddlmZm Z m Z m Z m Z m Z mZddlmZmZmZddlmZmZmZddlmZdd lmZdd lmZdd lmZmZerdd l m!Z!d Z"e dZ#ejHdk\reejJefeddddddddddd d"dZ&eejJefeddddddddddd d#dZ&nzeejJefeddddddddd d$dZ&eejJefeddddddddd d%dZ&eejJef d&ddddddddddd d'dZ&ee'Z(dejHcxkrdkrnnd(dZ)e)ejT_+ddddd d)d Z,d*d!Z-y)+z7Provide 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_onlyslotsc yNrs SD:\DhanshreeandTeamAI\Github\ai_code\venv\Lib\site-packages\pydantic/dataclasses.pyrrs c yr+r,) _clsr r!r"r#r$r%r&r'r(r)s r-rr+s r.r r!r"r#r$r%r&r'cyr+r,r1s r-rr?s r.cyr+r,) r0r r!r"r#r$r%r&r's r-rrNs r.c  |dusJd|dusJdtjdk\rt| |  d d ni d d d  fd } || 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)c |jD]O} t||}t|ts!|j s.t ||tj|dQy#t$rY^wxYw)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)cls field_name field_values r-make_pydantic_fields_compatiblez2dataclass..make_pydantic_fields_compatiblesg"11 ")#z":Kk95+:M:MC[->->{\`-ab2&s A  A,+A,cyr+r,)_s r-rAz2dataclass..make_pydantic_fields_compatiblesr.c f|} }|t|dd}||}tj|}tjj |}|j }tj|rNd}|f}t|trt|j}||fz}tj|j|}|tj |fd  d }||_||_|j$|_|j&|_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. N__pydantic_config__T)r r!r"r#r$r%F raise_errorstypes_namespace)r8r ConfigWrapperrDecoratorInfosbuild__doc___pydantic_dataclassesis_builtin_dataclass issubclassr__parameters__types new_class__name__r<r__pydantic_decorators__ __module__ __qualname__complete_dataclass__pydantic_complete__)r> original_cls config_dict cls_configconfig_wrapper decorators original_docbases generic_basepydantic_completer&r"r%kwargsrAr#r!r$s r-create_dataclassz#dataclass..create_dataclasssK    &;TBJ%(  ..{; //55c: {{ 5 5c : L FE#w'&s'9'9: ///#,,6C',##   #    '1#" %00'441DD eT %6! r.)r> type[Any]returnNone)rerf)r>rdretype[PydanticDataclass])sys version_infodict)r0r r!r"r#r$r%r&r'r(r)rcrbrAs `````` @@r-rr_s{\ 5=SSS= 5 (Y*YY ( 7"gU3 c" ==~ | D !!r.)r)r ctd)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)argsrbs r- _call_initvarrps :;;r.)forcerG_parent_namespace_depth_types_namespacec*|s |jry||j}n:|dkDrtj|xsi}|}ni}tj||}t j |tj|jd||S)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)checkrF) rXcopyrparent_frame_namespaceget_cls_types_namespacerMrWrrIrE)r>rrrGrsrtrHframe_parent_nss r-rrs4 S..  '5E5J5J5LO&*"/"F"FTk"l"rpr#2"$+CCCYO$77   ! !#"9"9 G%+   r.cLtj|xrd|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__)r< is_dataclass__dict__)__clss r-is_pydantic_dataclassr/s$  # #E * Y/G5>>/YYr.)r Literal[False]r!boolr"rr#rr$rr%rr& ConfigDict | type[object] | Noner' bool | Noner(rr)rre-Callable[[type[_T]], type[PydanticDataclass]])r0type[_T]r rr!rr"rr#rr$rr%rr&rr'rr(rr)rrerg)r rr!rr"rr#rr$rr%rr&rr'rrer)r0rr rr!rr"rr#rr$rr%rr&rr'rrergr+)r0ztype[_T] | Noner rr!rr"rr#rr$rr%rr&rr'rr(rr)rrezGCallable[[type[_T]], type[PydanticDataclass]] | type[PydanticDataclass])rorrbrrer) r>rgrrrrGrrsintrtzdict[str, Any] | Nonerer)rrdrez"TypeGuard[type[PydanticDataclass]]).rL __future__r _annotationsr<rhrQtypingrrrrrr r typing_extensionsr r r _internalrrrrrM _migrationrr&rfieldsrr_internal._dataclassesr__all__rrir=rrS __getattr__rpInitVar__call__rrr,r.r-rsu=2 UUUEE::<)$9 * T]w;+<+