o iK@sUdZddlmZddlZddlZddlZddlZddlmZddlm Z ddl m Z m Z m Z mZmZmZddlZddlmZddlZddlmZd d lmZmZd d lmZe rodd lmZdd lmZmZddlm Z m!Z!ej"#dZ$e$%dZ&GdddeddZ'GdddeddZ(GdddZ)dNddZ*eGdd d Z+e+Z,d!Z-d"e.d#<e/d$d%e-DZ0d"e.d&<da1d'e.d(<dOd*d+Z2dPd.d/Z3dQd3d4Z4e d5d6Z5e d7d8Z6ed9Z7ed:Z8dRd?d@Z9dSdDdEZ:dTdHdIZ;dUdLdMZd2d3Zd4S)?_ValidateWrapperz2Decorator factory for one schema validator method.)validation_method schema_name_record_logfirer)?Literal['validate_python', 'validate_json', 'validate_strings']schemar_configCoreConfig | None_plugin_settingsPluginSettings | dict[str, Any]schema_type_pathrr'r&returnNonecCsZ||_t||_||_tj|_|did}|r+t|t r"|g}|jj ||_dSdS)Nrr%) r)get_schema_namer*r+rDEFAULT_LOGFIRE_INSTANCEr,get isinstancestr with_tags)selfr)r.r/r1r3r'r%r r r!__init__Os  z_ValidateWrapper.__init__ validatorrcsddlmjdkrtdfd d }|Sjd kr0tdfd d }|Sjd ks7Jtdfdd }|S)zEDecorator which wraps a schema validator method with instrumentation.rrall input_datarargskwargsr4c stjs|g|Ri|S|}z |g|Ri|}Wn1y9}z ||jddd}~wtyQ}z ||jddd}~ww||jdd|SNF)successT)r _initialized _on_enter_on_error_span_count_validation Exception_on_exception_span _on_success)r@rArBspanresulterror exceptionrr<r>r r!wrapped_validatorqs&       z4_ValidateWrapper.__call__..wrapped_validatorfailurec stjs|g|Ri|Sz |g|Ri|}Wn/y3}z jdd|d}~wtyJ}z jdd|d}~wwjdd|SrC)rrErH _on_error_logrI_on_exception_log)r@rArBrMrNrOrPr r!rQs"     metricscsdtjs|g|Ri|Sz |g|Ri|}Wnty)jddwjdd|SrC)rrErIrH)r@rArBrM)r<r>r r!rQs   N)r@rrArrBrr4r)pydanticrr+ functoolswraps)r<r>rQr rPr!__call__js  D $z_ValidateWrapper.__call__r@cCs(|jjd|j|j|dd|jdS)Nz*Pydantic {schema_name} {validation_method}infoz pydantic.)r*r)r@_level _span_name)r,rLr*r) __enter__)r<r@r r r!rFs z_ValidateWrapper._on_enterrLr rMcCs$|j|dd|d|ddddS)NT succeeded)rDstatusrM)_set_span_attributes__exit__)r<rLrMr r r!rKsz_ValidateWrapper._on_successrNrcCs,|jjdd|j||jdddddS)Nwarnz"Validation on {schema_name} failedF include_url)r* error_counterrors)level msg_template attributes)r,logr*rerf)r<rNr r r!rSs  z_ValidateWrapper._on_error_logcCs<|j|dd||jddd|d|ddddS)NFfailedrc)rDr_rerfrb)r`rerf set_levelra)r<rLrNr r r!rGs  z_ValidateWrapper._on_error_spanrOrIcCs&|jjdd|jt|jd|ddS)NrNz3Validation on {schema_name} raised {exception_type})r*exception_type)rgrhriexc_info)r,rjr*typer)r<rOr r r!rTs z"_ValidateWrapper._on_exception_logcCs4|j|ddt|jd|t|||jdS)NFzraised )rDr_)r`rorra __traceback__)r<rLrOr r r!rJs z#_ValidateWrapper._on_exception_spanr_r:rDboolricKs4|r|d|i||jd|7_dSdS)NrD ) is_recordingset_attributesmessage)r<rLr_rDrir r r!r`sz%_ValidateWrapper._set_span_attributescCstd||j|jddS)N)rDr*r))validation_counteraddr*r))r<rDr r r!rHsz"_ValidateWrapper._count_validationN)r)r-r.rr/r0r1r2r3rr'r&r4r5)r>rr4r)r@r)rLr rMr)rNr)rLr rNr)rOrI)rLr rOrI) rLr r_r:rDrqrirr4r5)rDrqr4r5)rrrr __slots__r=rYrFrKrSrGrTrJr`rHr r r r!r(Es   N r(r.rr4r:cs|ddvr |djS|ddvrt|dS|ddkr?|d}|ddkr;|dfd d |dD\}t|St|S|dS) aFind the best name to use for a schema. The follow rules are used: * If the schema represents a model or dataclass, use the name of the class. * If the root schema is a wrap/before/after validator, look at its `schema` property. * Otherwise use the schema's `type` property. Args: schema: The schema to get the name for. Returns: The name of the schema. ro>modelrcls> function-wrapfunction-afterfunction-beforer. definitionszdefinition-ref schema_refcsg|] }|dkr|qS)refr ).0 definitionrr r! s  z#get_schema_name..)rr6)r. inner_schemaschema_definitionr rr!r6s       r6c@sJeZdZdZeejedksej ddkrdd d Z dSddd Z dS)LogfirePydanticPluginaImplements a new API for pydantic plugins. Patches Pydantic to accept this new API shape. Set the `LOGFIRE_PYDANTIC_RECORD` environment variable to `"off"` to disable the plugin, or `PYDANTIC_DISABLE_PLUGINS` to `true` to disable all Pydantic plugins. z2.5.0LOGFIRE_PYDANTIC_RECORDoff_r__r4/tuple[_ValidateWrapper, ...] | tuple[None, ...]cOsdS)zBackwards compatibility for Pydantic < 2.5.0. This method is called every time a new `SchemaValidator` is created, and is a NO-OP for Pydantic < 2.5.0. NNNr )r<rrr r r!new_schema_validator-sz*LogfirePydanticPlugin.new_schema_validatorr.r schema_typer3r schema_kindrconfigr0plugin_settingsdict[str, Any]c Cst|d}|rd|vr|d}ntj}|dkrdSt|r>ttd|||||td|||||td|||||fSdS)a.This method is called every time a new `SchemaValidator` is created. Args: schema: The schema to validate against. schema_type: The original type which the schema was created from, e.g. the model class. schema_type_path: Path defining where `schema_type` was defined, or where `TypeAdapter` was called. schema_kind: The kind of schema to validate against. config: The config to use for validation. plugin_settings: The plugin settings. Returns: A tuple of decorator factories for each of the three validation methods - `validate_python`, `validate_json`, `validate_strings` or a tuple of three `None` if recording is `off`. rr'rrvalidate_python validate_jsonvalidate_strings)_patch_PluggableSchemaValidatorr8get_pydantic_plugin_configr'_include_model_patch_build_wrapperr() r<r.rr3rrrlogfire_settingsr'r r r!r7s   N)rrrrr4r)r.rrrr3rrrrr0rrr4r) rrrrrrV __version__osenvironr8rr r r r!r!s " r)fastapilogfire_backendfastuiztuple[str, ...]IGNORED_MODULESccs|]}|dVqdS).Nr )rmoduler r r! jsrIGNORED_MODULE_PREFIXESPydanticPlugin | None_pydantic_plugin_config_valuercCstdurtStjjS)zGet the Pydantic plugin config.N)rr param_managerpydantic_pluginr r r r!rosr plugin_configr5cCs|adS)zSet the pydantic plugin config.N)r)rr r r!set_pydantic_plugin_configwsrr3rrqcsnt}|j}|j}jtstvrdS|r'tfdd|Dr'dS|r5tfdd|DSdS)z-Check whether a model should be instrumented.Fc3.|]}t|ddjVqdS$z::Nresearchnamerpatternrr3r r!r,z!_include_model..c3rrrrrr r!rrT)rincludeexcluder startswithrrany)r3rrrr rr!r}srcCsddlm}t|_dS)a%The old pydantic plugin API required managing state between event handler methods. This was messy, especially in a way that handles concurrency and nested validation: see https://pydantic.slack.com/archives/C05AF4A4WRM/p1710503400257589. The 'new API' simply requires decorating the validation methods, returning a new wrapper function. At the time of writing, this API doesn't actually exist yet in pydantic. We patch it here so that this will also work for older versions of pydantic without needing to support both APIs. r_schema_validatorN)pydantic.pluginr_build_wrapper build_wrapperrr r r!rs rcCs<ddlm}t|jdkrd d d }||_d Sd S) zPatch a 'bug' in PluggableSchemaValidator. Getting an attribute before proper initializing (e.g. when using cloudpickle) leads to infinite recursion trying to get _schema_validator. r)PluggableSchemaValidatorzg def __getattr__(self, name: str) -> Any: return getattr(self._schema_validator, name) r<rrr:r4cSs|dkrt|t|j|S)Nr)AttributeErrorgetattrr)r<rr r r! __getattr__s z4_patch_PluggableSchemaValidator..__getattr__N)r<rrr:r4r)!pydantic.plugin._schema_validatorrinspect getsourcerstrip)rrr r r!rs   rPRfuncCallable[P, R]event_handlers list[Any]cCs2|D]}t||}|r|}qt|r||}q|SN)_wrap_with_old_handlercallable)rrhandler old_wrappedr r r!rs rrrrcOsdSrr )rrr r r!_noopsrrCallable[P, R] | Nonecsddlmt|dt|dt|dt|dur/ur/ur/tur0dStdfd d }|S)Nrron_enter on_successon_error on_exceptionrAP.argsrBP.kwargsr4rc sp|i|z |i|}Wn!y!}z|d}~wty1}z|d}~ww||Sr)rI)rArBrMrNrOrrrrrrr r!wrappersz'_wrap_with_old_handler..wrapper)rArrBrr4r)rVr_get_handler_methodrrWrX)rrrr rr!rs     &r method_nameCallable[..., None]cCs*t||d}|dur tS|jdkrtS|S)Nzpydantic.plugin)rrr)rrr r r!rs  r)r.rr4r:)r4r)rrr4r5)r3rr4rq)rrrrr4r)rrrr)rrrrr4r)rrrr:r4r)=r __future__rrWrrr dataclassesrrtypingrrrrr r rVtyping_extensionsr rr _internal.configrr_internal.utilsrrrrr pydantic_corerr_meter_provider get_meterMETERcreate_counterrwrrr(r6rpluginrrtuplerrrrrrrrrrrrrr r r r!sZ           ;"D