o i) @sddlZddlmZddlmZddlZddlmZmZddl m Z ddl m Z ddl mZmZmZddlmZmZdd lmZdd lmZmZdd lmZmZmZmZerdd lmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%dd l&m'Z'm(Z(ddlm)Z)e%dZ*e%dZ+e%dZ,Gddde#Z-edZ.dd%dd&d'd(e/d)dd*d+ddf d,d-Z4d?d/d0Z5d@d2d3Z6dAd5d6Z7Gd7d8d8Z8ddd e1dd9fd:d;Z9dS)BN)partial) TYPE_CHECKING) _get_headersget_host)continue_trace)OP)DEFAULT_HTTP_METHODS_TO_CAPTURE_filter_headers nullcontext)should_send_default_piiuse_isolation_scope) track_session) TransactionTransactionSource) ContextVarcapture_internal_exceptionsevent_from_exceptionreraise)AnyCallableDictIteratorOptionalProtocolTupleTypeVar)EventEventProcessor)ExcInfoWsgiResponseIterWsgiResponseHeaders WsgiExcInfoc @s*eZdZ d deddddddfd d ZdS) StartResponseNstatusresponse_headersr exc_infoOptional[WsgiExcInfo]returnrcCsdSN)selfr#r$r%r)r)Y/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/sentry_sdk/integrations/wsgi.py__call__#szStartResponse.__call__r()__name__ __module__ __qualname__strr,r)r)r)r+r""sr"sentry_wsgi_middleware_appliedutf-8replacescharseterrorsr'cCs|d||S)Nlatin1)encodedecode)r4r5r6r)r)r+wsgi_decoding_dance/sr:FenvironDict[str, str]use_x_forwarded_forcCsV|ddd}|ddd}|d|}d|dt||t|dfS)zPReturn the absolute URL without query string for the given WSGI environment. SCRIPT_NAME/ PATH_INFOz %s://%s/%szwsgi.url_scheme)getrstriplstriprr:)r;r= script_name path_infopathr)r)r+get_request_url3s rHc @sLeZdZdZddefdddededd d d f d d Z     dddZd S)SentryWsgiMiddlewareappr= span_originhttp_methods_to_captureFmanualrKz3Callable[[Dict[str, str], Callable[..., Any]], Any]r=rLrMzTuple[str, ...]r'NcCs||_||_||_||_dSr(rJ)r*rKr=rLrMr)r)r+__init__Ks zSentryWsgiMiddleware.__init__r;r<start_responseCallable[..., Any]_ScopedResponsec Cstdr |||Stdzt}t|ddt|d|_ | t ||j Wdn1s;wY|dd }d}||jvr[t|tjdtj|jd }|durhtj|d |id nt}|#z ||tt||}WntyttYnwWdn1swYWdn1swYWdn1swYWtdntdwt||S) NFTrequest) session_modewsgiREQUEST_METHODr?zgeneric WSGI request)opnamesourceorigin wsgi_environ)custom_sampling_context)_wsgi_middleware_appliedrBrKset sentry_sdkisolation_scoper rclear_breadcrumbs_nameadd_event_processor_make_wsgi_event_processorr=upperrMrr HTTP_SERVERrROUTErLstart_transactionr r_sentry_start_response BaseExceptionr_capture_exceptionrR)r*r;rPscopemethod transactiontransaction_contextresponser)r)r+r,Wsh        ) zSentryWsgiMiddleware.__call__)r;r<rPrQr'rR) r-r.r/ __slots__rboolr0rOr,r)r)r)r+rICs.   rIold_start_responsernzOptional[Transaction]r#r$r%r&cCsjtt|ddd}|dur||Wdn1s!wY|dur/|||S||||S)N r)rintsplitset_http_status)rsrnr#r$r% status_intr)r)r+ris   riIterator[Tuple[str, str]]ccs>ddg}tr |dg7}|D] }||vr|||fVqdS)z Returns our explicitly included environment variables we want to capture (server name, port and remote addr if pii is enabled). SERVER_NAME SERVER_PORT REMOTE_ADDRN)r )r;keyskeyr)r)r+ _get_environs r Optional[Any]c CsXz |dddWSttfyYnwz|dWSty&Ynw|dS)z Infer the user IP address from various headers. This cannot be used in security sensitive situations since the value may be forged from a client, but it's good enough for the event payload. HTTP_X_FORWARDED_FOR,rHTTP_X_REAL_IPr})rwstripKeyError IndexErrorrB)r;r)r)r+ get_client_ips   rrcCsXt}|d}t|to|jdv}|s*t|tjdddd\}}tj ||d|S)z Captures the current exception and sends it to Sentry. Returns the ExcInfo tuple to it can be reraised afterwards. ru)rNrUF)typehandled)client_options mechanism)hint) sysr% isinstance SystemExitcoderr_ get_clientoptions capture_event)r%eshould_skip_captureeventrr)r)r+rks rkc@s>eZdZdZdZ      dd d Zdd d Zdd dZdS)rRaY Users a separate scope for each response chunk. This will make WSGI apps more tolerant against: - WSGI servers streaming responses from a different thread/from different threads than the one that called start_response - close() not being called - WSGI servers streaming responses interleaved from the same thread ) _response_scoperlsentry_sdk.scope.ScoperpIterator[bytes]r'NcCs||_||_dSr()rr)r*rlrpr)r)r+rOs z_ScopedResponse.__init__c cst|j} t|j,zt|}Wnty#YWddSty0ttYnwWdn1s;wY|Vqr() iterrr rnext StopIterationrjrrk)r*iteratorchunkr)r)r+__iter__s      z_ScopedResponse.__iter__c Cst|j7z|jWntyYnty#ttYnwWddSWddSWddS1s?wYdSr()r rrcloseAttributeErrorrjrrk)r*r)r)r+rs   "z_ScopedResponse.close)rlrrprr'N)r'r)r'N)r-r.r/__doc__rqrOrrr)r)r)r+rRs  rRrcs^t|t|||d|dtt|ttt|d fdd }|S) N QUERY_STRINGrVrrrDict[str, Any]r'cst4|di}tr|di}r|d|d<|d<|d<|d<|d<Wd|S1s:wY|S) NrSuser ip_addressurl query_stringrmenvheaders)r setdefaultr )rr request_info user_info client_iprrrmr request_urlr)r+event_processors     z3_make_wsgi_event_processor..event_processor)rrrrr'r)rrHrBdictrr r)r;r=rr)rr+rds    rd)r2r3)Fr()r;r<r'rz)r;r<r'r)r'r):r functoolsrtypingrr_sentry_sdk._werkzeugrrsentry_sdk.apirsentry_sdk.constsr$sentry_sdk.integrations._wsgi_commonrr r sentry_sdk.scoper r sentry_sdk.sessionsr sentry_sdk.tracingrrsentry_sdk.utilsrrrrrrrrrrrrsentry_sdk._typesrrrrr r!r"r]r0r:rrrHrIrirrrkrRrdr)r)r)r+sp     (   O   +