o h@sdddlZddlZddlZddlZddlZddlZddlmZddlm Z ddl m Z ddl m Z mZddlZddlZddlmZmZmZmZddlmZmZmZmZmZmZmZmZmZm Z m!Z!m"Z"m#Z#ddl$m%Z%e%rdd l$m&Z&dd l$m'Z'dd l$m(Z(dd l$m)Z)dd l$m*Z*ddl+m,Z,e-dZ.dZ/GdddeZ0ddZ1ej2  dKddZ3ddZ4ddZ5ddZ6dd Z7d!d"Z8d#d$Z9Gd%d&d&Z:Gd'd(d(Z;d)d*Zfd-d.Z?dLd/d0Z@dLd1d2ZAd3d4d5d6ZBd7d8ZCd9d:ZDdLd;d<ZEd=d>ZFd?d@ZGdAdBZHdCdDZIdEdFZJdGdHZKddIlLmMZMmNZNmOZOe%r0ddJlLmPZPdSdS)MN)Mapping) timedelta)Random)quoteunquote)OPSPANDATA SPANSTATUS SPANTEMPLATE) capture_internal_exceptionsfilename_for_moduleDsnloggermatch_regex_listqualname_from_function safe_repr to_string try_convert is_sentry_url_is_external_source_is_in_project_root_module_in_list) TYPE_CHECKING)Any)Dict) Generator)Optional)Union) FrameTypez5^[ ]*([0-9a-f]{32})?-?([0-9a-f]{16})?-?([01])?[ ]*$z(([a-zA-Z0-9+/]{4})*([a-zA-Z0-9+/]{2,3})?c@s0eZdZ d ddZddZddZdd Zd S) EnvironHeadersHTTP_cCs||_||_dSN)environprefix)selfr"r#r%Z/var/www/html/pro-man-master/venv/lib/python3.10/site-packages/sentry_sdk/tracing_utils.py__init__As zEnvironHeaders.__init__cCs|j|j|ddSN-_)r"r#replaceupper)r$keyr%r%r& __getitem__JzEnvironHeaders.__getitem__cCstddt|DS)Ncss|]}dVqdS)Nr%).0r*r%r%r& Psz)EnvironHeaders.__len__..)sumiterr$r%r%r&__len__NszEnvironHeaders.__len__ccsP|jD]!}t|ts q|dd}||jsq|t|jdVqdSr()r" isinstancestrr+r, startswithr#len)r$kr%r%r&__iter__Rs   zEnvironHeaders.__iter__N)r )__name__ __module__ __qualname__r'r.r6r<r%r%r%r&r@s    rcCs:|durdSt|dduo|ddup|dduS)z} Returns True if either traces_sample_rate or traces_sampler is defined and enable_tracing is set and not false. NFenable_tracingtraces_sample_ratetraces_sampler)boolget)optionsr%r%r&has_tracing_enabled_s rFFmanualc cs tjdddr|r|dgkrd}|dkrd}nd}d}t||}i}|dur/||d<|dur7||d<|r=d|d <|rG|durG||d <ttj|d |d Wdn1s]wYtjtj ||d }| D] \} } | | | qp|VWddS1swYdS)N _experimentsrecord_sql_paramsFpyformatformatz db.paramsz db.paramstyleTzdb.executemanyz db.cursorquery)messagecategorydata)opnameorigin) sentry_sdk get_clientrErD _format_sqlr add_breadcrumb start_spanrDBitemsset_data) cursorrL params_list paramstyle executemanyrecord_cursor_repr span_originrOspanr;vr%r%r&record_sql_queriesqs@   "rccCs|jtjkr|j|jdd|jddS|jtjkrZd}|jt j }|rAd|kr.dkr3nnd}nd|kr=dkrAnnd}|rO|jd d |j|d dS|jd d |jd dS|jd krl|jd d |j|jddSdS)Nredis)rMtyperNrOiiWerroriiwarninghttphttplib)rerNrOlevel)rerNrO subprocess)rerNrMrO) rPrDB_REDISrV description_tags HTTP_CLIENT_datarDrHTTP_STATUS_CODE)scoperarj status_coder%r%r&"maybe_create_breadcrumbs_from_spans2       rtcCs z|jjWStyYdSwr!)f_code co_filename Exception)framer%r%r&_get_frame_module_abs_paths   rycCs6t||}t|p t||}| o|pt||o| Sr!)rrr)is_sentry_sdk_frame namespacein_app_includein_app_excludeabs_path project_rootshould_be_includedshould_be_excludedr%r%r&_should_be_includeds rcCsDt}|s dS|jdus|jdurdS|jdd}|s!dS|j|j}|jdd}|tdd|k}|s;dS|jd}|jd }|jd }t } | durt | } z| j d } Wn t ykd} Ynw| duot| d } t| | ||| |d } | rn | j} | dusTd} | durz| j}Wn t yd}Ynw|dur|tj| jz| j d } Wn t yd} Ynw| dur|tj| t | }|dur| durt| |}n|dur| |r||dtj}n|}|tj|z| jj}Wn t y d}Ynw|dur |tj| jjdSdSdS)zB Adds OTel compatible source code information to the span Nenable_db_query_sourceTdb_query_source_threshold_msrr0) millisecondsrr|r}r=z sentry_sdk.)rzr{r|r}r~r)rSrT is_active timestampstart_timestamprErDrsys _getframery f_globalsrwr9rf_backf_linenorZr CODE_LINENOCODE_NAMESPACEr r+lstripossep CODE_FILEPATHruco_name CODE_FUNCTION)raclientshould_add_query_sourceduration threshold slow_queryrr|r}rxr~r{rzrlinenofilepath in_app_path code_functionr%r%r&add_query_sources             rcCs|sdS|dr|dr|dd}t|}|sdS|\}}}d}|r0dt|d}|r:dt|d}|r@|d k}|||d S) zL Given a `sentry-trace` header string, return a dictionary of data. Nz00-z-00z{:032x}z{:016x}0)trace_idparent_span_idparent_sampled)r9endswithSENTRY_TRACE_REGEXmatchgroupsrKint)headerrrr sampled_strrr%r%r&extract_sentrytrace_data8s&  rcCsXd}zt|dr||}t|tr||jj}Wn ty%d}Ynw|p+t|S)Nmogrify) hasattrrr7bytesdecode connectionencodingrwr)r[sqlreal_sqlr%r%r&rUXs     rUc@seZdZdZdZ     dddZeddZedd Z e j d d Z ed d Z e j d d Z ddZ ddZ ddZddZdS)PropagationContextzJ The PropagationContext represents the data of a trace in Sentry.  _trace_id_span_idrrdynamic_sampling_contextNcCs*||_ ||_ ||_ ||_ ||_dSr!r)r$rspan_idrrrr%r%r&r'ys zPropagationContext.__init__cCszd}t|}|t}|rt}t||_|t}|r3t|}|dur3|dur.t}| ||dur;| |Sr!) normalize_incoming_datarDBAGGAGE_HEADER_NAMErBaggagefrom_incoming_headerrSENTRY_TRACE_HEADER_NAMErupdate_fill_sample_rand)cls incoming_datapropagation_contextnormalized_databaggage_headersentry_trace_headersentrytrace_datar%r%r&from_incoming_datas(   z%PropagationContext.from_incoming_datacCs|js tj|_|jS)z!The trace id of the Sentry trace.)ruuiduuid4hexr5r%r%r&rs zPropagationContext.trace_idcC ||_dSr!)rr$valuer%r%r&r cCs |js tjdd|_|jS)z+The span id of the currently executed span.rN)rrrrr5r%r%r&rszPropagationContext.span_idcCrr!)rrr%r%r&rrc Cs8|D]\}}zt|||WqtyYqwdS)zU Updates the PropagationContext with data from the given dictionary. N)rYsetattrAttributeError)r$ other_dictr-rr%r%r&rs zPropagationContext.updatecCsd|j|j|j|j|jS)Nzm)rKrrrrrr5r%r%r&__repr__szPropagationContext.__repr__c Cs|jdurdStt|jd}|dur d|krdkr dStt|jd}t|j|\}}z t|j||fd}WntySt d|jd|d YdSw|d |jd<dS) aJ Ensure that there is a valid sample_rand value in the dynamic_sampling_context. If there is a valid sample_rand value in the dynamic_sampling_context, we keep it. Otherwise, we generate a sample_rand value according to the following: - If we have a parent_sampled value and a sample_rate in the DSC, we compute a sample_rand value randomly in the range: - [0, sample_rate) if parent_sampled is True, - or, in the range [sample_rate, 1) if parent_sampled is False. - If either parent_sampled or sample_rate is missing, we generate a random value in the range [0, 1). The sample_rand is deterministically generated from the trace_id, if present. This function does nothing if there is no dynamic_sampling_context. N sample_randrr0 sample_rateintervalz5Could not backfill sample_rand, since parent_sampled=z and sample_rate=..6f) rrfloatrD_sample_rand_ranger_generate_sample_randr ValueErrorrdebug)r$rrlowerr,r%r%r&rs0      z$PropagationContext._fill_sample_randcCs|jdurdS|jdS)zRConvenience method to get the sample_rand value from the dynamic_sampling_context.Nr)rrDr5r%r%r& _sample_rands  zPropagationContext._sample_rand)NNNNN)r=r>r?__doc__ __slots__r' classmethodrpropertyrsetterrrrrrr%r%r%r&rls.         6rc@seZdZdZdZdZedZ  dddZ e d d d d Z e d dZ e ddZ ddZddZdddZeddZddZddZd S) raF The W3C Baggage header information (see https://www.w3.org/TR/baggage/). Before mutating a `Baggage` object, calling code must check that `mutable` is `True`. Mutating a `Baggage` object that has `mutable` set to `False` is not allowed, but it is the caller's responsibility to enforce this restriction.  sentry_itemsthird_party_itemsmutablezsentry-z^sentry-rTcCs||_||_||_dSr!r)r$rrrr%r%r&r'-s zBaggage.__init__N)rc Csi}d}d}|rU|dD]G}d|vrq t5|}|d\}}tj|r;t|dd} t||| <d}n ||r@dnd|7}Wdn1sOwYq |durat||d <d}t|||S) zF freeze if incoming header already has sentry baggage rT,=r)r0FNr)splitr striprSENTRY_PREFIX_REGEXrrr8) rrrrrritemr-val baggage_keyr%r%r&r7s,     zBaggage.from_incoming_headercCsi}d}d}t}|r|jdurt|S|j}|j}|dur&|j|d<|dr1|d|d<|dr<|d|d<|drJt|dj |d<|drWt |d|d <t|||S) NrFr environmentreleasedsn public_keyrAr) rSrTr_propagation_contextrrErrDr rr8)rrrrrrrrErr%r%r& from_options[s&        zBaggage.from_optionscCst}i}|st|S|jpi}|j|d<|jd|d<|dr*|d|d<|dr5|d|d<|drCt|dj |d<|j rP|j t vrP|j |d<|j d ur\t|j |d <|jd urj|jrfd nd |d <|jrx|jjrx||jjt|ddS)z Populate fresh baggage entry with sentry_items and make it immutable if this is the head SDK which originates traces. rrrrrrr transactionNrtruefalsesampledFr)rSrTrrrErrrDr rrQsourceLOW_QUALITY_TRANSACTION_SOURCESrr8r_baggagerr)rrrrrEr%r%r&populate_from_transaction|s0            z!Baggage.populate_from_transactioncCs d|_dS)NFrr5r%r%r&freezes zBaggage.freezecCs$i}|jD]\}}|||<q|Sr!)rrY)r$rr-rr%r%r&rs z Baggage.dynamic_sampling_contextFc Csg}|jD]+\}}ttjt|dtt|}||Wdn1s-wYq|r;||jd |S)Nrr) rrYr r SENTRY_PREFIXrr8appendrjoin)r$include_third_partyrYr-rrr%r%r& serializes   zBaggage.serializecCsddd|dDS)zRemove Sentry baggage from the given header. Given a Baggage header, return a new Baggage header with all Sentry baggage items removed. rcss$|] }tj|s|VqdSr!)rrrr)r1rr%r%r&r2s z/Baggage.strip_sentry_baggage..)rr)rr%r%r&strip_sentry_baggages zBaggage.strip_sentry_baggagecCs:tt|jd}|durd|krdkr|SdSdS)zConvenience method to get the sample_rand value from the sentry_items. We validate the value and parse it as a float before returning it. The value is considered valid if it is a float in the range [0, 1). rN?)rrrrD)r$rr%r%r&rs zBaggage._sample_randcCsd|jddd|jdS)Nz )r rr5r%r%r&rr/zBaggage.__repr__)rT)F)r=r>r?rrrrecompilerr'rrrrrrr  staticmethodr rrr%r%r%r&rs,   #  .   rcCs&|jd}t||r dSt||ddS)zy Returns True if url matches trace_propagation_targets configured in the given client. Otherwise, returns False. trace_propagation_targetsFT)substring_matching)rErr)rurlrr%r%r&should_propagate_traces  rcCsHi}|D]\}}|dr|dd}|dd}|||<q|S)z Normalizes incoming data so the keys are all lowercase with dashes instead of underscores and stripped from known prefixes. r Nr*r))rYr9r+r)rrOr-rr%r%r&rs   rcs$ddlmfdd}|S)a Create a span decorator that can wrap both sync and async functions. :param op: The operation type for the span. :type op: str or :py:class:`sentry_sdk.consts.OP` or None :param name: The name of the span. :type name: str or None :param attributes: Additional attributes to set on the span. :type attributes: dict or None :param template: The type of span to create. This determines what kind of span instrumentation and data collection will be applied. Use predefined constants from :py:class:`sentry_sdk.consts.SPANTEMPLATE`. The default is `SPANTEMPLATE.DEFAULT` which is the right choice for most use cases. :type template: :py:class:`sentry_sdk.consts.SPANTEMPLATE` r)should_send_default_piicstfdd}zt|_Wn ty!Ynwtfdd}zt|_Wn tyCYnwtrK|S|S)zD Decorator to create a span for the given function. c st}|durtdt|i|IdHSpt}p'tp'd}t||}}|j||d,}|p>it||||||i|IdH}t ||||WdS1sgwYdSNzbCannot create a child span for %s. Please start a Sentry transaction before calling this function.r)rPrQ get_current_spanrrr _get_span_op_get_span_name start_child update_data_set_input_attributes_set_output_attributes argskwargs current_spanspan_op function_name span_namesend_piiraresult attributesfrQrPrtemplater%r& async_wrapper!s2  $zDcreate_span_decorator..span_decorator..async_wrapperc st}|durtdt|i|Spt}p#tp#d}t||}}|j||d)}|p:it||||||i|}t ||||WdS1s`wYdSrrr!r*r%r& sync_wrapperGs0  $zCcreate_span_decorator..span_decorator..sync_wrapper) functoolswrapsinspect signature __signature__rwiscoroutinefunction)r,r.r/r+rQrPrr-)r,r&span_decorators"     z-create_span_decorator..span_decorator)sentry_sdk.scoper)rPrQr+r-r7r%r6r&create_span_decorators Wr9cCs|pt}|j}|S)zU Returns the currently active span if there is one running, otherwise `None` )rSget_current_scopera)rrr$r%r%r&rus rcCsB|pt}|dur|tj|jdur|jtjdSdSdS)z Set the status of the current or given span to ERROR. Also sets the status of the transaction (root span) to ERROR. N)r set_statusr ERRORcontaining_transaction)rar%r%r&set_span_erroreds   r>r r rcCsl|\}}||ks tdt|}t|d}t|d}z |||}W|dSty5|}Y|dSw)aGenerate a sample_rand value from a trace ID. The generated value will be pseudorandomly chosen from the provided interval. Specifically, given (lower, upper) = interval, the generated value will be in the range [lower, upper). The value has 6-digit precision, so when printing with .6f, the value will never be rounded up. The pseudorandom number generator is seeded with the trace ID. z/Invalid interval: lower must be less than upperi@B)rrr randrange)rrrr,rng lower_scaled upper_scaledsample_rand_scaledr%r%r&rs   rcCs,|dus|dur dS|durd|fS|dfS)z Compute the lower (inclusive) and upper (exclusive) bounds of the range of values that a generated sample_rand value must fall into, given the parent_sampled and sample_rate values. Nr?Tr r r%)rrr%r%r&rs rcCsTd}t|tr||}|St||r(zt||}W|Sty'd}Y|Sw|S)z Gets a value from a source object. The source can be a dict or an object. It is checked for dictionary keys and object attributes. N)r7dictrDrgetattrrw)rr-rr%r%r& _get_values     rGcCs|}|tjkr/d}|r"dD]}||r!t||tr!||}nq |r+d|}|Sd}|S|tjkr;d|}|S|tjkrEd|}|S)zF Get the name of the span based on the template and the name. N)model model_namezchat chatz invoke_agent z execute_tool )r AI_CHATrDr7r8AI_AGENTAI_TOOL)r-rQr#r'rHr-r%r%r&rs&   rcCs2tjtjtjtjtjtji}||tj }t |S)z> Get the operation of the span based on the template. ) r rKr GEN_AI_CHATrLGEN_AI_INVOKE_AGENTrMGEN_AI_EXECUTE_TOOLrDFUNCTIONr8)r-mappingrPr%r%r&rs rc s<i|tjtjtjfvr|tjtftjtftjtftjtftjt ftj t ftj t ftj t ftjt ftjt fd fdd}t|D]7\}}|dkr_t|tr_tjgd|dqD|dkrvt|trvtjgd|dqD|||qD|tjkr|rt||d tj<tjvrttjtj<S) z; Get input attributes for the given span template. ) rHrIagent agent_name max_tokensfrequency_penaltypresence_penalty temperaturetop_ptop_kcs>|vr|\}}|durt||r||<dSdSdSdSr!)r7)r-r attribute data_typer+rRr%r& _set_from_key s   z,_get_input_attributes.._set_from_keypromptuser)rolecontent system_promptsystem)r"r#)r rLrMrKrGEN_AI_REQUEST_MODELr8GEN_AI_AGENT_NAMEGEN_AI_REQUEST_MAX_TOKENSr GEN_AI_REQUEST_FREQUENCY_PENALTYrGEN_AI_REQUEST_PRESENCE_PENALTYGEN_AI_REQUEST_TEMPERATUREGEN_AI_REQUEST_TOP_PGEN_AI_REQUEST_TOP_KlistrYr7 setdefaultGEN_AI_REQUEST_MESSAGESrrGEN_AI_TOOL_INPUT)r-r(r"r#r^r-rr%r]r&_get_input_attributessF    rqcs:ifdd}|tjd|tjd|tjdS)z Get usage attributes. cs2|D]}t|}|durt|tr||<qdSr!)rGr7r)r[keysr-rr+usager%r&_set_from_keys6s  z-_get_usage_attributes.._set_from_keys) prompt_tokens input_tokens)completion_tokens output_tokens) total_tokens)rGEN_AI_USAGE_INPUT_TOKENSGEN_AI_USAGE_OUTPUT_TOKENSGEN_AI_USAGE_TOTAL_TOKENS)rtrur%rsr&_get_usage_attributes/sr~c Csi}|tjtjtjfvrqtZ|g}t|d}||t|d}t|d}|||D] }|dur;|t|q.t|d}|durOt |t rO||t j <t|d}|durbt |t rb||t j <Wdn1slwY|tjkr|rt ||t j<|S)z< Get output attributes for the given span template. rtmetadataNrHrI)r rLrMrKr rGrrr~r7r8rGEN_AI_RESPONSE_MODELrGEN_AI_TOOL_OUTPUT) r-r(r)r+usage_candidatesrtmetausage_candidaterIr%r%r&_get_output_attributesMs0         rc Csi}|tjkrtjdtj|i}n$|tjkrtjdi}n|tjkr4tjdtj|i}|j}|dur4||tj <| t ||||| |pCidS)a Set span input attributes based on the given span template. :param span: The span to set attributes on. :param template: The template to use to set attributes on the span. :param send_pii: Whether to send PII data. :param f: The wrapped function. :param args: The arguments to the wrapped function. :param kwargs: The keyword arguments to the wrapped function. invoke_agentrJ execute_toolN) r rLrGEN_AI_OPERATION_NAMErfrKrMGEN_AI_TOOL_NAMErGEN_AI_TOOL_DESCRIPTIONrrqr) rar-r(rQr,r"r#r+ docstringr%r%r&rts"    rcCs|t|||p idS)a% Set span output attributes based on the given span template. :param span: The span to set attributes on. :param template: The template to use to set attributes on the span. :param send_pii: Whether to send PII data. :param result: The result of the wrapped function. N)rr)rar-r(r)r%r%r&r s r )rrr)Span)FrGr!)Q contextlibr0r2rrrcollections.abcrdatetimerrandomr urllib.parserrrrSsentry_sdk.constsrrr r sentry_sdk.utilsr r r rrrrrrrrrrtypingrrrrrrtypesrrrbase64_strippedrrFcontextmanagerrcrtryrrrrUrrrrDEFAULTr9rr>rrrGrrrqr~rrr sentry_sdk.tracingrrrrr%r%r%r&s   <          / [ 4J  p    9'%