R3idZddlZddlmZddlmZmZddlmZmZm Z m Z m Z m Z ddl mZddlmZerdd lmZGd d e ZGd d e ZGddZgdZy)z&Progress tracking for task executions.N)contextmanager)datetimetimezone) TYPE_CHECKINGAnyAsyncGenerator GeneratorLiteral TypedDict)suppress_instrumentation)Self)DocketcbeZdZUeded<eed<edzed<eed<edzed<edzed<y) ProgressEventprogresstypekeyNcurrenttotalmessage updated_at)__name__ __module__ __qualname__r __annotations__strintd/mnt/e/genesis-system/.venvs/voice-bridge/lib/python3.12/site-packages/docket/_execution_progress.pyrrs5 *  H 4Z J 4Zd r rc|eZdZUeded<eed<eed<eed<edzed<edzed<edzed<edzed <y) StateEventstaterrwhenNworker started_at completed_aterror)rrrr rrrr r!r#r#sD '  H J I $Jd * :r r#ceZdZdZdddeddfdZededfd Ze dddede fd Z d e ddfd Z dd e ddfdZdedzddfdZddZddZdeeefddfdZdeedffdZy)ExecutionProgressaManages user-reported progress for a task execution. Progress data is stored in Redis hash {docket}:progress:{key} and includes: - current: Current progress value (integer) - total: Total/target value (integer) - message: User-provided status message (string) - updated_at: Timestamp of last update (ISO 8601 string) This data is ephemeral and deleted when the task completes. docketrrreturnNc||_||_|jd||_d|_d|_d|_d|_y)zInitialize progress tracker for a specific task. Args: docket: The docket instance key: The task execution key progress:Nr )r,r _redis_keyrrrr)selfr,rs r!__init__zExecutionProgress.__init__/sE  **y%67#'  #' +/r )NNNc#K|jjst5ddddydy#1swYyxYww)zASuppress OTel auto-instrumentation for internal Redis operations.N)r,enable_internal_instrumentationr )r1s r!_maybe_suppress_instrumentationz1ExecutionProgress._maybe_suppress_instrumentation>s<{{::)+      s!A6A?AcTK|||}|jd{|S7w)aCreate and initialize progress tracker by reading from Redis. Args: docket: The docket instance key: The task execution key Returns: ExecutionProgress instance with attributes populated from Redis N)sync)clsr,rinstances r!createzExecutionProgress.createGs-vs#mmo s (&(rcK|dkr tdtjtj}|j }|j j4d{}|j|jt||dd{dddd{||_ ||_ |j||dd{y7p7A73#1d{7swYCxYw7 w)zSet the total/target value for progress tracking. Args: total: The total number of units to complete. Must be at least 1. r zTotal must be at least 1N)rrmapping) ValueErrorrnowrutc isoformatr,redishsetr0rrr_publish)r1r updated_at_dtrrBs r! set_totalzExecutionProgress.set_totalVs 1978 8 X\\2 ",,. ;;$$&  %** Z",     'mme:FGGG       HsfA"C2$C%C2(-CCC C2%C&)C2C0C2CC2C-!C$ "C-)C2amountcK|dkr tdtjtj}|j }|j j4d{}|j|jd|d{}|j|jd|d{dddd{|_ ||_ |j||dd{y77f7C75#1d{7swYExYw7"w)zAtomically increment the current progress value. Args: amount: Amount to increment by. Must be at least 1. r zAmount must be at least 1Nrr)rr)r>rr?rr@rAr,rBhincrbyr0rCrrrD)r1rGrErrB new_currents r! incrementzExecutionProgress.incrementos A:89 9 X\\2 ",,. ;;$$&  % % dooy& QQK**    # 'mm :NOOO Q      PsxA"D $C.%D (!C6 C0 $C6.C2/C63 D >C4?)D (D )D 0C62C64D 6D<C? =DD rcKtjtj}|j }|j j 4d{}|j|j||dd{dddd{||_ ||_ |j||dd{y7g7A73#1d{7swYCxYw7 w)z{Update the progress status message. Args: message: Status message describing current progress N)rrr<) rr?rr@rAr,rBrCr0rrrD)r1rrErrBs r! set_messagezExecutionProgress.set_messages ! X\\2 ",,. ;;$$&  %**&",     'mmzJKKK       LsfACB<C$C<B>=C C C )C6C7C>CCCC CCcK|j5|jj4d{}|j|jd{}|rt |j dd|_t |j dd|_d|vr|djnd|_ d|vr&tj|djnd|_ nd|_d|_d|_ d|_ dddd{dddy777#1d{7swY!xYw#1swYyxYww) zSynchronize instance attributes with current progress data from Redis. Updates self.current, self.total, self.message, and self.updated_at with values from Redis. Sets attributes to None if no data exists. Nscurrent0stotals100smessages updated_atd)r5r,rBhgetallr0rgetrrdecoderr fromisoformatr)r1rBdatas r!r7zExecutionProgress.syncs*  1 1 3 +{{((* + +e"]]4??;;#&txx D'A#BDL!$TXXh%?!@DJ5?45GZ(//1TL )D0!..tM/B/I/I/KL!O $(DL!$DJ#'DL&*DO# + + + + +; + + + + + +stED<D!D<D'D#B2D' D<D%D< E!D<#D'%D<'D9 -D0.D9 5D<<EEcpK|j5|jj4d{}|j|jd{dddd{dddd|_d|_d|_d|_y7\7;7-#1d{7swY=xYw#1swYAxYww)zgDelete the progress data from Redis. Called internally when task execution completes. NrP) r5r,rBdeleter0rrrr)r1rBs r!rWzExecutionProgress.deletes  1 1 3 4{{((* 4 4ell4??333 4 4 4     43 4 4 4 4 4 4ssB6B*BB*BBB B*%B&B**%B6B*BB*B' BB' #B**B3/B6rUcjK|jjd|j}d|j|j |jnd|j|j|j dd}|jj |tj|d{y7w)zPublish progress update to Redis pub/sub channel. Args: data: Progress data to publish (partial update) r/rNrr)rrrrrr) r,rrrrrRrDjsondumps)r1rUchannelpayloads r!rDzExecutionProgress._publishs ++//IdhhZ"8988'+||'?t||QZZ||((<0 " kk""7DJJw,?@@@sB)B3+B1,B3cK|jjd|j}|jj4d{}|j|d{|j 23d{}|ddk(st j |d.7[7D7-6dddd{7y#1d{7swYyxYww)a~Subscribe to progress updates for this task. Yields: Dict containing progress update events with fields: - type: "progress" - key: task key - current: current progress value - total: total/target value (or None) - message: status message (or None) - updated_at: ISO 8601 timestamp r/NrrrU)r,r_pubsub subscribelistenrYloads)r1r[pubsubrs r!r_zExecutionProgress.subscribes++//IdhhZ"89;;&&( 6 6F""7+ + +!' 6 6g6?i/**WV_55  6 + 6 6 6 6 6 6sAC B% C B>"B'#B>7B+;B)<B+? B> B>%C'B>)B++B>, C7B:8C>CC C C)r )r-N)rrr__doc__rr2rr r5 classmethodrr:rrFrKrMr7rWdictrrDrrr_rr r!r+r+#s  0x 0c 0d 0;K1L (    HSHTH2PcP$P0LtLL,+4 A4S>AdA"6 t0C!D6r r+)r+rr#)rcrY contextlibrrrtypingrrrr r r _telemetryr typing_extensionsrr,rrr#r+__all__rr r!rksO, %'TT0"ID6D6N r