a ~gx @sUdZddlmZddlZddlmZmZmZmZm Z ddl m Z m Z m Z mZddlmZmZerzddlmZmZmZmZed Ze d Zed ZeejZd ed <d$dddddddddddZGdddZddddddZ ddd d!d"d#Z!dS)%z(Pluggable schema validator for pydantic.) annotationsN) TYPE_CHECKINGAnyCallableIterableTypeVar) CoreConfig CoreSchemaSchemaValidatorValidationError)Literal 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|||||||| |pcz build_wrapper..css|]}t|dr|jVqdS) on_successN)rGrMrHr%r%r&rKdrLcss|]}t|dr|jVqdS)on_errorN)rGrNrHr%r%r&rKerLcss|]}t|dr|jVqdS) on_exceptionN)rGrOrHr%r%r&rKfrLzP.argszP.kwargsr)argskwargsrc sD]}||i|qz|i|}Wnntyb}z D] }||q>WYd}~nVd}~0ty}z D] }||qtWYd}~n d}~00D] }||q|SdSr>)r Exception) rPrQZon_enter_handlerresulterrorZon_error_handler exceptionZon_exception_handlerZon_success_handlerrDZ on_entersZ on_errorsZ on_exceptionsZ on_successesr%r&wrapperhs   zbuild_wrapper..wrapper)tuple functoolswraps)rDrErWr%rVr&r5_s r5rbool) 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?r1)r\r]handlerr%r%r&rGs   rG)NN)"rB __future__rrYtypingrrrrr pydantic_corerr r r typing_extensionsr r r!rrrrrrEventlist__args__r__annotations__r'r#r5rGr%r%r%r&s"   *