a y{Jh)@sdZddlmZddlZddlZddlZddlZddlm Z m Z ddlm Z m Z ddlm Z mZmZddlmZmZmZmZmZddlmZd d lmZd d lmZd d lmZd d lmZddlm Z m!Z!m"Z"ddl m#Z#ddl$m%Z%ddl&m'Z'm(Z(ddl)m*Z*ddl+m,Z,ddl-m.Z.ddl/m0Z0ej1rdd dl2m3Z3Gdddej4Z5Gddde5ej4Z6neZ7d8ddddd d!Z8d"d#d$d%d&dd&d'd(d)Z9d*d*d+d,d-Z:dd.d/d0d1d2d3Z;d$d4d5d6d7Z.__init__z .__init__Z__get_pydantic_core_schema__F)Zfrom_dunder_get_core_schemaunpack)Zref_mode`zall referenced typesztype[PydanticDataclass] dataclassstr)instance__field__valuer*cs|||dSr+)validate_assignment)rIrJrK validatorr,r.validated_setattrsz-complete_dataclass..validated_setattrT)(hasattrwarningswarnDeprecationWarningrZget_cls_types_namespacer?rrgenerate_dataclass_signaturer7r2r/Z config_dictr5 __signature__getattrrrZgenerate_schemarrr0name core_configZ clean_schemaZCollectedInvalidtypingcastr6rr1Zplugin_settingsr9r r8rLr __setattr____get__) r;rBr@r<r=Z gen_schemasigr/Zget_core_schemaZschemaerXrOr,rMr.complete_dataclassUsj          r_r)paramr*cCs|j}t|tr|j}|dkr"t}|j}|j}|j}|durTt|trTt |rT|}nt|trjt |rj|}|j}|t ur|j t urt j j}ntj}|j|||dS|S)zCustom processing where the parameter default is of type FieldInfo Args: param (Parameter): The parameter Returns: Parameter: The custom processed parameter rN) annotationrWdefault)rb isinstancerrarrWaliasvalidation_aliasrHr r default_factoryinspectrempty dataclasses_HAS_DEFAULT_FACTORYreplace)r`Z param_defaultrarWrdrerbr,r,r.process_param_defaultss&    rlzdict[str, FieldInfo]rr)r;r>rBr*cCst|j||tdS)zGenerate signature for a pydantic dataclass. Args: cls: The dataclass. fields: The model fields. config_wrapper: The config wrapper instance. Returns: The dataclass signature. )initr>rBZpost_process_parameter)rr/rl)r;r>rBr,r,r.rTs  rTz"TypeGuard[type[StandardDataclass]])_clsr*cCs2t|o0t|d o0t|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. r9r3)ri is_dataclassrPsetr# issupersetrV)rnr,r,r.is_builtin_dataclasss   rr)N)=r: __future__rZ _annotationsrirgrYrQ functoolsrrrrrrr Z pydantic_corer r r r rtyping_extensionsrerrorsrr>rZplugin._schema_validatorrrrrrr_fieldsrZ_generate_schemarrZ _genericsrZ _mock_val_serrZ_schema_generation_sharedrZ_utilsr TYPE_CHECKINGconfigr!Protocolr"r4rSr?r_rlrTrrr,r,r,r.sB             i'