KiWUdZddlZddlZddlZddlZddlmZmZddlm Z ddl m Z m Z m Z mZmZmZmZmZmZddlZddlZddlmZddlmZddlmZd d lmZmZd d l m!Z!e rd d l"m#Z#d d l$m%Z%ejLe'Z(ejRe*d<e+e,e,fZ-Gdde.ej^Z0edZ1GddejdZ3Gdde3Z4Gdde3Z5e6e0efZ7e+e.e8e7fZ9e+e.e9fZ:GddZ;y)aStrike list with optional Redis synchronization. This module provides the StrikeList class which manages strike conditions for blocking task execution. When connected to Redis, it monitors a stream for strike/restore instructions issued by external processes. N)AsyncExitStackcontextmanager) TracebackType) TYPE_CHECKINGAnyCallable GeneratorHashableLiteralMappingNoReturncast)Redis) RedisCluster)Self)CANCEL_MSG_CLEANUP cancel_task)suppress_instrumentation) Execution)RedisConnectionloggerc(eZdZdZdZdZdZdZdZdZ y) Operator==!=>>=<<=betweenN) __name__ __module__ __qualname__EQUAL NOT_EQUAL GREATER_THANGREATER_THAN_OR_EQUAL LESS_THANLESS_THAN_OR_EQUALBETWEEN]/home/jay/workspace/scripts/.codegraph-venv/lib/python3.12/site-packages/docket/strikelist.pyrr-s' EIL IGr-r)rrrrrr r!c eZdZUeded<eed<dedzdedzdededdf d Zde fd Z e d e ddfd Z de eeffd ZdefdZy)StrikeInstructionstrikerestore directionoperatorfunctionN parametervaluereturnc<||_||_||_||_yN)r6r7r5r8)selfr6r7r5r8s r.__init__zStrikeInstruction.__init__>s ! "   r-c`d|jji}|jr|jj|d<|jr|jj|d<|jj|d<t j |j|d<|S)N directionfunction parameteroperatorvalue)r4encoder6r7r5 cloudpickledumpsr8)r<messages r. as_messagezStrikeInstruction.as_messageJs'3T^^5J5J5L&M ==#'==#7#7#9GK >>$(NN$9$9$;GL !#}}335 '--djj9r-rGcbttd|dj}d|vr|djnd}d|vr|djnd}tt|dj}t j |d}|dk(rt ||||St||||S)Nr1r?r@rArBrCr2)rr decoderrEloadsStrikeRestore)clsrGr4r6r7r5r8s r. from_messagezStrikeInstruction.from_messageTs!45w|7L7S7S7UV 4?74J7;'..0PT6Bg6MGL)002SW '+"6"="="?@!!'("34  (Ix? ?8Y%@ @r-ci}|jr|j|d<|jr6|j|d<|j|d<t|j|d<|S)Nz docket.taskzdocket.parameterzdocket.operatorz docket.valuer6r7r5reprr8)r<labelss r.rSzStrikeInstruction.labels`sY!# ==$(MMF= ! >>)-F% &(, F$ %%)$**%5F> " r-c |jxsdd|jxsdd|jd|jrt|jdSddS)N*( )rQr<s r. call_reprzStrikeInstruction.call_reprlsi}}#$ ~~$% }}o #'>>tDJJ;   8;;   r-)r"r#r$r __annotations__rstrr r=MessagerH classmethodrOr rSrZr,r-r.r0r0:s*++ * :      G A7 A/B A A S)   3  r-r0c"eZdZUdZeded<y)rLr2r1r4Nr"r#r$r4r r[r,r-r.rLrLys.6Iw*+6r-rLc"eZdZUdZeded<y)rMr3r1r4Nr`r,r-r.rMrM}s.7Iw*+7r-rMc eZdZUdZeed<eed<eedge fed<e dzed<e je dzed<e jdzed <eed < d:d edzd ed e ddfdZedefdZedefdZededfdZdefdZdeedzdedzdedzddfdZdedge fddfdZdedge fddfdZd;dZ de!ddfdZ" dz'StrikeList.__aenter__..sWT3Dd%Kr-z - strike monitor)rlctddS)NrhrxrYsr.rzz'StrikeList.__aenter__..sWT?D%Ir-)rrj __aenter__rg is_connectedenter_async_contextasyncioEventricallback create_task_monitor_strikesrlrhpush_async_callbackrrrYs`r.r|zStrikeList.__aenter__s$& kk$$&&& ;; K;;++J-JJ+kk--dkk:::&}} KL$00  ! ! #TYYK7H*I  IJ '' ++-?  ' ' ;s#-EE AEE  CE Eexc_type exc_value tracebackcnK |jj|||d{|`y7#|`wxYww)z5Async context manager exit - closes Redis connection.N)rj __aexit__)r<rrrs r.rzStrikeList.__aexit__s8 ++'')YG G G  H s 5 .,.5.25 conditionc<|jjd|y)zCAdds a temporary condition that indicates an execution is stricken.rN)rfinsertr<rs r. add_conditionzStrikeList.add_condition s 9-r-cZ||jusJ|jj|y)zFRemoves a temporary condition that indicates an execution is stricken.N)rorfremovers r.remove_conditionzStrikeList.remove_conditions* F FFFF  *r-cnK|jy|jjd{y7w)alWait for all existing strikes to be loaded from the stream. This method blocks until the strike monitor has completed its initial non-blocking read of all existing strike messages. Call this before making decisions that depend on the current strike state. If not connected to Redis (local-only mode), returns immediately. N)riwaitrYs r.wait_for_strikes_loadedz"StrikeList.wait_for_strikes_loadeds/    ' ""'')))s +535 instructioncK|j|jjs td|jj4d{}|j |j |j d{dddd{|j|y7X7(7#1d{7swY*xYww)zSend a strike instruction to Redis and update local state. Args: instruction: The Strike or Restore instruction to send. Raises: RuntimeError: If not connected to Redis. NzeCannot send strike instruction: not connected to Redis. Use connect() or async context manager first.)rgr} RuntimeErrorclientxaddrsrHupdate)r<rrs r.send_instructionzStrikeList.send_instruction!s ;; dkk&>&>@  ;;%%' D D1&&+*@*@*BC C C D D K  D C D D D DsZA CB'C.B-B)B- CB+C)B-+C-B?3B6 4B?;Cr6r7r5zOperator | LiteralOperatorr8cpKt||t||}|j|d{y7w)aGIssue a strike to block matching tasks or parameters. Args: function: Task function name to strike, or None for all tasks. parameter: Parameter name to match, or None for entire task. operator: Comparison operator for the value. value: Value to compare against. N)rLrrr<r6r7r5r8rs r.r2zStrikeList.strike5s1Xy(82DeL ##K000 ,646cpKt||t||}|j|d{y7w)a6Restore a previously issued strike. Args: function: Task function name to restore, or None for all tasks. parameter: Parameter name to match, or None for entire task. operator: Comparison operator for the value. value: Value to compare against. N)rMrrrs r.r3zStrikeList.restoreGs1h 8H3EuM ##K000rtargetzExecution | Mapping[str, Any]cttr|jStfd|jDS)a4Check if a target matches any strike condition. Args: target: Either an Execution object (for Docket/Worker use) or a dictionary of parameter names to values (for standalone use). Returns: True if any parameter matches a strike condition. c3.K|] }|ywr;r,).0rrs r. z)StrikeList.is_stricken..hsG9V$Gs) isinstancer _is_dict_strickenanyrf)r<rs `r. is_strickenzStrikeList.is_strickenYs9 fg &))&1 1Gd6F6FGGGr-paramsc|jD]A\}}||jvr|j|D]\}}|j|||syCy)zCheck if a parameter dict matches any strike condition. Args: params: Dictionary of parameter names to values. Returns: True if any parameter matches a strike condition. TF)itemsre _is_match)r<rr7argumentr5 strike_values r.rzStrikeList._is_dict_strickenjsf$*<<> Ix 6 66*.*@*@*K &,>>(HlC  r- executionczddlm}|j}|jj |i}||jvr|sy||j } |j |ji|j}|ji|j|jjDcic]\}}||jvr||c}}} | jD]A\} } ||jfD]+} | | vr| | D]\} }|j| | |sy-Cy#t$rYywxYwcc}}w)Nr) get_signatureTF)rr function_namerdgetr6bindargskwargsapply_defaults TypeError argumentsrrer)r<rrrrd signature bound_argskv all_argumentsr7r strike_sourcer5rs r.roz,StrikeList._matches_task_or_parameter_strike}sa,!// ((,,]B? D-- -l!)"4"45  'L9;K;KLJ  % % '  "" &,,224AqJ0001 $1#6#6#8 $ Ix".0F0F!G $ M1.;I.F$*Hl~~h,G#$  $ $-   s6D(3D7( D43D4rc8 |xdk(r||k(Sxdk(r||k7Sxdk(r||kDSxdk(r||k\Sxdk(r||kSxdk(r||kSdk(r|\}}||cxkxr|kScS td|#ttf$rtjd |||d Yy wxYw) z1Determines if a value matches a strike condition.rrrrrr r!zUnknown operator: z0Incompatible type for strike condition: %r %s %rT)exc_infoF) ValueErrorrrwarning)r<r8r5rloweruppers r.rzStrikeList._is_matchs  L00 L00 <// L00 <// L00#/LE5 E2U2222$'9(%DEEI&  NNB   s< A- A- A- A- A- A-A-A-A--)BBc, t|jt |tr|j|yt |tr|j|yy#t$r.tj d|j |jYywxYw)Nz-Incompatible type for strike condition: %s %r) hashr8rrrr5rrL_strikerM_restore)r<rs r.rzStrikeList.updates  "" # k6 * LL %  W - MM+ &.  NN?$$!!    sA4BBr2c|jr^|jrR |j|j} ||j}|j |j |jfy|jr |j|j}y|jrB |j|j}|j |j |jfyy#t$rix}|j|j<YwxYw#t$rt x}||j<YwxYw#t$rix}|j|j<YywxYw#t$r&t x}|j|j<YwxYwr;) r6r7rdKeyErrorsetaddr5r8re)r<r2rdres r.rzStrikeList._strikes{ ??v// G#00A  K$01A1A$B!  ! !6??FLL"A B __ G#00A    U$($:$:6;K;K$L!  ! !6??FLL"A B ! GDFF t00A G  KEHUJ!L1A1A$B K GDFF t00A G  UORuT!D$:$:6;K;K$L UsFC!D 8D3E!$DD "D0/D03$EE,F  F r3c|jr|jr |j|j} ||j} |j |j |jf|sF|j |jd|s'|jj |jdyyy|jrD |j|j}|s'|jj |jdyy|jrl |j|j} |j |j |jf|s'|jj |jdyyy#t$rYywxYw#t$r|j |jdYywxYw#t$rYXwxYw#t$rYywxYw#t$rYywxYw#t$rYwxYwr;) r6r7rdrpoprr5r8re)r<r3rdres r.rzStrikeList._restores    1 1 #001A1AB  $01B1B$C!  !(('*:*:GMM)JK%  !2!2D9#%%))'*:*:DA$%    #001A1AB  !!%%g&6&6=    $($:$:7;L;L$M! !(('*:*:GMM)JK%&&**7+<+?KLmmA&&& '  !;<mmA&&& 'sD0A=A#A=A'A%A'#A=%A''A9-A0 .A95A=<D0=A(D+%C(&D++D0-5D+"D%#D+(D0*D++D0rrrstrikes_in_effectcK|j5|j|j|id|rdndd{}ddds0|s.d}|jJ|jj ||fS|sJ|rH|j r8|j j drtjdd{||fS|D]\}}|D]\}} |}tj| } |j| tjd| jd k(rd nd | j|j!| jd k(rd nd d|j"i| j%||fS7@#1swY@xYw7ͭw)zRead and process strike messages from Redis stream. Returns: Tuple of (last_id, initial_load_complete) to allow state persistence. di`N)countblockTz memory://g?z%s %rr2Striking Restoringrz docket.name)ruxreadrsrirrk startswithrrr0rOrrinfor4rZrrlrS) r<rrrrstreams_messages message_idrGrs r.rzStrikeList._read_strikes7s 1 1 3 GG'* 5f4$G  4$( !''3 33  $ $ &11 10xxDHH// <mmC(((11 1" KAx'/ # G$/<E;E>A=F =F >B=F ;E>>F F )N strikelistF)r9N)NNrN)7r"r#r$__doc__ TaskStrikesr[ParameterStrikeslistrboolrrTaskr rrr\r=propertyrqrsrr rurr|type BaseExceptionrrrrrr0rr r2r3rr rrrorrrrLrrMrrrrtuplerr,r-r.rcrcs+!F''h }d2344 d ""<<)D00]]T))  05 $ 4Z$$*. $  $4   (C((;K1L$0 }%, !4' !4'   .x t0C'D..+(K=$3F*G+D+ *!2C!!, $ $15!% 1*1:1/ 1 $ 1  1( $ $15!% 1*1:1/ 1 $ 1  1$H"AHdH"S(9d&%;%4%NshcdB'"3'' CfCC8-D-DD-D^''.6. < 6.6. $ 6.  6. sDy  6.r-rc)r s  5   .":0$#***844 ue| sDIIEF< < ~7 788h() S//03(() g.g.r-