a h @sUdZddlmZddlZddlmZddlmZmZm Z m Z m Z ddl m Z mZmZmZddlmZerdd lmZmZmZmZed Ze d Ze d ZeejZd ed<d%dddddddddddZGdddZ ddddddZ!d dd!d"d#d$Z"dS)&z(Pluggable schema validator for pydantic.) annotationsN)Iterable) TYPE_CHECKINGAnyCallableLiteralTypeVar) CoreConfig CoreSchemaSchemaValidatorValidationError) ParamSpec)BaseValidateHandlerProtocolPydanticPluginProtocol SchemaKindSchemaTypePathPR)Zon_validate_pythonZon_validate_jsonZon_validate_stringsz list[Event]eventsr rstrrCoreConfig | Nonezdict[str, Any] | Nonez*SchemaValidator | PluggableSchemaValidator)schema schema_typeschema_type_moduleschema_type_name schema_kindconfigplugin_settingsreturnc CsNddlm}ddlm}|} | r@t|||||||| |pdz build_wrapper..css|]}t|dr|jVqdS) on_successN)rDrJrEr$r$r%rHerIcss|]}t|dr|jVqdS)on_errorN)rDrKrEr$r$r%rHfrIcss|]}t|dr|jVqdS) on_exceptionN)rDrLrEr$r$r%rHgrIzP.argszP.kwargsr)argskwargsrc sD]}||i|qz|i|}Wnntyb}z D] }||q>WYd}~nVd}~0ty}z D] }||qtWYd}~n d}~00D] }||q|SdSr;)r Exception) rMrNZon_enter_handlerresulterrorZon_error_handler exceptionZon_exception_handlerZon_success_handlerrAZ on_entersZ on_errorsZ on_exceptionsZ on_successesr$r%wrapperis   zbuild_wrapper..wrapper)tuple functoolswraps)rArBrTr$rSr%r3`s r3rbool) handler_cls method_namercCs.t||d}|durdS|jdkr&dSdSdS)zFilter out handler methods which are not implemented by the plugin directly - e.g. are missing or are inherited from the protocol. NFzpydantic.pluginT)r<r/)rYrZhandlerr$r$r%rDs   rD)NN)#r? __future__rrVcollections.abcrtypingrrrrrZ pydantic_corer r r r Ztyping_extensionsr r!rrrrrrEventlist__args__r__annotations__r&r"r3rDr$r$r$r%s$     *