U e<@sddlmZddlmZmZmZmZddlmZm Z ddl m Z ddl m Z mZerlddlmZddlmZed e eZGd d d eeZdd ddddddZddddddddZdS)) annotations) TYPE_CHECKINGCallableGenericTypeVar)SchemaSerializerSchemaValidator)Literal)PydanticErrorCodesPydanticUserError)PydanticDataclass) BaseModelValSerc@sNeZdZdZdZddddddd d d d Zdd d ddZddddZdS) MockValSerzMocker for `pydantic_core.SchemaValidator` or `pydantic_core.SchemaSerializer` which optionally attempts to rebuild the thing it's mocking when one of its methods is accessed and raises an error if that fails. )_error_message_code _val_or_ser_attempt_rebuildN)attempt_rebuildstrr z$Literal[('validator', 'serializer')]z"Callable[[], ValSer | None] | NoneNone) error_messagecode val_or_serrreturncCs(||_|dkrtnt|_||_||_dS)N validator)rrrrrr)selfrrrrrD/tmp/pip-unpacked-wheel-6q_yhcy8/pydantic/_internal/_mock_val_ser.py__init__szMockValSer.__init__)itemrcCsDd}|jr$|}|dk r$t||St|j|t|j|jddS)NTr)rgetattrrr rr)rr!Z__tracebackhide__val_serrrr __getattr__&s  zMockValSer.__getattr__z ValSer | NonercCs.|jr*|}|dk r|St|j|jddS)Nr")rr rr)rr$rrrrebuild1s zMockValSer.rebuild)__name__ __module__ __qualname____doc__ __slots__r r%r'rrrrrs   rall referenced typesztype[BaseModel]rr)clscls_nameundefined_namercsdd|d|d|d}ddfdd }t|d d |d _d dfd d }t|d d|d _dS)a&Set `__pydantic_validator__` and `__pydantic_serializer__` to `MockValSer`s on a model. Args: cls: The model class to set the mocks on cls_name: Name of the model class, used in error messages undefined_name: Name of the undefined thing, used in error messages `*` is not fully defined; you should define z , then call `z.model_rebuild()`.SchemaValidator | Noner&cs jddddk rjSdSdSNF) raise_errorsZ_parent_namespace_depth) model_rebuild__pydantic_validator__rr.rrattempt_rebuild_validatorHsz2set_model_mocks..attempt_rebuild_validatorclass-not-fully-definedrrrrSchemaSerializer | Nonecs jddddk rjSdSdSr4)r7__pydantic_serializer__rr9rrattempt_rebuild_serializerUsz3set_model_mocks..attempt_rebuild_serializer serializerN)rr8r>r.r/r0Zundefined_type_error_messager:r?rr9rset_model_mocks;s rBztype[PydanticDataclass]cstddlmd|d|d|d}ddfd d }t|d d |d _ddfdd }t|d d |d _dS)a*Set `__pydantic_validator__` and `__pydantic_serializer__` to `MockValSer`s on a dataclass. Args: cls: The model class to set the mocks on cls_name: Name of the model class, used in error messages undefined_name: Name of the undefined thing, used in error messages r )rebuild_dataclassr1r2z4, then call `pydantic.dataclasses.rebuild_dataclass(z)`.r3r&cs ddddk rjSdSdSr4)r8rr.rCrrr:tsz6set_dataclass_mocks..attempt_rebuild_validatorr;rr<r=cs ddddk rjSdSdSr4)r>rrDrrr?sz7set_dataclass_mocks..attempt_rebuild_serializerN) dataclassesrCrr8r>rArrDrset_dataclass_mockscs" rFN)r-)r-) __future__rtypingrrrrZ pydantic_corerrZtyping_extensionsr errorsr r rEr mainrrrrBrFrrrrs     ))