a y{Jh0@sdZddlmZddlZddlZddlZddlmZm Z m Z m Z m Z m Z mZddlmZmZmZddlmZmZmZddlmZdd lmZdd lmZdd lmZmZerdd l m!Z!d Z"e dZ#ej$dkrpeej%efdeddddddddddd dddddddddddd ddZ&eej%efdeddddddddddd ddddddddddddd d dZ&neej%efdeddddddddd!dddddddddd" d#dZ&eej%efdeddddddddd!ddddddddddd$ d%dZ&eej%efdd.make_pydantic_fields_compatible)r*cSsdSr+r,)_r,r,r-r=sr.c s|}}|dur(t|dd}|dur(|}t|}tj|}|j}t|rd}|f}t |t rvt |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)r6r ConfigWrapperrZDecoratorInfosbuild__doc___pydantic_dataclassesZis_builtin_dataclass issubclassr__parameters__types new_class__name__r:rZ__pydantic_decorators__ __module__ __qualname__complete_dataclass__pydantic_complete__) r3Z original_clsZ config_dictZ cls_configZconfig_wrapperZ decoratorsZ original_docbases generic_baseZpydantic_completer%r!r$kwargsr=r"r r#r,r-create_dataclasssL          z#dataclass..create_dataclassN)sys version_infodict) r/rr r!r"r#r$r%r&r'r(rTr,rRr-r_s.  "?)r)r rr)argsrSr*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)rZrSr,r,r- _call_initvarsr\)forcerA_parent_namespace_depth_types_namespaceintzdict[str, Any] | None)r3r^rAr_r`r*cCsp|s|jrdS|dur |}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)checkr@) rOcopyrZparent_frame_namespaceZget_cls_types_namespacerFrNrrCr?)r3r^rAr_r`rBZframe_parent_nsr,r,r-rs   rr1z"TypeGuard[type[PydanticDataclass]])__clsr*cCst|od|jvS)zWhether a class is a pydantic dataclass. Args: __cls: The class. Returns: `True` if the class is a pydantic dataclass, `False` otherwise. Z__pydantic_validator__)r: is_dataclass__dict__)rdr,r,r-is_pydantic_dataclass/s rg)N).rE __future__rZ _annotationsr:rUrItypingrrrrrrr typing_extensionsr r r _internalrrrrrFZ _migrationrr%rfieldsrrZ_internal._dataclassesr__all__rrVr;rrK __getattr__r\InitVar__call__rrgr,r,r,r-s $     *.& (*1