o $6dV*@sdZddlmZddlZddlZddlZddlZddlm Z m Z ddlm Z m Z m Z ddlmZmZmZddlmZmZmZmZmZddlmZd d lmZd d lmZd d lmZd dlmZm Z m!Z!m"Z"d dl#m$Z$m%Z%m&Z&d dl'm(Z(d dl)m*Z*d dl+m,Z,d dl-m.Z.d dl/m0Z0ej1rd dl2m3Z3Gdddej4Z5Gddde5ej4Z6neZ7d5d6d d!Z8d"d#d7d)d*Z9d8d,d-Z:d9d0d1Z;d:d3d4Zr@fieldsr-r-r/set_dataclass_fields@s rBT) raise_errors type[Any]config_wrapper_config.ConfigWrapperrCboolc st|dr tdt|durt|}t||t|}t|||}dd d }|j d |_ t |}||_ ||_ |j |_t|dd}z|rV||tt|jdd|dd} n|j|dd} Wn"ty} z|rit||jd| jdWYd} ~ dSd} ~ ww||} || } t| } t| rt||jddStd|}tt| |_} t | } t!| | |_"t#| | |_$|j%rt&|j'dfdd } | (d||_'dS)aFinish building a pydantic dataclass. This logic is called on a class which has already been wrapped in `dataclasses.dataclass()`. This is somewhat analogous to `pydantic._internal._model_construction.complete_model_class`. Args: cls: The class. config_wrapper: The config wrapper instance. raise_errors: Whether to raise errors, defaults to `True`. types_namespace: The types namespace. Returns: `True` if building a pydantic dataclass is successfully completed, `False` otherwise. Raises: PydanticUndefinedAnnotation: If `raise_error` is `True` and there is an undefined annotations. Z__post_init_post_parse__zVSupport for `__post_init_post_parse__` has been dropped, the method will not be calledN__dataclass_self__r5r'rr)r*r+c_s"d}|}|jjt|||ddS)NT)Z self_instance)r:Zvalidate_pythonr )rHr'r)Z__tracebackhide__sr-r-r/r0ysz$complete_dataclass..__init__z .__init__Z__get_pydantic_core_schema__F)Zfrom_dunder_get_core_schemaunpack)Zref_mode`zall referenced typesztype[PydanticDataclass]instance__fieldstr__valuecs|||dSr,)validate_assignment)rLrMrO validatorr-r/validated_setattrsz-complete_dataclass..validated_setattrT)rHr5r'rr)rr*r+)rLrrMrNrOrNr*r+))hasattrwarningswarnDeprecationWarningrZget_cls_types_namespacerBrrr3generate_dataclass_signaturer0 __signature__Z config_dictr6getattrr!rZgenerate_schemarr r1name core_configZcollect_definitionsrrtypingcastrZapply_discriminatorsr7rrr:r r9rPr __setattr____get__)r<rErCr>r@Z gen_schemar0sigZget_core_schemaZschemaer\Zsimplified_core_schemarSr-rQr/complete_dataclassMsn           rcrcCst|}i}|jD]O}|j}t|trU|j}|dkrt}|j}|j dur0t|j t r0|j }n t|j t r9|j }|j}|t urM|j t urJtjj}ntj}|j|||d}|||j<q tt|ddS)aGenerate signature for a pydantic dataclass. This implementation assumes we do not support custom `__init__`, which is currently true for pydantic dataclasses. If we change this eventually, we should make this function's logic more closely mirror that from `pydantic._internal._model_construction.generate_model_signature`. Args: cls: The dataclass. Returns: The signature. rN) annotationr[default) parametersreturn_annotation)rrfvaluesre isinstancerrdrr[Zvalidation_aliasaliasrNr default_factoryinspectrempty dataclasses_HAS_DEFAULT_FACTORYreplacelist)r<raZ final_paramsparamZ param_defaultrdr[rer-r-r/rXs*      rX_cls"TypeGuard[type[StandardDataclass]]cCs2t|ot|d ot|jtt|diS)a>Returns True if a class is a stdlib dataclass and *not* a pydantic dataclass. We check that - `_cls` is a dataclass - `_cls` does not inherit from a processed pydantic dataclass (and thus have a `__pydantic_validator__`) - `_cls` does not have any annotations that are not dataclass fields e.g. ```py import dataclasses import pydantic.dataclasses @dataclasses.dataclass class A: x: int @pydantic.dataclasses.dataclass class B(A): y: int ``` In this case, when we first check `B`, we make an extra check and look at the annotations ('y'), which won't be a superset of all the dataclass fields (only the stdlib fields i.e. 'x') Args: cls: The class. Returns: `True` if the class is a stdlib dataclass, `False` otherwise. r:r4)rn is_dataclassrTsetr$ issupersetrZrsr-r-r/is_builtin_dataclasss  ry"TypeGuard[type[PydanticDataclass]]cCst|o d|jvS)zWhether a class is a pydantic dataclass. Args: cls: The class. Returns: `True` if the class is a pydantic dataclass, `False` otherwise. r:)rnru__dict__rxr-r-r/is_pydantic_dataclasss r|r,)r<r=r>r?r*r+) r<rDrErFrCrGr>r?r*rG)r<r=r*r)rsrDr*rt)rsrDr*rz)=r; __future__rZ _annotationsrnrlr]rU functoolsrrrrrrr r Z pydantic_corer r r rrZtyping_extensionsrerrorsrrArrrrrrZ _core_utilsrrr_fieldsrZ_generate_schemarZ _genericsrZ _mock_val_serr Z_schema_generation_sharedr! TYPE_CHECKINGconfigr"Protocolr#r5rWrBrcrXryr|r-r-r-r/s@             f 0%