o hl@sUdZddlmZddlZddlmZmZmZmZm Z ddl m Z m Z m Z mZddlmZmZer=ddlmZmZmZmZed Ze d Zed ZeejZd ed <  d-d.ddZGdd d Zd/d%d&Z d0d+d,Z!dS)1z(Pluggable schema validator for pydantic.) annotationsN) TYPE_CHECKINGAnyCallableIterableTypeVar) CoreConfig CoreSchemaSchemaValidatorValidationError)Literal ParamSpec)BaseValidateHandlerProtocolPydanticPluginProtocol SchemaKindSchemaTypePathPR)on_validate_pythonon_validate_jsonon_validate_stringsz list[Event]eventsschemar schema_typerschema_type_modulestrschema_type_name schema_kindrconfigCoreConfig | Noneplugin_settingsdict[str, Any] | Nonereturnr c CsJddlm}ddlm}|} | r t|||||||| |piSt||S)zCreate a `SchemaValidator` or `PluggableSchemaValidator` if plugins are installed. Returns: If plugins are installed then return `PluggableSchemaValidator`, otherwise return `SchemaValidator`. r)r) get_plugins)r_loaderr$PluggableSchemaValidatorr ) rrrrrrr!rr$pluginsr)W/var/www/html/ai/venv/lib/python3.10/site-packages/pydantic/plugin/_schema_validator.pycreate_schema_validators   r+c@s(eZdZdZdZdddZdddZdS)r'zPluggable schema validator.)_schema_validator validate_jsonvalidate_pythonvalidate_stringsrr rrschema_type_pathrrrrr r( Iterable[PydanticPluginProtocol]r!dict[str, Any]r#Nonec Cst|||_g}g} g} |D]J} z| ||||||\} } }Wnty<}ztd| jd| jjd||d}~ww| durF|| | durO| | |durX| |qt|jj ||_ t|jj | |_ t|jj | |_ dS)NzError using plugin `:z`: ) r r,new_schema_validator TypeError __module__ __class____name__append build_wrapperr.r-r/)selfrrr0rrr(r!python_event_handlersjson_event_handlersstrings_event_handlerspluginpjser)r)r*__init__9s.  "   z!PluggableSchemaValidator.__init__namercCs t|j|SN)getattrr,)r<rFr)r)r* __getattr__Zs z$PluggableSchemaValidator.__getattr__N)rr rrr0rrrrr r(r1r!r2r#r3)rFrr#r)r9r7 __qualname____doc__ __slots__rErIr)r)r)r*r'4s  !r'funcCallable[P, R]event_handlers!list[BaseValidateHandlerProtocol]cst|sStdd|Dtdd|Dtdd|Dtdd|Dtdfd d }|S)Ncs |] }t|dr|jVqdS)on_enterN)filter_handlersrR.0hr)r)r* bz build_wrapper..csrQ) on_successN)rSrYrTr)r)r*rWcrXcsrQ)on_errorN)rSrZrTr)r)r*rWdrXcsrQ) on_exceptionN)rSr[rTr)r)r*rWerXargsP.argskwargsP.kwargsr#rc sD] }||i|qz |i|}Wn+ty+}z D]}||qd}~wty@}z D]}||q4d}~wwD]}||qC|SrG)r Exception) r\r^on_enter_handlerresulterroron_error_handler exceptionon_exception_handleron_success_handlerrM on_enters on_errors on_exceptions on_successesr)r*wrappergs$   zbuild_wrapper..wrapper)r\r]r^r_r#r)tuple functoolswraps)rMrOrmr)rhr*r;^sr; handler_clsr method_nameboolcCs*t||d}|dur dS|jdkrdSdS)zFilter out handler methods which are not implemented by the plugin directly - e.g. are missing or are inherited from the protocol. NFzpydantic.pluginT)rHr7)rqrrhandlerr)r)r*rS~s  rS)NN)rr rrrrrrrrrr r!r"r#r )rMrNrOrPr#rN)rqrrrrr#rs)"rK __future__rrotypingrrrrr pydantic_corerr r r typing_extensionsr r r%rrrrrrEventlist__args__r__annotations__r+r'r;rSr)r)r)r*s$    *