U
췀g @ s| d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl m
Z
m Z d dl m
Z
e
rd dl mZ d dl mZ d dl mZ d dl mZ d dl mZ d d lmZmZmZ d d
lmZ ed Zd Zd
Z G dd dZ zxd dl m Z d dl m!Z!m"Z"m#Z# d dl$m%Z% dZ&dZ'dZ(dZ)dZ*e+dd e e,e*- e*. D Z/G dd de Z0W n e1k
rn dZ%Y nX d d Z2dS ) N)chainproduct)
TYPE_CHECKING)Any)Callable)Dict)Optional)Self)logger env_to_boolcapture_internal_exceptions)Envelope spotlightz http://localhost:8969/streamz(sentry_sdk.spotlight.SpotlightMiddlewarec @ s e Zd Zdd Zdd ZdS )SpotlightClientc C s || _ t | _d| _d S )Nr )urlurllib3 PoolManagerhttptries)selfr r B/opt/hc_python/lib/python3.8/site-packages/sentry_sdk/spotlight.py__init__% s
zSpotlightClient.__init__c
C sr t }|| z*| jj| j| dddid}| W n0 tk
rl } zt
t | W 5 d }~X Y nX d S )NPOST Content-Typez application/x-sentry-envelope)r bodymethodheaders) ioBytesIOZserialize_intor requestr getvalueclose Exception
sentry_loggerwarningstr)r enveloper reqer r r capture_envelope+ s
z SpotlightClient.capture_envelopeN)__name__
__module__ __qualname__r r* r r r r r $ s r )MiddlewareMixin)HttpResponseServerError HttpResponse HttpRequest)settingsz/assets/main.jsz4z
zcharset=r c c s | ]}d d|V qdS )z N)formatjoin).0charsr r r M s r8 c s< e Zd ZdZ fddZedd Zdd Zdd Z Z S )
SpotlightMiddlewareNc sT t | dd l}|j| _| j j}|d kr>td d S t j
|j d| _
d S )Nr zPCannot find Spotlight client from SpotlightMiddleware, disabling the middleware.z../)superr Zsentry_sdk.apiapi
sentry_sdkZ
get_clientr r$ r% urllibparseurljoinr _spotlight_url)r Z get_responser< Zspotlight_client __class__r r r U s z SpotlightMiddleware.__init__c
C s | j d kr~z r? r@ SPOTLIGHT_JS_ENTRY_PATHr Requesturlopen SPOTLIGHT_JS_SNIPPET_PATTERNr4 errorURLErrorr$ debug)r Zspotlight_js_urlr( errr r r spotlight_scriptf s*
z$SpotlightMiddleware.spotlight_scriptc s t dd jdd dD }|d }t|dkr^|d tr^|d ttd nd | jd k rj s|d krtj
}| j }t d
d fd dt
D D |}j
d | | j
|d _
d r|t| jd < S )
Nc s s | ]}| V qd S N)strip)r6 pr r r r8 s z7SpotlightMiddleware.process_response..r r3 ;r utf-8z text/htmlc s s | ]}|d kr|V qdS )Nr )r6 idxr r r r8 s c 3 s | ]}j | V qd S rO )contentrfindencode)r6 Z body_variantencodingresponser r r8 s zContent-Length)tupler getlowersplitlen
startswithCHARSET_PREFIXrN Z streamingrW rY next BODY_CLOSE_TAG_POSSIBILITIES
has_header)r _requestr\ Zcontent_type_header content_typecontent_lengthZ injectionZinjection_siter rZ r process_response sD
z$SpotlightMiddleware.process_responsec C sn t js
d S z tj| j d}W n tjj k
r@ Y d S X | j
|}t |
dtj| j|dS d S )NrT z)Z
spotlight_urlevent_id)r2 DEBUGr= r rH r@ readdecoderJ rK r< Zcapture_exceptionr/ replace SPOTLIGHT_ERROR_PAGE_SNIPPETr4 )r rg exceptionr rk r r r process_exception s" z%SpotlightMiddleware.process_exception)
r+ r, r- rE r propertyrN rj rr
__classcell__r r rA r r9 R s
-r9 c C s t tj}|t d t| tt j |
d}|dkrHt }t |t
sVd S td k rtjrttj
ddrttj
ddrt 4 tj}t|krt|t|tft_td W 5 Q R X t|}td| |S ) Nz' [spotlight] %(levelname)s: %(message)sr TZSENTRY_SPOTLIGHT_ON_ERROR1ZSENTRY_SPOTLIGHT_MIDDLEWAREz(Enabled Spotlight integration for Djangoz%Enabled Spotlight using sidecar at %s)logging
StreamHandlersysstderr setFormatter Formatterr
addHandlersetLevelINFOr^ DEFAULT_SPOTLIGHT_URL
isinstancer& r2 rl r osenvironr Z
MIDDLEWARE DJANGO_SPOTLIGHT_MIDDLEWARE_PATHtyper infor )optionsZ_handlerr Z
middlewareclientr r r setup_spotlight s6
r )3r rv r urllib.parser= urllib.request urllib.errorr rx itertoolsr r typingr r r r r r Zsentry_sdk.utilsr
r$ r r Zsentry_sdk.enveloper
getLoggerr r r Zdjango.utils.deprecationr. Z django.httpr/ r0 r1 Z django.confr2 rF rI rp rc Z
BODY_TAG_NAMEr] zipupperr_ re r9 ImportErrorr r r r r sN
p