o i2@sddlmZddlmZddlmZddlmZddlmZddlmZddlm Z erDddl m Z dd l m Z dd lmZdd lmZdd lmZdd lZddlmZddlmZddlmZddlmZddlmZddlmZddlm Z ddl!m"Z"ddl#m$Z$ddl%m&Z&ddl'm(Z(ddl'm)Z)ddl*m+Z+ddl,m-Z-ddl.m/Z/ddl0m1Z1e$e2Z3e/Z4e5de6ddd d!e7fd"d#Z8d!e6e7e7ffd$d%Z9Gd&d'd'e:Z;d(d)ZGd/d0d0e;Z?d S)1)Mapping) TYPE_CHECKING)Any)Callable)Iterable)Optional) SpanDirection)Pin)Config)Span)Tracer)quoteN)config) SPAN_KIND) trace_utils)SpanKind) SpanTypes)core)BlockingException) COMPONENT) get_logger)schematize_url_operation) get_blocked) set_blocked)DDTraceDeprecationWarning)from_wsgi_header)HTTPPropagator) deprecatewsgiT)_default_servicedistributed_tracingreturncCdS)Nr$r$r$V/home/ubuntu/.local/lib/python3.10/site-packages/ddtrace/contrib/internal/wsgi/wsgi.py get_version4sr&cCsddiS)Nr*r$r$r$r$r%_supported_versions8sr(c@seZdZdZ d+dededdddd d ed d f d dZed e fddZ ed e fddZ ed e fddZ dede d ejfddZ d,de ddddde deded d fddZd,d dded!eed d fd"d#Zddded$ed d fd%d&Zd'dd(ed d fd)d*Zd S)-_DDWSGIMiddlewareBaseaBase WSGI middleware class. :param application: The WSGI application to apply the middleware to. :param tracer: [Deprecated] Global tracer will be used instead. :param int_config: Integration specific configuration object. :param pin: Set tracing metadata on a particular traced connection :param app_is_iterator: Boolean indicating whether the wrapped app is a Python iterator F applicationtracerr int_configr pinr app_is_iteratorr!NcCs4|dur tddtdd||_||_||_||_dS)Nz"The tracer parameter is deprecatedz'The global tracer will be used instead.z5.0.0)messagecategoryremoval_version)rrapp_config_pinr.)selfr*r+r,r-r.r$r$r%__init__Fs z_DDWSGIMiddlewareBase.__init__cCt)zReturns the name of a response span. Example: `flask.response`r8r:r$r$r%_response_span_namedr<z)_DDWSGIMiddlewareBase._response_span_nameenvironstart_responsec s2t|}d}d}tjd|d|dt|j|jtj t |dr"|j n|j |j|j|||dddމ dtfd d }trR|\}}}|t|||g}d }td |fd} |rtd |fz ||d}WnRty} z$t| jd|\}}}|t|||g}tdfWYd} ~ nDd} ~ wty} z | } WYd} ~ n0d} ~ wtytdfwtrtdtfjjpd\} } }|g}td|ftd||jfj} | r| jr| jtd|f| | r| jngWdS1swYdS)Nr$Tz wsgi.__call__ REMOTE_ADDR_request_call_namereq_span) remote_addrheadersheaders_case_sensitiveservice span_type span_namemiddleware_configintegration_configdistributed_headersr? middlewarespan_keyactivate_distributed_headerszwsgi.construct_urlcs>tdtfj}|r|j\}}}ndgd}}}|||fS)Nwsgi.block.startedir#)rdispatch_with_results construct_urlstatus_headers_contentvalue)resultstatusrEcontentctxr$r% blocked_views z4_DDWSGIMiddlewareBase.__call__..blocked_viewFzwsgi.block_decidedzwsgi.request.prepareintercept_start_responserzwsgi.app.exceptionrP)NNr#zwsgi.app.successzwsgi.request.complete)get_request_headersrcontext_with_datagetr int_servicer4r3rWEBhasattrrBr;set_itemrRrstrdispatchr2 find_itemrrargs StopIteration BaseExceptionrQrSrTr.traced_iterable _finish_spans) r5r?r@rEclosing_iterable not_blockedrZrWrVstop_iteration_exceptione_rUr$rXr%__call__is        &z_DDWSGIMiddlewareBase.__call__ request_spanr app_spanrVexc_infocCshtjd|||||tjtd|jdt|jjt t j id ||||WdS1s-wYdS)zDsets the status code on a request span when start_response is called wsgi.responseNF) rMrq parent_callrVr?rHrG start_spantags) rr]rr`rr_r3rintegration_namerrSERVER)r5r@rqrrrVr?rsr$r$r%_traced_start_responses   $z,_DDWSGIMiddlewareBase._traced_start_responserCparsed_headerscCr"z7Implement to modify span attributes on the request_spanNr$)r5rCr?r{r$r$r%_request_span_modifierz,_DDWSGIMiddlewareBase._request_span_modifierrUcCr")z;Implement to modify span attributes on the application_spanNr$)r5rrr?rUr$r$r%_application_span_modifierr~z0_DDWSGIMiddlewareBase._application_span_modifier resp_spanresponsecCr"r|r$r5rrr$r$r%_response_span_modifierr~z-_DDWSGIMiddlewareBase._response_span_modifier)FN)__name__ __module__ __qualname____doc__rrboolr6propertyrcr;r=r>rwrapt ObjectProxyrpdictrrzr}rrr$r$r$r%r)<sT \ r)cCs<|dd}|dr||d7}n)||d7}|ddkr-|ddkr,|d|d7}n|dd kr;|d|d7}|t|d d 7}|d re||d 7}|d rcd|d vrc|d|d 7}|S|dr||d7}|d rd|dvr|d|d 7}|S|t|dd 7}|d r|d|d 7}|S)zF https://www.python.org/dev/peps/pep-3333/#url-reconstruction zwsgi.url_schemez:// HTTP_HOST SERVER_NAMEhttps SERVER_PORT443:80 SCRIPT_NAMEr#RAW_URI QUERY_STRING? REQUEST_URI PATH_INFO)r^r )r?urlr$r$r%rRs2          rRr?cCs8i}|D]}|drt|}|r||||<q|S)zH Manually grab the request headers from the environ dictionary. HTTP_)keys startswithr)r?request_headerskeynamer$r$r%r\s   r\cCsd|d|d|_dS)Nz{} {}REQUEST_METHODr)formatresource)spanr?r$r$r%default_wsgi_span_modifiersrc seZdZdZeddejdZdZdZ de dfd e d e d d e d eeefgdfdeddf fdd ZdddZdddZddZZS)DDWSGIMiddlewareaWSGI middleware providing tracing around an application. :param application: The WSGI application to apply the middleware to. :param tracer: [Deprecated] Global tracer will be used. :param span_modifier: "Span" modifier that can add tags to the root span. Defaults to using the request method and url in the resource. :param app_is_iterator: Boolean indicating whether the wrapped WSGI app is a Python iterator z wsgi.requesthttp)protocol directionzwsgi.applicationrtNFr*r+r span_modifierr r.r!cs&tt|j||tjd|d||_dS)N)r.)superrr6rrr)r5r*r+rr. __class__r$r%r6%s zDDWSGIMiddleware.__init__cCstjd|||||tjdtd|jdt|jjt t j id )}|j ||||WdWdS1s9wYWddS1sIwYdS)Nrtzwsgi.start_responseT) rMrqrurVr?rHrIrGrvrw) rr]rr`rr_r3rrxrrryr)r5r@rqrrrVr?rsrYr$r$r%rz/s&   Rz'DDWSGIMiddleware._traced_start_responsecCs6t|}|dur |nt|}td|||||fdS)Nzwsgi.request.prepared)rRr\rrd)r5rCr?r{rrr$r$r%r}Bsz'DDWSGIMiddleware._request_span_modifiercCstd||fdS)Nzwsgi.response.prepared)rrdrr$r$r%rGsz(DDWSGIMiddleware._response_span_modifierr)rrrrrrINBOUNDr;r=r>rrrrrrcrr6rzr}r __classcell__r$r$rr%rs,  r)@collections.abcrtypingrrrrr-ddtrace.internal.schema.span_attribute_schemarddtrace._trace.pinr !ddtrace.internal.settings._configr ddtrace.tracer r urllib.parser rddtracerddtrace.constantsrddtrace.contribr ddtrace.extrrddtrace.internalrddtrace.internal._exceptionsrddtrace.internal.constantsrddtrace.internal.loggerrddtrace.internal.schemarddtrace.internal.utilsrr#ddtrace.internal.utils.deprecationsrddtrace.propagation._utilsrddtrace.propagation.httprddtrace.vendor.debtcollectorrrlog propagator_addrrcr&r(objectr)rRr\rrr$r$r$r%s^                            %"&