U ={f4@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_compatibler-c s|}}|dkr(t|dd}|dk r(|}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)rrr 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__) r2Z original_clsZ config_dictZ cls_configZconfig_wrapperZ decoratorsZ original_docbasesZ generic_baseZpydantic_completer$r r#kwargsr@r!rr"r+r,create_dataclasssR          z#dataclass..create_dataclassN)AssertionErrorr8r9dict) r.rrr r!r"r#r$r%r&r'rUr+rSr,r_s. #"?)r)r rr)argsrTr)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)rZrTr+r+r, _call_initvarsr\)forcerC_parent_namespace_depth_types_namespaceintzdict[str, Any] | None)r2r^rCr_r`r)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)checkrB) rQcopyrZparent_frame_namespaceZget_cls_types_namespacerHrPrrErA)r2r^rCr_r`rDZframe_parent_nsr+r+r,r s   rr0z"TypeGuard[type[PydanticDataclass]])__clsr)cCst|od|jkS)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_dataclassr=)rdr+r+r,is_pydantic_dataclass>s rf)N).rG __future__rZ _annotationsr;r8rKtypingrrrrrrr Ztyping_extensionsr r r _internalrrrrrHZ _migrationrr$rfieldsrrZ_internal._dataclassesr__all__rr9r<rrM __getattr__r\InitVar__call__rrfr+r+r+r,s $     *.& (*1