eϸ d dl mZ d dlmZ d dlmZ d dlmZmZ d dl m
Z
m Z d dl m
Z
d dlmZmZ d dlmZmZ d d lmZmZmZ d d
lmZ d d lmZ d d l m Z d d
l m Z d dl m!Z!m"Z"m#Z# er0d dl m$Z$m%Z%m&Z&m'Z'm(Z( d dl)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2 d dlm3Z3 d dl4m5Z5 n# e6$ r e
d w xY wdZ7 G d de
Z8 G d de Z9d)dZ:d)d Z;d*d Zd%e?d!d"d&d'ddfd(Z@dS ), )
TYPE_CHECKING) BaseModel)OP)Hub_should_send_default_pii) DidNotEnable Integration)SentryAsgiMiddleware)SOURCE_FOR_STYLETRANSACTION_SOURCE_ROUTE)event_from_exceptiontransaction_from_function)RequestStarliteState)BaseRouteHandler)DefineMiddleware)get_plugin_for_value) HTTPRoute)ConnectionDataExtractoris_async_callableRef)AnyDictListOptionalUnion) ASGIAppHTTPReceiveMessage HTTPScopeMessage
MiddlewareReceiveScopeSendWebSocketReceiveMessage)MiddlewareProtocol)EventzStarlite is not installedzgeneric Starlite requestc e Zd Zd fd Z xZS ) SentryStarliteASGIMiddlewareappr c R t |ddd d S )NFendpointasgi)r+ unsafe_context_datatransaction_stylemechanism_type)super__init__)selfr+ __class__s r/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib/python3.11/site-packages/sentry_sdk/integrations/starlite.pyr3 z%SentryStarliteASGIMiddleware.__init__* s:
% (!
)r+ r )__name__
__module__ __qualname__r3
__classcell__)r5 s @r6 r* r* ) s=
r7 r* c * e Zd ZdZedd ZdS )StarliteIntegrationstarlitereturnNc Z t t t d S )N)patch_app_initpatch_middlewarespatch_http_route_handle r7 r6
setup_oncez StarliteIntegration.setup_once6 s, !!!!!r7 r? N)r8 r9 r:
identifier staticmethodrE rD r7 r6 r= r= 3 s7 J" " " \" " "r7 r= r? Nc D t j d fd } | t _ dS )
a
Replaces the Starlite class's `__init__` function in order to inject `after_exception` handlers and set the
`SentryStarliteASGIMiddleware` as the outmost middleware in the stack.
See:
- https://starlite-api.github.io/starlite/usage/0-the-starlite-app/5-application-hooks/#after-exception
- https://starlite-api.github.io/starlite/usage/7-middleware/0-middleware-intro/
r4 r argsr kwargsr? Nc . | dg }| t gt |t r|n|g t
j t
_ | dd pg }t
g||d< | g|R i | d S )Nafter_exception)rM
middleware)popupdateexception_handler
isinstancelistr*
_run_asgi3__call__)r4 rJ rK rM rN old__init__s r6 injection_wrapperz)patch_app_init..injection_wrapperG s **%6;;
! "/488+OO)*
1M0W$-ZZ d339r
< Jz J| D*4***6*****r7 )r4 r rJ r rK r r? N)r r3 )rW rV s @r6 rA rA = s; #K+ + + + + +$ *Hr7 c D t j dfd } | t _ d S )Nr4 r r? List[Middleware]c . d | D S )Nc , g | ]}t | S rD )enable_span_for_middleware).0rN s r6
zIpatch_middlewares..resolve_middleware_wrapper..` s0
'z 2 2
r7 rD )r4 old__resolve_middleware_stacks r6 resolve_middleware_wrapperz5patch_middlewares..resolve_middleware_wrapper_ s/
; ;D A A
r7 )r4 r r? rY )r resolve_middleware)r` r_ s @r6 rB rB \ s< $4$G!
+E'''r7 rN r" c t | d r | t u r| S t | t r
| j j n| j d
fd }j d v}|r)t | t r
|| j _ n|| _ | S )NrU r4 r' scoper$ receiver# sendr% r? c 8 K t j }| t }|| j j | t j 5 }| d dfd }t d t }|d
k } | s|n}
dfd
} t d t } | dk }
|
s| n} | ||
| d {V cd d d S # 1 swxY w Y d S | | d {V S )Nop descriptionstarlite.middleware_namerJ r rK r? 2Union[HTTPReceiveMessage, WebSocketReceiveMessage]c
K t j }| t j t dt
5 }| d | i | d {V cd d d S # 1 swxY w Y d S Nr: rg rj )r current
start_spanr MIDDLEWARE_STARLITE_RECEIVEgetattrstrset_tag)rJ rK hubspanmiddleware_namerd s r6 _sentry_receivezNenable_span_for_middleware.._create_span_call.._sentry_receive s +C 9$+G^S\\$R$R ( > %? QQQ%,Wd%=f%=%= = = = = = = > > > > > > > > > > > > > > > > > >s
$A>>BBr8 rw messager! c
K t j }| t j t dt
5 }| d | d {V cd d d S # 1 swxY w Y d S rm )r rn ro r MIDDLEWARE_STARLITE_SENDrq rr rs )rx rt ru rv re s r6 _sentry_sendzKenable_span_for_middleware.._create_span_call.._sentry_send s +C 6$+D.#d))$L$L ( 3 %? QQQ%)T']] 2 2 2 2 2 2 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3s
'BBBr{ )rJ r rK r r? rk )rx r! r? N) r rn get_integrationr= r5 r8 ro r MIDDLEWARE_STARLITErs rq rr )r4 rc rd re rt integrationmiddleware_spanrw receive_namereceive_patched new_receiver{ send_name send_patchednew_sendrv old_calls `` @r6 _create_span_callz5enable_span_for_middleware.._create_span_callt s k))*=>> " "n 5O ) #
J ''(BOTTT > >,1 >I > > > > > > > 'w
CLL I I ".2C"C5D Qoo' 3 3 3 3 3 3 3 $D*c$ii @ @ (N : /;E<< %XdE; I IIIIIIIG#
J #
J #
J #
J #
J #
J #
J #
J #
J #
J #
J #
J #
J #
J #
J #
J #
J #
JJ "$w======== =s BC;;C?C?)r )
r4 r' rc r$ rd r# re r% r? N)hasattrr* rR r rN rU r8 )rN r not_yet_patchedr s @r6 r\ r\ h s J
+ + 5 5 5* .// ' ( 3 < &,>",>+2,>=F,>NT,>
,> ,> ,> ,> ,> ,>\ '/DDO 4 j"2 3 3 4->J ! * *"3J r7 c X t j d fd
} | t _ d S ) Nr4 r rc r rd r# re r% r? c
K t j }| t }| | || d {V S | 5 }d || } t
dd |
d }| d {V d
fd } t j |_ |
| | || d {V cd d d S # 1 swxY w Y d S )
Nr+ )rc rd re T)
parse_body parse_querybodyeventr( _Dict[str, Any]r? c d }| di }t dd |d< t r d |d< |d< d }|j |j }n.t |j t r
|j j }n|j }|t | }dt d i}|st }dt i}| |||
| S ) N
route_handlerrequest_bodyr7 content_lengthcookiesdatasourcer- )r transactiontransaction_info)
getlenr namerR fnr valuer r _DEFAULT_TRANSACTION_NAMEr rP )
r r r request_infofunctx_nametx_infoextracted_request_data request_datarc s
r6 event_processorzHpatch_http_route_handle..handle_wrapper..event_processor s % / : :
$yyB 7 7 14UYYw5L5L1M1M -.+-- P.DY.OL+ ++7L( %1 + 0GG
0#66 ,(+1DD(+D# 7 = =G#%5j%AB C 7G ')A BG (gPW r7 )r r( r r r? r( ) r rn r| r= configure_scope
request_classr rO rG _nameadd_event_processor)
r4 rc rd re rt r~ sentry_scoper r r r r
old_handles
` @@r6 handle_wrapperz/patch_http_route_handle..handle_wrapper s k+.+>+>?R+S+S #D%$???????? ?
"
" - @l+0<+E+EW4 ,F , ,G&%< T& & && & " *--f55D!%::::::L
> "5!?L , ,_ = = =#D%$????????[- @ - @ - @ - @ - @ - @ - @ - @ - @ - @ - @ - @ - @ - @ - @ - @ - @ - @s BC;;C?C?)
r4 r rc r rd r# re r% r? N)r handle)r r s @r6 rC rC sm !J5@5@"-5@8A5@IO5@
5@ 5@ 5@ 5@ 5@ 5@n &Ir7 rc r$ Optional[Dict[str, Any]]c v | di }|sd S t |t r|S t |t r| S t |d r| S t
| }|r)t |j s| | S d S )Nuserasdict) r rR dictr r r r r to_dict)rc
scope_userplugins r6 retrieve_user_from_scoper s 62&&J t*d## *i(( ! z8$$ # """
!*
-
-F
*'77 *~~j))) 4r7 excr r c t j }| t d S d }t rt | }|rVt
|t rA| 5 }| | d d d n # 1 swxY w Y t | |j r |j j nd t j
dd \ }}| || d S )NF)typehandled)client_options mechanism)hint)r rn r| r= r r rR r r set_userr
clientoptionsrG
capture_event)r rc r rt user_infor r r s r6 rQ rQ s4
+C
.//7,0I !! 4,U33 -Z 400 -
"
" -l ! !) , , , - - - - - - - - - - - - - - - ' -0ZAsz))T.9eLL KE4 e$'''''s 3BB BrF )rN r" r? r" )rc r$ r? r )Atypingr pydanticr sentry_sdk.constsr sentry_sdk.hubr r sentry_sdk.integrationsr r sentry_sdk.integrations.asgir
sentry_sdk.tracingr r sentry_sdk.utilsr
r r> r r r starlite.handlers.baser starlite.middlewarer starlite.plugins.baser starlite.routes.httpr starlite.utilsr r r r r r r r starlite.typesr r r r! r" r# r$ r% r& r' sentry_sdk._typesr( ImportErrorr r* r= rA rB r\ rC r ExceptionrQ rD r7 r6 r s. 8 8 8 8 8 8 8 8 = = = = = = = = = = = = = = I I I I I I I I L L L L L L L L41111111111777777444444::::::......NNNNNNNNNN ,;;;;;;;;;;;;;;
0/////++++++ 4 4 4
,2
3
334 7
#7
" " " " "+ " " " * * * *> E E E EB B B BJ:& :& :& :&z $(9 (W ( (T ( ( ( ( ( (s A B B*