R3iT" HdZddlZddlZddlZddlZddlZddlZddlmZm Z ddl m Z ddl m Z m Z mZddlmZmZddlZddlmZddlmZd d lmZmZd e eeefd ed eeeefdzdffdZGddeej<ZGddeej<Z de d efdZ!d efdZ"dee zd e fdZ#de d dfdZ$ded dfdZ%ded efdZ&ded edzfdZ'dedzd efd Z(d!e de d efd"Z)e Gd#d$Z*d%ed eee*ffd&Z+ d,d'ed(eed)edzd dfd*Z,gd+Z-y)-z%Support functions for the docket CLI.N)AsyncGenerator AsyncIterator) dataclass)datetime timedeltatimezone)Any Collection)Console)Table)DocketSnapshot WorkerInfoiteratortimeoutreturnc0Kddl} |j|j|d{.7#|j$rdYt$rYnwxYw |j d{7y#|j d{7wxYww)aIterate over an async iterator with timeout, ensuring proper cleanup. Wraps an async iterator to add timeout support and guaranteed cleanup. On timeout, yields None to allow the caller to handle polling fallback. Args: iterator: An async generator (must have __anext__ and aclose methods) timeout: Timeout in seconds for each iteration Yields: Items from the iterator, or None if timeout expires rN)r)asynciowait_for __anext__ TimeoutErrorStopAsyncIterationaclose)rrrs ]/mnt/e/genesis-system/.venvs/voice-bridge/lib/python3.12/site-packages/docket/_cli_support.pyiterate_with_timeoutrs   #,,X-?-?-A7,SSSS''  %  oohooslBA8$979A89A A8AA8AA8B1A42B8B B BBc eZdZdZdZdZdZdZy)LogLevelDEBUGINFOWARNINGERRORCRITICALN)__name__ __module__ __qualname__rrr r!r"rrr3s E DG EHr'rceZdZdZdZdZy) LogFormatrichplainjsonN)r#r$r%RICHPLAINJSONr&r'rr)r);s D E Dr'r)whenc@|jjdS)Nz%Y-%m-%d %H:%M:%S %z) astimezonestrftime)r0s r local_timer4As ??  % %&< ==r'cXtjdtjS)N#)socket gethostnameosgetpidr&r'rdefault_worker_namer;Es#  " #1RYY[M 22r' duration_strcHt|tr|Sd|vr}|jd}t|dk(r t t |\}}t||St|dk(r"t t |\}}}t|||St d||jdrtt |dd  S|jd rtt |dd  S|jd rtt |dd Stt | S)z Parse a duration string into a timedelta. Supported formats: - 123 = 123 seconds - 123s = 123 seconds - 123m = 123 minutes - 123h = 123 hours - 00:00 = mm:ss - 00:00:00 = hh:mm:ss :)minutesseconds)hoursr@rAzInvalid duration string: sNrAmr@h)rC) isinstancersplitlenmapint ValueErrorendswith)r<partsr@rArCs rdurationrRIs, * l""3' u:?"3 GWWg> > Z1_&)#uo #E7G5'7K K8GH H   s #\#2%6!788   s #\#2%6!788   s #s<#4566\!233r'formatcntj}|tjk(rWddlm}|dd}tj tj}|j||j|y|tjk(r^tj tj}tjdd }|j||j|ydd l m}|}tjd d }|j||j|y) Nr) JsonFormatterz-{name}{asctime}{levelname}{message}{exc_info}{style)streamz4[%(asctime)s] %(levelname)s - %(name)s - %(message)sz%Y-%m-%d %H:%M:%S)datefmt) RichHandlerz %(message)sz[%X])logging getLoggerr)r/pythonjsonlogger.jsonrU StreamHandlersysstdout setFormatter addHandlerr. Formatter rich.loggingr[)rS root_loggerrU formatterhandlerr[s rset_logging_formatrils##%K 7! ;3 ''szz:Y'w' 9?? "''szz:%% B'  Y'w',-%%mVD Y'w'r'levelc^tjj|jyN)r\r]setLevelvalue)rjs rset_logging_levelros   -r'urlcR|jdrtjd|S)z Validate that the provided URL is compatible with the CLI. The memory:// backend is not compatible with the CLI as it doesn't persist across processes. z memory://zThe memory:// URL scheme is not supported by the CLI. The memory backend does not persist across processes. Please use a persistent backend like Redis or Valkey.) startswithtyper BadParameter)rps r validate_urlrus0 ~~k"   D  Jr'rnc|dvry|S)N)*r&)rns rhandle_strike_wildcardrys  Lr'c.|y|jd\}}}|s|S|jd\}}}tj|xsd}t||}|turt t |S|t ur|jdk(S||S)Nr>.builtinsrFtrue) rpartition importlib import_modulegetattrrrNboollower)rntype_ module_name member_namemodulemembers rinterpret_python_valuers }%%c*ND!U  "&//#"6KK  $ $[%>J ?F V[ )FU,, 4{{}&&e}r'nowc||z }|td krdt|S|tdkrd| S|tdkr|dSdt|S)NrHzat rzin  ago)rr4)rr0deltas r relative_timerst $JE "%%%Z%&'' 1 eVH~ 2& &~Z%&''r'ceZdZUdZdZeed<dZeed<dZeed<dZ e dzed<dZ e dzed<dZ e dzed <dZ e dzed <y) TaskStatsz&Statistics for a single task function.rrunningqueuedtotalN oldest_queued latest_queuedoldest_startedlatest_started)r#r$r%__doc__rrN__annotations__rrrrrrrr&r'rrrsY0GSFCOE3N%)M8d?)%)M8d?)&*NHtO*&*NHtO*r'rsnapshotci}|jD]}|j}||vr t||<||}|xjdz c_|xjdz c_|j}|j }|||kr||_|j }|||kDs||_|jD]}|j}||vr t||<||}|xjdz c_|xjdz c_|j}|j}|||kr||_ |j}|||kDs||_ |S)z?Get task count statistics by function name with timestamp data.r ) r function_namerrstartedrrfuturerr0rr) rstats execution func_name task_statsroldestlatestr0s rget_task_statsrsb"$E%%0 ++ E !({E) 9% aA##** >Wv-(/J %** >Wv-(/J %0$__, ++ E !({E) 9% QA~~)) >TF]'+J $)) >TF]'+J $," Lr' docket_nameworkershighlight_taskc$t|dd}td|}|jdd|jd d |jd d tjt j }|D]q}||jz }t|jDcgc]}||k(rd |dn|} }|j|j|d| rdj| ndst} | j|ycc}w)Nc|jSrl) last_seen)ws rzprint_workers..s 1;;r'T)keyreversezWorkers in Docket: )titleNamecyanrWz Last SeengreenTasksyellowz[bold]z[/bold]r z(none))sortedr add_columnrrrutcrtasksadd_rownamejoinr print) rrrsorted_workerstablerworkertime_agotaskrconsoles r print_workersrs G)>MN -k]; rs+  9!22" . DcN+ 6; DcNT)4/0 <sDIITYY >X>#>3S3 43? 4y 4F(y(T(8.X.$. c c  ##* #*, (x (x (C (  + +  +(^(S)^0D(\"&  #$J B r'