U #iV9@sdZddlmZddlZddlZddlZddlmZm Z m Z m Z m Z m Z mZddlmZddlmZmZmZddlmZmZmZdd lmZdd lmZdd lmZdd lm Z dd l!m"Z"m#Z#m$Z$erddl%m&Z&dZ'e dZ(ej)dkreej*e"e$fdeddddddddddd dddddddddddd ddZ+eej*e"e$fdeddddddddddd dddddddddddd d! d"dZ+neej*e"e$fdeddddddddd#dddddddddd$ d%dZ+eej*e"e$fdeddddddddd#dddddddddd d& d'dZ+eej*e"e$fdd>ddddddddddd d(ddddddddddd)d! d*dZ+ee,Z-d+ej)krd,krnnd-d-d.d/d0d1Z.e.ej/_0ddd2dd3d ddd4d5dd6d7d8Z1d9d:d;d.make_pydantic_fields_compatibler1c s8ddlm}||r*td|jddd|}dk r:n t|dd}t|}tj |}|j }t |rd}|f}t |trt|j}||f}t|j|}|dk rֈ} |jrtd |jd td d n |jpd } tj|fd| d}||_||_ |j|_|j|_d |_t j||d 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)codeN__pydantic_config__zN`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.)category stacklevelFT)r"r#r$r%r&r' raise_errorstypes_namespace)Z_internal._utilsrEr__name__r:r ConfigWrapperrZDecoratorInfosbuild__doc___pydantic_dataclassesZis_builtin_dataclass issubclassr__parameters__types new_classr'r UserWarningr?rZ__pydantic_decorators__ __module__ __qualname____pydantic_complete__complete_dataclass) r6rEZ original_clsZ config_dictZconfig_wrapperZ decoratorsZ original_docbases generic_baseZfrozen_r(r$r'kwargsrDr%r#r&r/r0create_dataclasss^            z#dataclass..create_dataclassN)AssertionErrorr<r=) r2r"r#r$r%r&r'r(r)r*r+r`r/r^r0r`s.  #"O)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)rdr_r/r/r0 _call_initvarsrfrH)forcerL_parent_namespace_depth_types_namespaceintzdict[str, Any] | None)r6rgrLrhrir-cCsp|s|jrdS|dk r |}n.|dkr>tj|dp6i}|}ni}t||}tj|tj|j dd||dSdS)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`. Nr)Z parent_depthF)checkrK) rZcopyrZparent_frame_namespaceZmerge_cls_and_parent_nsrRr[rrOrG)r6rgrLrhrirMZframe_parent_nsr/r/r0rs   rr4z"TypeGuard[type[PydanticDataclass]])class_r-cCs2zd|jkot|WStk r,YdSXdS)zWhether a class is a pydantic dataclass. Args: class_: The class. Returns: `True` if the class is a pydantic dataclass, `False` otherwise. Z__pydantic_validator__FN)rAr? is_dataclassAttributeError)rmr/r/r0is_pydantic_dataclassLs rp)N)3rQ __future__rZ _annotationsr?r<rUtypingrrrrrrr warningsr typing_extensionsr r r _internalrrrrrRZ _migrationrr(rerrorsrfieldsrrrZ_internal._dataclassesr__all__rr=r@rrN __getattr__rfInitVar__call__rrpr/r/r/r0s $       *.& (*+1