a Ofx @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|||||||| |pr$r$r$r%r"5s!r"zCallable[P, R]z!list[BaseValidateHandlerProtocol])funcevent_handlersrcs|sStdd|Dtdd|Dtdd|Dtdd|Dtdddd fd d }|SdS) Ncss|]}t|dr|jVqdS)on_enterN)filter_handlersrD.0hr$r$r% cz build_wrapper..css|]}t|dr|jVqdS) on_successN)rErKrFr$r$r%rIdrJcss|]}t|dr|jVqdS)on_errorN)rErLrFr$r$r%rIerJcss|]}t|dr|jVqdS) on_exceptionN)rErMrFr$r$r%rIfrJzP.argszP.kwargsr)argskwargsrc sD]}||i|qz|i|}Wnntyb}z D] }||q>WYd}~nVd}~0ty}z D] }||qtWYd}~n d}~00D] }||q|SdSr<)r Exception) rNrOZon_enter_handlerresulterrorZon_error_handler exceptionZon_exception_handlerZon_success_handlerrBZ on_entersZ on_errorsZ on_exceptionsZ on_successesr$r%wrapperhs   zbuild_wrapper..wrapper)tuple functoolswraps)rBrCrUr$rTr%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/)rZr[handlerr$r$r%rEs   rE)NN)"r@ __future__rrWtypingrrrrrZ pydantic_corerr r r typing_extensionsr r r!rrrrrrEventlist__args__r__annotations__r&r"r3rEr$r$r$r%s"   *