AHjdZddlmZddlZddlmZddlmZddlm Z m Z m Z m Z ddl ZddlmZmZmZe r ddlmZdd lmZej.j1d Gd d Zy)z Contains `WorkerManager`, a class to manage [workers](/guide/workers) for an app. You access this object via [App.workers][textual.app.App.workers] or [Widget.workers][textual.dom.DOMNode.workers]. ) annotationsN)Counter) attrgetter) TYPE_CHECKINGAnyIterableIterator)Worker WorkerStateWorkType)App)DOMNodeT)angularceZdZdZddZddZddZddZddZddZ ddZ d dd Z d d d d d d d d ddZ d dZ d!dZd!dZd"dZd#dZd$d%dZy)& WorkerManagerzAn object to manager a number of workers. You will not have to construct this class manually, as widgets, screens, and apps have a worker manager accessibly via a `workers` attribute. c2||_ t|_y)zVInitialize a worker manager. Args: app: An App instance. N)_appset_workers)selfapps N/root/tools/cai/cai_env/lib/python3.12/site-packages/textual/worker_manager.py__init__zWorkerManager.__init__s  %%(U (c#Kt}|jd|jDt|j D]\}}|j |fyw)Nc34K|]}|jywN)state).0workers r z.WorkerManager.__rich_repr__..,s@v||@s)rupdatersorteditemsname)rcounterrcounts r __rich_repr__zWorkerManager.__rich_repr__*sP(/ @$--@@"7==?3 $LE5**e# # $sAA!cTtt|jtdS)N _created_time)keyiterr#rrrs r__iter__zWorkerManager.__iter__0sF4==j.IJKKrcVtt|jtddS)Nr*T)r+reverser,r.s r __reversed__zWorkerManager.__reversed__3s$ 4==j&A4 P  rc,t|jSr)boolrr.s r__bool__zWorkerManager.__bool__8sDMM""rc,t|jSr)lenrr.s r__len__zWorkerManager.__len__;s4==!!rc||jvSr)rrr s r __contains__zWorkerManager.__contains__>s&&rTc|r2|jr&|j|j|j|jj ||r'|j |j |jyy)zAdd a new worker. Args: worker: A Worker instance. start: Start the worker if True, otherwise the worker must be started manually. exclusive: Cancel all workers in the same group as `worker`. N)group cancel_groupnoderadd_startr_remove_worker)rr start exclusives r add_workerzWorkerManager.add_workerAsW    fkk6<< 8 &!  MM$))T%8%8 9 rdefaultF)r%r= description exit_on_errorrCrDthreadc t|||xst|ddxsd||xs t||| } |j| ||| S)acCreate a worker from a function, coroutine, or awaitable. Args: work: A callable, a coroutine, or other awaitable. name: A name to identify the worker. group: The worker group. description: A description of the worker. exit_on_error: Exit the app if the worker raises an error. Set to `False` to suppress exceptions. start: Automatically start the worker. exclusive: Cancel all workers in the same group. thread: Mark the worker as a thread worker. Returns: A Worker instance. __name__rF)r%r=rHrIrJ)rCrD)r getattrreprrE) rworkr?r%r=rHrIrCrDrJr s r _new_workerzWorkerManager._new_workerQsX8%  <z26<"#1tDz'  eyA rc:|jj|y)z`Remove a worker from the manager. Args: worker: A Worker instance. N)rdiscardr:s rrBzWorkerManager._remove_workerys f%rcr|jD](}|j|j|j*y)zStart all the workers.N)rrArrBr:s r start_allzWorkerManager.start_alls-mm :F MM$))T%8%8 9 :rcF|jD]}|jy)zCancel all workers.N)rcancelr:s r cancel_allzWorkerManager.cancel_allsmm F MMO rc|jDcgc]"}|j|k(r|j|k(r|$}}|D]}|j|Scc}w)zCancel a single group. Args: node: Worker DOM node. group: A group name. Returns: A list of workers that were cancelled. )rr=r?rV)rr?r=r workerss rr>zWorkerManager.cancel_groups`--  %&++*=    F MMO  s'Ac|jDcgc]}|j|k(s|}}|D]}|j|Scc}w)zCancel all workers associated with a given node Args: node: A DOM node (widget, screen, or App). Returns: List of cancelled workers. )rr?rV)rr?r rYs r cancel_nodezWorkerManager.cancel_nodesI)- Mf9L6MM F MMO Ns AANcK tj|xs|Dcgc]}|jc}d{ycc}w7 #tj$rYywxYww)zWait for workers to complete. Args: workers: An iterable of workers or None to wait for all workers in the manager. N)asynciogatherwaitCancelledError)rrYr s rwait_for_completezWorkerManager.wait_for_completesO  ..4"QV6;;="QR R R"Q R%%   s?AA?AAAAAAAAA)rr returnNone)rbzrich.repr.Result)rbzIterator[Worker[Any]])rbr4)rbint)r objectrbr4)TT)r r rCr4rDr4rbrc)rOr r?rr%z str | Noner=strrHrfrIr4rCr4rDr4rJr4rbr )r r rbrc)rbrc)r?rr=rfrb list[Worker])r?rrbrgr)rYzIterable[Worker] | Nonerbrc)rL __module__ __qualname____doc__rr(r/r2r5r8r;rErPrBrTrWr>r[rarrrrs )$ L #"'EI::%):=A: :*"&&&  &  &&&&&& &P&:  &  rr)rj __future__rr] collectionsroperatorrtypingrrrr rich.reprrichtextual.workerr r r textual.appr textual.domrrNautorrkrrrvs[ #9988#]]]r