jjb)dZddlmZddlZddlZddlZddlZddlmZm Z ejdkrddl m Z ejr ddlZddlmZmZmZmZdd lmZmZed Zd ZGd d ej2j4ZGddeZy)zMTrio Runner based urwid EventLoop implementation. Trio library is required. ) annotationsN) EventLoop ExitMainLoop) )BaseExceptionGroup) AwaitableCallableHashableMapping) Concatenate ParamSpec_Spec) TrioEventLoopc$eZdZdZdZddZddZy)_TrioIdleCallbackInstrumentz IDLE callbacks emulation helper.idle_callbacksc||_y)Nr)selfrs c/home/ubuntu/.cache/pipx/cdb5059130eadca/lib/python3.12/site-packages/urwid/event_loop/trio_loop.py__init__z$_TrioIdleCallbackInstrument.__init__7s ,c\|dkDr'|jjD] }| yy)Nr)rvalues)rtimeout idle_callbacks rbefore_io_waitz*_TrioIdleCallbackInstrument.before_io_wait:s1 Q;!%!4!4!;!;!=   rN)rz+Mapping[Hashable, Callable[[], typing.Any]])rfloatreturnNone)__name__ __module__ __qualname____doc__ __slots__rrrrrr2s*#I- rrceZdZdZdfd Z ddZddZddZddZddZ ddZ dd Z dd Z dd Z dd Zdd ZddZddZ ddZ ddZxZS)rzn Event loop based on the ``trio`` module. ``trio`` is an async library for Python 3.5 and later. cDt|tjtj |j j|_d|_i|_ g|_ d|_ tj|_tjj |_y)z Constructor.rN)superrlogging getLoggerr#getChild __class__logger _idle_handle_idle_callbacks_pending_tasks_nurserytriosleep_sleeplowlevel wait_readable_wait_readable)rr/s rrzTrioEventLoop.__init__Gsq ''1::4>>;R;RS DFeg-1 jj "mm99rc<|j|j||S)aCalls `callback()` a given time from now. :param seconds: time in seconds to wait before calling the callback :type seconds: float :param callback: function to call from the event loop :type callback: Callable[[], typing.Any] :return: a handle that may be passed to `remove_alarm()` :rtype: trio.CancelScope No parameters are passed to the callback. ) _start_task _alarm_task)rsecondscallbacks ralarmzTrioEventLoop.alarmUs  0 0'8DDrcv|xjdz c_||j|j<|jS)zCalls `callback()` when the event loop enters the idle state. There is no such thing as being idle in a Trio event loop so we simulate it by repeatedly calling `callback()` with a short delay. r)r1r2)rr?s r enter_idlezTrioEventLoop.enter_idlegs7 Q2:T../   rc$|j|S)zeRemoves an alarm. Parameters: handle: the handle of the alarm to remove  _cancel_scoperhandles r remove_alarmzTrioEventLoop.remove_alarmqs !!&))rc> |j|=y#t$rYywxYw)zuRemoves an idle callback. Parameters: handle: the handle of the idle callback to remove FT)r2KeyErrorrFs rremove_enter_idlezTrioEventLoop.remove_enter_idleys.  $$V,  s  c$|j|S)zRemoves a file descriptor being watched for input. Parameters: handle: the handle of the file descriptor callback to remove Returns: True if the file descriptor was watched, False otherwise rDrFs rremove_watch_filezTrioEventLoop.remove_watch_files!!&))rc@|j }|j|S)zCancels the given Trio cancellation scope. Returns: True if the scope was cancelled, False if it was cancelled already before invoking this function ) cancel_calledcancel)rscopeexisteds rrEzTrioEventLoop._cancel_scopes ))) rct|j} tj|j|gy#t $r}|j |Yd}~yd}~wwxYw)zStarts the event loop. Exits the loop when any callback raises an exception. If ExitMainLoop is raised, exits cleanly. ) instrumentsN)rr2r5run _main_task BaseException_handle_main_loop_exceptionremulate_idle_callbacksexcs rrUzTrioEventLoop.runsQ "=T=Q=Q!R 2 HHT__3I2J K 2  , ,S 1 1 2s": AAAcKt|j} tjj | |j d{tjj |y7$#tjj |wxYw#t$r}|j|Yd}~yd}~wwxYww)aBStarts the main loop and blocks asynchronously until the main loop exits. This allows one to embed an urwid app in a Trio app even if the Trio event loop is already running. Example:: with trio.open_nursery() as nursery: event_loop = urwid.TrioEventLoop() # [...launch other async tasks in the nursery...] loop = urwid.MainLoop(widget, event_loop=event_loop) with loop.start(): await event_loop.run_async() nursery.cancel_scope.cancel() N) rr2r5r8add_instrumentrVremove_instrumentrWrXrYs r run_asynczTrioEventLoop.run_asyncs$"=T=Q=Q!R 2 MM ( ()? @ Hoo''' //0FG( //0FG 2  , ,S 1 1 2sWB>BA3 A1 A3B0B>1A33!BB B; B61B>6B;;B>c<|j|j||S)abCalls `callback()` when the given file descriptor has some data to read. No parameters are passed to the callback. Parameters: fd: file descriptor to watch for input callback: function to call when some input is available Returns: a handle that may be passed to `remove_watch_file()` )r< _watch_task)rfdr?s r watch_filezTrioEventLoop.watch_files 0 0"h??rc~K|5|j|d{|dddy7#1swYyxYww)a0Asynchronous task that sleeps for a given number of seconds and then calls the given callback. Parameters: scope: the cancellation scope that can be used to cancel the task seconds: the number of seconds to wait callback: the callback to call N)r7)rrQr>r?s rr=zTrioEventLoop._alarm_tasks? ++g& & & J   &  s =1/ 1 =1:=c|jjt|tr't |j dk(r|j d}t|t ry|j|jd)a5Handles exceptions raised from the main loop, catching ExitMainLoop instead of letting it propagate through. Note that since Trio may collect multiple exceptions from tasks into an ExceptionGroup, we cannot simply use a try..catch clause, we need a helper function like this. rrN) r2clear isinstancer len exceptionsrwith_traceback __traceback__)rr[s rrXz)TrioEventLoop._handle_main_loop_exceptionse ""$ c- .3s~~3F!3K..#C c< (   !2!23=rc&K tj4d{|_|jtjd{dddd{d|_y7M77#1d{7swY xYw#d|_wxYww)zzMain Trio task that opens a nursery and then sleeps until the user exits the app by raising ExitMainLoop. N)r5 open_nurseryr4_schedule_pending_tasks sleep_forever)rs rrVzTrioEventLoop._main_tasks !((* + +dm,,.((*** + +!DM  +* + + + +!DMssBBA*B-A0 A,A0 BA.B"B*B,A0.B0B6A9 7B>B BBc|jD]$\}}}|jj||g|&|jjy)zSchedules all pending asynchronous tasks that were created before the nursery to be executed on the nursery soon. N)r3r4 start_soonrf)rtaskrQargss rrnz%TrioEventLoop._schedule_pending_taskssM"&!4!4 9 D% $DMM $ $T5 84 8 9 !!#rctj}|jr |jj||g||S|jj |||f|S)apStarts an asynchronous task in the Trio nursery managed by the main loop. If the nursery has not started yet, store a reference to the task and the arguments so we can start the task when the nursery is open. Parameters: task: a Trio task to run Returns: a cancellation scope for the Trio task )r5 CancelScoper4rqr3append)rrrrsrQs rr<zTrioEventLoop._start_task s]   " == $DMM $ $T5 84 8     & &eT': ; rcK|5|js-|j|d{||js-dddy7!#1swYyxYww)aPAsynchronous task that watches the given file descriptor and calls the given callback whenever the file descriptor becomes readable. Parameters: scope: the cancellation scope that can be used to cancel the task fd: the file descriptor to watch callback: the callback to call N)rOr:)rrQrbr?s rrazTrioEventLoop._watch_task!sY ))))"--- ))    .   s+A!A AA  AA  AA)r!r")r>r r?Callable[[], typing.Any]r!trio.CancelScope)r?rxr!int)rGryr!bool)rGrzr!r{)rQryr!r{)rbint | io.IOBaser?rxr!ry)rQryr>r r?rxr!r")r[rWr!r")rrz9Callable[Concatenate[trio.CancelScope, _Spec], Awaitable]rsz _Spec.argsr!ry)rQryrbr|r?rxr!r")r#r$r%r&rr@rBrHrKrMrErUr_rcr=rXrVrnr<ra __classcell__)r/s@rrr@s :EE+E  E$!*  *  22:@ @+@  @"+   $> !$G  . +   rr)r& __future__rr,systypingr5 abstract_looprr version_infoexceptiongroupr TYPE_CHECKINGiocollections.abcr r r r typing_extensionsrrr__all__abc Instrumentrrr(rrrsw* # 2g1  FF8 g E   $(("5"5  uIur