
    ghC                        S SK r S SKJr  S SKrS SKJrJrJr  S SKJrJ	r	  S SK
JrJr  S SKJr  S SKJrJrJr  S SKJrJrJrJrJr  S S	KJr  \(       a*  S S
KJrJrJrJrJr  S SK J!r!J"r"J#r#  S SKJ$r$  \" S\S\4   S9r% S SK&J'r'J(r(J)r)J*r*J+r+  S SK,J-r-J.r.J/r/  \-\.\/4r1 " S S\5      r2S r3S r4S r5S r6S r7g! \0 a	    \" S5      ef = f)    N)datetime)continue_traceget_baggageget_traceparent)OP
SPANSTATUS)DidNotEnableIntegration)should_send_default_pii)BAGGAGE_HEADER_NAMESENTRY_TRACE_HEADER_NAMETransactionSource)capture_internal_exceptionsensure_integration_enabledevent_from_exceptionSENSITIVE_DATA_SUBSTITUTEreraise)TYPE_CHECKING)AnyCallableOptionalUnionTypeVar)EventProcessorEventHint)ExcInfoF.)bound)HueyResultResultGroupTaskPeriodicTask)CancelExecution	RetryTaskTaskLockedExceptionzHuey is not installedc                   2    \ rS rSrSrS\ 3r\S 5       rSrg)HueyIntegration*   hueyzauto.queue.c                  ,    [        5         [        5         g N)patch_enqueuepatch_execute     _/home/kali/devsecops-assessor/venv/lib/python3.13/site-packages/sentry_sdk/integrations/huey.py
setup_onceHueyIntegration.setup_once.   s     	r1   r0   N)	__name__
__module____qualname____firstlineno__
identifieroriginstaticmethodr3   __static_attributes__r0   r1   r2   r)   r)   *   s%    J:,'F r1   r)   c                  n   ^ [         R                  m[        [        T5      U4S j5       n U [         l        g )Nc                 B  > [         R                  " [        R                  UR                  [
        R                  S9   [        U[        5      (       d+  [        [        5       [        [        5       0UR                  S'   T" X5      sS S S 5        $ ! , (       d  f       g = f)N)opnamer:   sentry_headers)
sentry_sdk
start_spanr   QUEUE_SUBMIT_HUEYr@   r)   r:   
isinstancer$   r   r   r   r   kwargs)selftaskold_enqueues     r2   _sentry_enqueue&patch_enqueue.<locals>._sentry_enqueue9   sw     ""##"))

 dL11
 (,o.?1,- t*
 
 
s   AB
B)r    enqueuer   r)   )rJ   rI   s    @r2   r.   r.   5   s.    ,,K=+ >+" #DLr1   c                    ^  U 4S jnU$ )Nc                   > [        5          U R                  S0 5      nTR                  US'   TR                  TR                  :  US'   U R                  S0 5      nTR
                  [        5       (       a  TR                  O[        [        5       (       a  TR                  O[        TR                  =(       d    STR                  -
  S.US'   S S S 5        U $ ! , (       d  f       U $ = f)Ntagshuey_task_idhuey_task_retryextrar   )rH   argsrF   retryzhuey-job)
r   
setdefaultiddefault_retriesretriesr@   r   rS   r   rF   )eventhintrO   rR   rH   s       r2   event_processor._make_event_processor.<locals>.event_processorP   s     )*##FB/D#'77D &*&:&:T\\&ID"#$$Wb1E		 /00 II2 /00 KK2..3!t||C!E* +( ) +*( s   CC
C'r0   )rH   r[   s   ` r2   _make_event_processorr]   N   s    2 r1   c                    [         R                  " 5       nU S   [        ;   a*  UR                  R	                  [
        R                  5        g UR                  R	                  [
        R                  5        [        U [         R                  " 5       R                  [        R                  SS.S9u  p#UR                  X#S9  g )Nr   F)typehandled)client_options	mechanism)rZ   )rB   get_current_scopeHUEY_CONTROL_FLOW_EXCEPTIONStransaction
set_statusr   ABORTEDINTERNAL_ERRORr   
get_clientoptionsr)   r9   capture_event)exc_infoscoperY   rZ   s       r2   _capture_exceptionrn   l   s    ((*E{22$$Z%7%78	  !:!:;&!,,.66*55%HKE
 
)r1   c                 :   ^  [        [        T 5      U 4S j5       nU$ )Nc                     >  T" U 0 UD6nU$ ! [          a,    [        R                  " 5       n[        U5        [	        U6    W$ f = fr-   )	Exceptionsysrl   rn   r   )rS   rF   resultrl   funcs       r2   _sentry_execute+_wrap_task_execute.<locals>._sentry_execute   sO    	4*6*F   	||~Hx(X	s    2AA)r   r)   )rt   ru   s   ` r2   _wrap_task_executerw   }   s&      6	 7	 r1   c                  r   ^ [         R                  m[        [        T5      SU4S jj5       n U [         l        g )Nc           	        > [         R                  " 5        n[        5          SUl        UR	                  5         UR                  [        U5      5        S S S 5        UR                  R                  SS 5      n[        U=(       d    0 UR                  [        R                  [        R                  [        R                   S9nUR#                  [$        R&                  5        [)        USS5      (       d!  [+        UR,                  5      Ul        SUl        [         R0                  " U5         T" XU5      sS S S 5        sS S S 5        $ ! , (       d  f       N= f! , (       d  f       O= f S S S 5        g ! , (       d  f       g = f)Nr+   rA   )r@   r?   sourcer:   _sentry_is_patchedFT)rB   isolation_scoper   _nameclear_breadcrumbsadd_event_processorr]   rF   popr   r@   r   QUEUE_TASK_HUEYr   TASKr)   r:   rf   r   OKgetattrrw   executer{   start_transaction)rG   rH   	timestamprm   rA   re   old_executes         r2   ru   &patch_execute.<locals>._sentry_execute   s    '')U,.$'')))*?*EF /
 "[[__-=tDN($"YY%%(--&--K "":==14!5u==1$,,?*.'--k:"4y9 ;:+ *)..( ;::+ *))s;   E22ECE2*	E3	E2
E	E2
E%	!E22
F r-   )r    _executer   r)   )ru   r   s    @r2   r/   r/      s.    --K=: >:4 $DMr1   )8rr   r   rB   sentry_sdk.apir   r   r   sentry_sdk.constsr   r   sentry_sdk.integrationsr	   r
   sentry_sdk.scoper   sentry_sdk.tracingr   r   r   sentry_sdk.utilsr   r   r   r   r   typingr   r   r   r   r   r   sentry_sdk._typesr   r   r   r   r   huey.apir    r!   r"   r#   r$   huey.exceptionsr%   r&   r'   ImportErrorrd   r)   r.   r]   rn   rw   r/   r0   r1   r2   <module>r      s    
   G G , = 4 
  !>>==(8CH-.A0FFOO
 !0<OP k #2<*"$$Y  0
.
//0s   ?B7 7C