o ic.@sdZddlmZddlZddlZddlmZddlmZddlmZddlm Z ddlm Z dd lm Z dd lm Z dd lm Z dd lmZdd lmZddlZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlm Z ddlm!Z!ddl"m#Z#ddl$m%Z%ddl$m&Z&ddl'm(Z(ddl)m*Z*ddl+m,Z,ddl-m.Z.ddl/m.Z0ddl1Z2dd l3m4Z4erdd!l5m6Z6ddl7mZdd"l7m8Z8e,e9Z:ej;Zj?j@Z@e2j=j>j?jAZAd#ZBd$ZCeDd%ZEd&ZFd'eGeHeHfd(ed)d*d+eHd,df d-d.ZIdzd'e eHeHfd0eJd,eHfd1d2ZKd3eHd,eHfd4d5ZL /d{d'e e eHeHfd6e eHd0eJd,eHfd7d8ZMd'eGeHeHfd(ed)d*d,dfd9d:ZNd'eGeHeHfd(ed)d*d,dfd;d<ZOd=eHd,eHfd>d?ZPd@dAZQdzdBd*dCeJd,eJfdDdEZRd|dFe edBd*dCe eHd,e eHfdGdHZSd|dFe edBd*dCe eHd,e eHfdIdJZT                 /  d}d(ed)d*dKe eHd=e eHdLe eHdMe eHdNe e eUeHfdOe eHdPe eHdQe e eHeHfdRe e eHeHfdSe e eHeHfdTe e eUeHfdUe eHdVe eGeHeHfdWe eGeHeHfdXe e eHeGeHeVeHffd6e eHd0eJdYe eHdZe eGeHeHfd,df,d[d\ZW   d~d]d^dBe d*dRe eGeHeHfd_e eJd,df d`daZXdbedcefdddeZY f g ddhedieHdjeHdke eeHgeJfd,e eZeHefddff dldmZ[ f  dd(edne eZeHefdieHdke eeHgeJfdoe eegefd,df dpdqZ\d=eHd,eZeHeHffdrdsZ]Gdtdudue^Z_dvdwZ`dxdyZadS)zJ This module contains utility functions for writing ddtrace integrations. )dequeN) TYPE_CHECKING)Any)Callable) Generator)Iterator)Mapping)Optional)Union)cast)parse)Pin)Span) _ORIGIN_KEY)USER_AGENT_PATTERNS)"_get_header_value_case_insensitive)_get_request_header_user_agent)_normalize_tag_name) _set_url_tag)set_user)http)net)core) ensure_text) ip_is_global)SAMPLING_DECISION_TRACE_TAG_KEY)dispatch) get_logger)config)HTTPPropagator)IntegrationConfig)Tracerrequestresponsez([^a-z0-9_\-:/]){1}) zx-forwarded-forz x-real-ipztrue-client-ipz x-client-ip forwardedz forwarded-forzx-cluster-client-ipzfastly-client-ipzcf-connecting-ipzcf-connecting-ipv6headersspanintegration_configr request_or_responsereturncCst|tszt|}Wn tyYdSw|dur!tddS|D]\}}||}|dur3q%||p;t|||q%dS)aP :param headers: A dict of http headers to be stored in the span :type headers: dict or list :param span: The Span instance where tags will be stored :type span: ddtrace.trace.Span :param integration_config: An integration specific config object. :type integration_config: ddtrace.settings.IntegrationConfig Nz>Skipping headers tracing as no integration config was provided) isinstancedict Exceptionlogdebugitems_header_tag_name _set_tag_strr)r%r&r'r( header_name header_valuetag_namer5X/home/ubuntu/.local/lib/python3.10/site-packages/ddtrace/contrib/internal/trace_utils.py_store_headersOs     r7Fheaders_are_case_sensitivec Cs^|r t|tj}n|tj}|r-zt|}|jr|jWSWdSttfy,YdSwdS)aXGet referrer host from request headers :param headers: A dict of http headers to be stored in the span :type headers: dict or list :param headers_are_case_sensitive: Whether the headers are case sensitive :type headers_are_case_sensitive: bool :return: The referrer host if found, empty string otherwise :rtype: str ) rrREFERER_HEADERgetr urlparsehostname ValueErrorAttributeError)r%r8referrer parsed_urlr5r5r6!_get_request_header_referrer_hostns   rBip_header_valuecCs8gd}|D]}t||tj}r|dSqdS)zParse the ip header, either in Forwarded-For format or Forwarded format. references: https://developer.mozilla.org/en-US/docs/Web/HTTP/Reference/Headers/Forwarded )z+^\s*(?P[0-9]+\.[0-9]+\.[0-9]+\.[0-9]+)$z6(?:^|;)\s*for="?(?P[0-9]+\.[0-9]+\.[0-9]+\.[0-9]+)z^\s*(?P[0-9a-fA-F:]+)$z((?:^|;)\s*for="\[(?P[0-9a-fA-F:]+)\]ipr9)research IGNORECASEgroup)rCIP_EXTRACTIONSpatternmr5r5r6_parse_ip_headers rLpeer_ipc sdtdttffdd }s%z tt|}W|Sty$YdSwd}tj}|r_|r7|ddn|}|sDt d|dSz tt|}Wn;ty^t d ||YdSwrzd d D}t D] }||vrx||}nqlnt D] }|vr|}nq|d} |r| d } | D]!} t| } | sqzt| r| WS| s| } WqtyYqwz t|s| s|WSW| StyY| Sw) Nkeyr)css|St|SN)r;r)rNr%r8r5r6get_header_values  z7_get_request_header_client_ip..get_header_valuer9_-zs"z1_get_request_header_client_ip..,)strr ipaddress ip_addressr>r_client_ip_headerrTrUr-r.r/ IP_PATTERNSsplitrLr) r%rMr8rQrRrCuser_configured_ip_header new_headers ip_headerprivate_ip_from_headersip_listrDr5rPr6_get_request_header_client_ips~        rgcCt|||tdS)a Store request headers as a span's tags :param headers: All the request's http headers, will be filtered through the whitelist :type headers: dict or list :param span: The Span instance where tags will be stored :type span: ddtrace.trace.Span :param integration_config: An integration specific config object. :type integration_config: ddtrace.settings.IntegrationConfig N)r7REQUESTr%r&r'r5r5r6_store_request_headers rkcCrh)a Store response headers as a span's tags :param headers: All the response's http headers, will be filtered through the whitelist :type headers: dict or list :param span: The Span instance where tags will be stored :type span: ddtrace.trace.Span :param integration_config: An integration specific config object. :type integration_config: ddtrace.settings.IntegrationConfig N)r7RESPONSErjr5r5r6_store_response_headersrlrnurlcCsZd|vr+t|}|j}d|vr|S||ddd}t|j||jd|jdfS|S)zA Sanitize url by removing parts with potential auth info @Nr9)r r<netlocindex urlunparseschemepathquery)roparsedrrr5r5r6_sanitized_urls   rycsfdd}|S)aHelper for providing tracing essentials (module and pin) for tracing wrappers. This helper enables tracing wrappers to dynamically be disabled when the corresponding pin is disabled. Usage:: @with_traced_module def my_traced_wrapper(django, pin, func, instance, args, kwargs): # Do tracing stuff pass def patch(): import django wrap(django.somefunc, my_traced_wrapper(django)) csfdd}|S)NcsVt|}|r|s||i|S|s"td|||i|S|||||S)Nz"Pin not found for traced method %r)r _findenabledr-r.)wrappedinstanceargskwargspin)funcmodr5r6wrapper.s   z5with_traced_module..with_mod..wrapperr5)rrr)rr6with_mod-s z$with_traced_module..with_modr5)rrr5rr6with_traced_modules  r int_configdefaultcCs4d|vr |jdur |jSd|vr|jdur|jS|S)zJReturns whether distributed tracing is enabled for this integration configdistributed_tracing_enabledNdistributed_tracing)rr)rrr5r5r6r<s rrcCs|dur |jr |jSd|vr|jdurtt|jSd|vr(|jdur(tt|jS|j}|r:||jjkr:tt|Sd|vrI|jdurItt|jS|durT|rTtt|S|S)aQReturns the service name for an integration which is internal to the application. Internal meaning that the work belongs to the user's application. Eg. Web framework, sqlalchemy, web servers. For internal integrations we prioritize overrides, then global defaults and lastly the default provided by the integration. Nservice service_name_default_service)rr r\r global_config _get_service_inferred_base_servicer)rrrglobal_servicer5r5r6 int_serviceEs        rcCsr|dur |jr |jSd|vr|jdurtt|jSd|vr(|jdur(tt|jSd|vr7|jdur7tt|jS|S)zReturns the service name for an integration which is external to the application. External meaning that the integration generates spans wrapping code that is outside the scope of the user's application. Eg. A database, RPC, cache, etc. Nrrr)rr r\rr)rrrr5r5r6 ext_servicejs   rmethod target_hostserver_address status_code status_msgrw parsed_queryrequest_headersresponse_headersretries_remainraw_urirequest_cookiesrequest_path_params request_bodyrouteresponse_cookiescCs |dur |tj||durt|}t|||||dur%|tj||dur0|tj||dur^zt|}Wnt t fyKt d|Ynw|tj t|tj|r^d|_|duri|tj||durw|jrw|tj||}| rt| |}|r|tj|t| |}|r|tj|tjstjrtd}|st| ||p|}|r|tj ||d||j!r t"t#| ||| dur|j!rt$t#| ||| dur|tj%t| t&d||| ||| || |||| |g |dur|tj'|dSdS)a Set HTTP metas on the span :param method: the HTTP method :param url: the HTTP URL :param status_code: the HTTP status code :param status_msg: the HTTP status message :param query: the HTTP query part of the URI as a string :param parsed_query: the HTTP query part of the URI as parsed by the framework and forwarded to the user code :param request_headers: the HTTP request headers :param response_headers: the HTTP response headers :param raw_uri: the full raw HTTP URI (including ports and query) :param request_cookies: the HTTP request cookies as a dict :param request_path_params: the parameters of the HTTP URL as set by the framework: /posts/ would give us { "id": } Nz,failed to convert http status code %r to intrqzhttp.request.remote_ipznetwork.client.ipset_http_meta_for_asm)(r1rMETHODryrr TARGET_HOSTSERVER_ADDRESSint TypeErrorr>r-r. STATUS_CODEr\r _http_server is_error_codeerror STATUS_MSGtrace_query_string QUERY_STRINGr USER_AGENTrBREFERRER_HOSTNAME asm_config _asm_enabled_retrieve_client_ipr find_itemrg CLIENT_IPis_header_tracing_configuredrkr+rnRETRIES_REMAINrROUTE)r&r'rrorrrrrwrrrrrrrrrMr8rrint_status_code request_ip user_agent referrer_hostr5r5r6 set_http_meta~s'        rtracerr!overridecCs|durdS|s|rWt|rYt|}|durdS|js$|js$|js$dS|}|rA|jr6|jrA|j|jkrAtd|j|j dS|j |t d||ft d|fdSdSdS)a+ Helper for activating a distributed trace headers' context if enabled in integration config. int_config will be used to check if distributed trace headers context will be activated, but override will override whatever value is set in int_config if passed any value other than None. FNzlwill not activate extracted Context(trace_id=%r, span_id=%r), a context with that trace id is already activez!http.activate_distributed_headerszdistributed_context.activated)rrextracttrace_id_baggage _span_linkscurrent_trace_contextr-r.span_idcontext_provideractivaterr)rrrrcontextcurrent_contextr5r5r6activate_distributed_headerss2     r target_spanparentcCs|jjD]\}}|j|||d||q|jjdur'|jj|j_|jjtr8|t|jjt|jjt rK|t |jjt dSdS)zZ Copies baggage, tags, origin, sampling decision from parent span to target span. zbaggage.N) rrr/set_baggage_itemr1sampling_priority_metar;rr)rrrNvaluer5r5r6_copy_trace_level_tags7s  r.r9objsepprefixexclude_policyc#sxt}|||f|r:|\}|dur|rq t|tr1|fdd|Dn|fV|s dSdS)Nc3s.|]\}}r|fn||fVqdSrO)joinrVprr5r6 Vs,z_flatten..)rappendpopr*r+extendr/)rrrrsrYr5rr6_flattenIs     rr/ processorc CsF|D]\}}t||||D]\}}|||dur||n|q qdSrO)rset_tag) r&r/rrrrrtagrYr5r5r6set_flattened_tags[s rcCsRt|}|j}|jstdj|d}|jddd}|ddd}||fS)Nz//{url})rorprq:r)r r<rwrrformatra)ro parse_resultrwrrr5r5r6&extract_netloc_and_query_info_from_urlgs rc@s eZdZdS)InterruptExceptionN)__name__ __module__ __qualname__r5r5r5r6rtsrcCs,|rt|ts t|trt|St|S|SrO)r*rfloatr\r)attrr5r5r6_convert_to_stringxs rc CsX|sdSz|}|dD]}t||sWdSt||}q WdSttfy+YdSw)a Helper function to safely check if a nested attribute path exists on a module. Args: module: The root module object attr_path: Dot-separated path to the attribute (e.g., "flows.llm_flows.functions") Returns: bool: True if the full path exists, False otherwise Example: check_module_path(adk, "flows.llm_flows.functions.__call_tool_async") check_module_path(adk, "agents.llm_agent.LlmAgent") FrT)rahasattrgetattr ImportErrorr?)module attr_pathcurrentrr5r5r6check_module_paths  r)F)NFrO)NNNNNNNNNNNNNNNNFNN)NNN)rr9N)rNN)b__doc__ collectionsrr]rEtypingrrrrrrr r r urllibr wraptddtrace._trace.pinr ddtrace._trace.spanrddtrace.constantsr)ddtrace.contrib.internal.trace_utils_baserrrrrr ddtrace.extrrddtrace.internalrddtrace.internal.compatrrddtrace.internal.constantsrddtrace.internal.core.event_hubrddtrace.internal.loggerr!ddtrace.internal.settings._configrddtrace.internal.settings.asmrddtrace.internal.utils.wrappersddtraceddtrace.propagation.httpr%ddtrace.internal.settings.integrationr ddtrace.tracer!rr-wrap_function_wrapperwrapinternalutilswrappersunwrap iswrappedrirmcompileNORMALIZE_PATTERNr`r+r\r7boolrBrLrgrkrnryrrrrrlistrrrtuplerrrr,rrrr5r5r5r6s                                        "O"  "( (%       7