U ={fl @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] | Noner )schema schema_typeschema_type_moduleschema_type_name schema_kindconfigplugin_settingsreturnc CsNddlm}ddlm}|} | r@t|||||||| |pr$r$r$r%r"4s!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% bs z build_wrapper..css|]}t|dr|jVqdS) on_successN)rErJrFr$r$r%rIcs css|]}t|dr|jVqdS)on_errorN)rErKrFr$r$r%rIds css|]}t|dr|jVqdS) on_exceptionN)rErLrFr$r$r%rIes zP.argszP.kwargsr)argskwargsrc sD]}|||qz||}Wnjtk rX}zD] }||q8W5d}~XYnLtk r}zD] }||qlW5d}~XYnXD] }||q|SdSr<)r Exception) rMrNZon_enter_handlerresulterrorZon_error_handler exceptionZon_exception_handlerZon_success_handlerrBZ on_entersZ on_errorsZ on_exceptionsZ on_successesr$r%wrappergs    zbuild_wrapper..wrapper)tuple functoolswraps)rBrCrTr$rSr%r3^s r3rbool) handler_cls method_namercCs.t||d}|dkrdS|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%rE~s   rE)NN)"r@ __future__rrVtypingrrrrrZ pydantic_corerr r r Ztyping_extensionsr r r!rrrrrrEventlist__args__r__annotations__r&r"r3rEr$r$r$r%s"   *