jj#&dZddlmZddlZddlZddlZddlZddlZddlZddl Z ddl m Z ddl Z ddl mZmZe j rDddlZddlmZddlmZmZdd lmZe j2d Ze j2d Zed ZGd deZy)zbZeroMQ based urwid EventLoop implementation. `ZeroMQ `_ library is required. ) annotationsN)count) EventLoop ExitMainLoop)Callable)ExecutorFuture) ParamSpecZMQAlarmHandle_T_SpecceZdZdZeZdfd Z ddZddZddZ e jf ddZ e jf ddZ ddZdd Zdd Zdd Zdd Zdd ZddZxZS) ZMQEventLoopac This class is an urwid event loop for `ZeroMQ`_ applications. It is very similar to :class:`SelectEventLoop`, supporting the usual :meth:`alarm` events and file watching (:meth:`watch_file`) capabilities, but also incorporates the ability to watch zmq queues for events (:meth:`watch_queue`). .. _ZeroMQ: https://zeromq.org/ ct|tjtj |j j|_d|_g|_ tj|_ i|_ d|_i|_y)NTr)super__init__logging getLogger__name__getChild __class__logger_did_something_alarmszmqPoller_poller_queue_callbacks _idle_handle_idle_callbacks)selfrs b/home/ubuntu/.cache/pipx/cdb5059130eadca/lib/python3.12/site-packages/urwid/event_loop/zmq_loop.pyrzZMQEventLoop.__init__Asi ''1::4>>;R;RS "JL zz| EGDFc.|j|g|i|S)aRun callable in executor. :param executor: Executor to use for running the function :type executor: concurrent.futures.Executor :param func: function to call :type func: Callable :param args: positional arguments to function :type args: object :param kwargs: keyword arguments to function :type kwargs: object :return: future object for the function call outcome. :rtype: concurrent.futures.Future )submit)r"executorfuncargskwargss r#run_in_executorzZMQEventLoop.run_in_executorKs(xt5d5f55r$ctj|zt|j|f}tj|j ||S)aK Call *callback* a given time from now. No parameters are passed to callback. Returns a handle that may be passed to :meth:`remove_alarm`. :param float seconds: floating point time to wait before calling callback. :param callback: function to call from event loop. )timenext _alarm_breakheapqheappushr)r"secondscallbackhandles r#alarmzZMQEventLoop.alarmas=))+'d.?.?)@(K t||V, r$c |jj|tj|jy#t$rYywxYw)ze Remove an alarm. Returns ``True`` if the alarm exists, ``False`` otherwise. FT)rremover0heapify ValueErrorr"r4s r# remove_alarmzZMQEventLoop.remove_alarmpsB   LL   ' MM$,, '   s:= A A c||jvrtd||jj||||j|<|S)a6 Call *callback* when zmq *queue* has something to read (when *flags* is set to ``POLLIN``, the default) or is available to write (when *flags* is set to ``POLLOUT``). No parameters are passed to the callback. Returns a handle that may be passed to :meth:`remove_watch_queue`. :param queue: The zmq queue to poll. :param callback: The function to call when the poll is successful. :param int flags: The condition to monitor on the queue (defaults to ``POLLIN``). zalready watching )rr9rregister)r"queuer3flagss r# watch_queuezZMQEventLoop.watch_queue~sN* D)) )0 :; ; eU+'/e$ r$ct|trtj|}|jj ||||j |j<|S)a Call *callback* when *fd* has some data to read. No parameters are passed to the callback. The *flags* are as for :meth:`watch_queue`. Returns a handle that may be passed to :meth:`remove_watch_file`. :param fd: The file-like object, or fileno to monitor. :param callback: The function to call when the file has data available. :param int flags: The condition to monitor on the file (defaults to ``POLLIN``). ) isinstanceintosfdopenrr=rfileno)r"fdr3r?s r# watch_filezZMQEventLoop.watch_filesJ( b# 2B b%(-5biik* r$c |jj||jj|dy#|jj|dwxYw#t$rYywxYw)z Remove a queue from background polling. Returns ``True`` if the queue was being monitored, ``False`` otherwise. NFT)r unregisterrpopKeyErrorr:s r#remove_watch_queuezZMQEventLoop.remove_watch_queuesd   8 ''/%%))&$7  %%))&$7  s ;AAA A('A(c |jj||jj|j dy#|jj|j dwxYw#t $rYywxYw)z Remove a file from background polling. Returns ``True`` if the file was being monitored, ``False`` otherwise. NFT)rrJrrKrFrLr:s r#remove_watch_filezZMQEventLoop.remove_watch_filesq   A ''/%%))&--/4@  %%))&--/4@  s"A *A8 ,A55A88 BBcv|xjdz c_||j|j<|jS)z Add a *callback* to be executed when the event loop detects it is idle. Returns a handle that may be passed to :meth:`remove_enter_idle`. r)r r!r"r3s r# enter_idlezZMQEventLoop.enter_idles7 Q2:T../   r$c> |j|=y#t$rYywxYw)zq Remove an idle callback. Returns ``True`` if *handle* was removed, ``False`` otherwise. FT)r!rLr:s r#remove_enter_idlezZMQEventLoop.remove_enter_idles.  $$V,  s  cbt|jjD] }| y)N)listr!valuesrQs r#_entering_idlezZMQEventLoop._entering_idles*T1188:; H J r$ctjt5 |j#tj j $r(}|jtjk7rYd}~Ad}~wwxYw#1swYyxYw)z Start the event loop. Exit the loop when any callback raises an exception. If :exc:`ExitMainLoop` is raised, exit cleanly. N) contextlibsuppressr_looprerrorZMQErrorerrnoEINTR)r"excs r#runzZMQEventLoop.runsl   . JJLyy))yyEKK/0   s1A6.A6A3 A.)A6.A33A66A?cd}|js |jrd}|jr3d}td|jddtjz }|jr!|jr|jr |dkDrd}d}t |j j |dz}n#t |j j }|sS|dk(r|jd|_n6|dk(r1tj|j\}}}|d|_|D]}|j|d|_y ) z7 A single iteration of the event loop. waitrr5gidleiFTN) rrmaxr-dictrpollrXr0heappopr)r"statetimeoutready_due _tie_breakr3r>s r#r\zZMQEventLoop._loops <<4..G||c4<<?1#5 #CD""DLLT\\gXYk**7T>:;E**,-E##%&+#'!-2]]4<<-H*j( &*# 'E (D ! !% ( *"&D  'r$)returnNone) r'r r(zCallable[_Spec, _T]r)z _Spec.argsr*z _Spec.kwargsroz Future[_T])r2floatr3Callable[[], typing.Any]ror )r4r robool)r> zmq.Socketr3rrr?rCrort)rGzint | io.TextIOWrapperr3rrr?rCroio.TextIOWrapper)r4rtrors)r4rurors)r3rrrorC)r4rCrors)r __module__ __qualname____doc__rr/rr+r5r;rPOLLINr@rHrMrOrRrTrXrbr\ __classcell__)rs@r#rr4s7LG66"6 6  6  6,  $ZZ +   >ZZ  "+   4  !  'r$r)rx __future__rrZr_r0rrDr-typing itertoolsrr abstract_looprr TYPE_CHECKINGiocollections.abcrconcurrent.futuresr r typing_extensionsr TypeVarr r rrr$r#rs{( #   2  (3+#V^^$45N  B g Ea'9a'r$