xjv=<ddlmZddlmZddlZddlmZmZddlm Z ddl m Z m Z m Z ddlmZddlmZdd lmZdd lmZmZdd lmZdd lmZmZmZmZ dd lmZm Z ddl!m"Z"ddl#m$Z$ddl%m&Z&ddl'm(Z(ddl)m*Z*n#e+$r e dwxYwddl,m-Z-e-r6ddl,m.Z.m/Z/m0Z0ddl'm1Z1ddl2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9ddl2m:Z;ddlm?Z?m@Z@dZAGdde ZBGddeZCd-d!ZDd-d"ZEd.d%ZFd-d&ZGd/d*ZHeeBd+eId'd(d dfd,ZJdS)0)Set)deepcopyN)OPSPANDATA)%_apply_key_value_collection_filtering)$_DEFAULT_FAILED_REQUEST_STATUS_CODES DidNotEnable Integration)SentryAsgiMiddleware) ignore_logger)should_send_default_pii)SOURCE_FOR_STYLETransactionSource)has_span_streaming_enabled)ensure_integration_enabledevent_from_exceptionhas_data_collection_enabledtransaction_from_function)LitestarRequest)ConnectionDataExtractor) HTTPException)BaseRouteHandler)DefineMiddleware) HTTPRoutezLitestar is not installed) TYPE_CHECKING)AnyOptionalUnion)MiddlewareProtocol)HTTPReceiveMessage HTTPScopeMessage MiddlewareReceiveSendWebSocketReceiveMessage)Scope)ASGIApp)EventHintzgeneric Litestar requestcFeZdZdZdeZef d dZed dZdS) LitestarIntegrationlitestarz auto.http.failed_request_status_codesSet[int]returnNc||_dS)N)r/)selfr/s QC:\Projects\Historical\venv\Lib\site-packages\sentry_sdk/integrations/litestar.py__init__zLitestarIntegration.__init__?s,G(((cxttttddS)Nr.)patch_app_initpatch_middlewarespatch_http_route_handler r6r4 setup_oncezLitestarIntegration.setup_onceEs<!!! j!!!!!r6)r/r0r1Nr1N) __name__ __module__ __qualname__ identifieroriginrr5 staticmethodr<r;r6r4r-r-;sxJ &* & &F3WG%/G GGGG  " " "\ " " "r6r-cJeZdZejfdddeddffd ZdeddfdZxZ S) SentryLitestarASGIMiddlewareappr) span_originr1NcVt|ddd|ddS)NFendpointasgi)rFunsafe_context_datatransaction_stylemechanism_typerG asgi_version)superr5)r3rFrG __class__s r4r5z%SentryLitestarASGIMiddleware.__init__WsB  %(!#      r6exccdS)zAvoid catching exceptions from request handlers. Those exceptions are already handled in Litestar.after_exception handler. We still catch exceptions from application lifespan handlers. Nr;)r3rRs r4_capture_request_exceptionz7SentryLitestarASGIMiddleware._capture_request_exceptioncs r6) r>r?r@r-rBstrr5 ExceptionrT __classcell__)rQs@r4rErEVs1D1K     +.          i D        r6rEr1cztjttd fd }|t_dS) a\ Replaces the Litestar class's `__init__` function in order to inject `after_exception` handlers and set the `SentryLitestarASGIMiddleware` as the outmost middleware in the stack. See: - https://docs.litestar.dev/2/usage/applications.html#after-exception - https://docs.litestar.dev/2/usage/middleware/using-middleware.html r3rargsrkwargsr1Nctg|dpg|d<|dpg}tg||d<|g|Ri|dS)Nafter_exception middleware)exception_handlergetrE)r3rYrZr] old__init__s r4injection_wrapperz)patch_app_init..injection_wrappervs % jj*++1r%  ! ZZ --3 <JzJ| D*4***6*****r6)r3rrYrrZrr1N)rr5rr-)rar`s @r4r8r8lsR#K 3[AA+++++BA+*Hr6cztjttdfd }|t_dS)Nr3rr1list[Middleware]c.d|DS)Nc,g|]}t|Sr;)enable_span_for_middleware).0r]s r4 zIpatch_middlewares..resolve_middleware_wrapper..s0    'z 2 2   r6r;)r3old_resolve_middleware_stacks r4resolve_middleware_wrapperz5patch_middlewares..resolve_middleware_wrappers/  ::4@@    r6)r3rr1rc)rresolve_middlewarerr-)rjris @r4r9r9sT#3#F  35QRR     SR +E'''r6r]r$ct|dr |tur|St|tr |jjn|j d fd }jd v}|r)t|tr ||j_n||_|S)N__call__r3r scope LitestarScopereceiver%sendr&r1c Ktj t||d{VS|jjt jr"tj ||d{VStj tj tj d5}|tj d fd }t#d t%}|d k}|s|n}d fd } t#d t%} | dk} | s| n} |||| d{VcdddS#1swxYwYdStj tj tj 5}|d d fd }t#d t%}|d k}|s|n}d fd } t#d t%} | dk} | s| n} |||| d{VcdddS#1swxYwYdS)Nz sentry.opz sentry.originname attributesrYrrZr12Union[HTTPReceiveMessage, WebSocketReceiveMessage]c Kt|i|d{VStj|i|d{VStjt dttj tj d5}| tj |i|d{VcdddS#1swxYwYdSNr@rsrt)get_integrationr- sentry_sdktracesget_current_span start_spangetattrrUrMIDDLEWARE_LITESTAR_RECEIVErB set_attributerMIDDLEWARE_NAMErYrZspanclientmiddleware_namerps r4_sentry_receivezNenable_span_for_middleware.._create_span_call.._sentry_receivesk--.ABBJ%,Wd%=f%=%========!(99;;C%,Wd%=f%=%========#*55$Wnc'llKK)+)G-@-G$$6> **8+C_UUU%,Wd%=f%=%=======>>>>>>>>>>>>>>>>>>s,.C''C+.C+r>rmessager#c Kt|d{VStj|d{VStjt dttj tj d5}| tj |d{VcdddS#1swxYwYdSry)rzr-r{r|r}r~rrUrMIDDLEWARE_LITESTAR_SENDrBrrrrrrrrqs r4 _sentry_sendzKenable_span_for_middleware.._create_span_call.._sentry_sendsT--.ABBJ%)T']]2222222!(99;;C%)T']]2222222#*55$T>3t99EE)+)D-@-G$$63 **8+C_UUU%)T']]222222333333333333333333s21C00C47C4roprurBlitestar.middleware_namec bKt|i|d{VStjtjt dttj5}| d|i|d{VcdddS#1swxYwYdSNr@rr) rzr-r{r~rrrrUrBset_tagrs r4rzNenable_span_for_middleware.._create_span_call.._sentry_receives--.ABBJ%,Wd%=f%=%========#.9$Wnc'llKK29> %?QQQ%,Wd%=f%=%======= >>>>>>>>>>>>>>>>>>s3$B$$B(+B(c nKt|d{VStjtjt dttj5}| d|d{VcdddS#1swxYwYdSr) rzr-r{r~rrrrUrBrrs r4rzKenable_span_for_middleware.._create_span_call.._sentry_sends --.ABBJ%)T']]2222222#.6$T>3t99EE293 %?QQQ%)T']]222222 333333333333333333s6'B**B.1B.)rYrrZrr1rw)rr#r1N)r{ get_clientrzr-rQr>roptionsr|r}r~rMIDDLEWARE_LITESTARrBrrrrrUr)r3rnrprqmiddleware_spanr receive_namereceive_patched new_receiver send_name send_patchednew_sendrrold_calls `` @@r4_create_span_callz5enable_span_for_middleware.._create_span_calls &((  ! !"5 6 6 >!$w======== =.1 %fn 5 5a J 1133;%XdE7DAAAAAAAAA"--$!#!7%8%?.3 J !--h.FXXX> >,1>I>>>>>>>>" 'w CLLII ".2C"C5DQoo'  3 3 3 3 3 3 3 3$D*c$ii@@ (N: /;E<<%XdE;IIIIIIIIg3 J3 J3 J3 J3 J3 J3 J3 J3 J3 J3 J3 J3 J3 J3 J3 J3 J3 Jj&)$*1) J!''(BOTTT >  >,1 >I > > > > > > > > 'w CLLII ".2C"C5DQoo'  3 3 3 3 3 3 3 3$D*c$ii@@ (N: /;E<<%XdE;IIIIIIIIS) J) J) J) J) J) J) J) J) J) J) J) J) J) J) J) J) J) Js&BE??FF7BIII)r) r3r rnrorpr%rqr&r1N)hasattrrE isinstancerr]rmr>)r]rnot_yet_patchedrs @r4rfrfs J + + 5 5 5*.//'(3<&lJ"lJlJlJ lJ  lJlJlJlJlJlJ\'/DDO4 j"2 3 3 4->J ! * *"3J  r6cXtj d fd }|t_dS) Nr3rrnr"rpr%rqr&r1cB Ktj t|||d{VStj}d||}t dd| d}|d{Vd}d}|j |j } n)t|j dr |j j }n|j }|t|} td} | st} t j} tj| | || | d fd } tj|_|| |||d{VS)NrF)rnrprqT) parse_body parse_querybody route_handlervaluerIeventr*_r+r1c|di}tdd|d<d}tjrEt djdd}|r||d<d jdd v}nt r d|d<|r|d <t ||d<|S) Nrequest_bodyr6content_lengthTcookiesdata_collection)items behaviourincoming_request http_bodiesdata)r_lenrrrr r) rr request_infoshould_attach_request_bodyrrextracted_request_data request_datarns r4event_processorzHpatch_http_route_handle..handle_wrapper..event_processorAs 99Y33L-07C1H1H-I-IL) *)- &*6>:: L?0;$n->? J6.5L+'~&78GH+*)** L*@*K Y'',F''3 V$' 55E) Lr6)rr*rr+r1r*)r{rrzr-get_isolation_scope request_classrpopr_rurfnrrr_DEFAULT_TRANSACTION_NAMErROUTEset_transaction_namerA_nameadd_event_processor)r3rnrprq sentry_scoperrrfuncrusourcerrrr old_handles ` @@@r4handle_wrapperz/patch_http_route_handle..handle_wrappers&((  ! !"5 6 6 >#D%$???????? ?!577 ',U|'A'At(B( ( " !8" " " ""&))&11!zzzzzz  /22    ) %DD ]%w / / $ #)DD #D  ,T22D!*- -,D&,F'f555))$777         41; ((999ZeWd;;;;;;;;;r6) r3rrnr"rpr%rqr&r1N)rhandle)rrs @r4r:r:sm!JE<E<"-E<8AE224 + ,[ 9 8077I " "4,U33 )Z 400)!577 i(((#}%% +--==>QRR  #{'NNN F& !,..6.9eLLKE4 U......r6r=)r]r$r1r$)rnror1r)Kcollections.abcrcopyrr{sentry_sdk.constsrrsentry_sdk.data_collectionrsentry_sdk.integrationsrr r sentry_sdk.integrations.asgir sentry_sdk.integrations.loggingr sentry_sdk.scoper sentry_sdk.tracingrrsentry_sdk.tracing_utilsrsentry_sdk.utilsrrrrr.rrlitestar.data_extractorsrlitestar.exceptionsrlitestar.handlers.baserlitestar.middlewarerlitestar.routes.httpr ImportErrortypingrrrrr litestar.typesr!r"r#r$r%r&r'r(rolitestar.types.asgi_typesr)sentry_sdk._typesr*r+rr-rEr8r9rfr:rrVr^r;r6r4rs********LLLLLL >=====999999444444BBBBBBBB??????4********@@@@@@111111777777444444.......444 ,2 3 334! .++++++++++666666211111--------6"""""+"""6     #7   ,****0 E E E EBBBBJJ&J&J&J&Z/00/9/_////10///s &A==B