o US`cO@sfddlZddlmZddlmZddlZddlmZmZm Z eddZ Gdd d Z ed d Z dS) N)contextmanager) OrderedDict) signal_raiseis_main_threadConflictDetectorc cspi}z$t|D] }t||||<qdVW|D] \}}t||qdS|D] \}}t||q,wN)setsignalitems)signalshandlerZoriginal_handlerssignumoriginal_handlerr8/usr/local/lib/python3.10/dist-packages/trio/_signals.py_signal_handler,s rc@s<eZdZddZddZddZddZd d Zd d Zd S)SignalReceivercCs(t|_tj|_td|_d|_dS)Nz8only one task can iterate on a signal receiver at a timeF) r_pendingtriolowlevelZ ParkingLot_lotr_conflict_detector_closedselfrrr__init__9s   zSignalReceiver.__init__cCs*|jr t|dSd|j|<|jdSr)rrrrZunpark)rrrrr_addBs  zSignalReceiver._addcsd_fdddS)NTcs:jrjjdd\}}z t|WdSwdS)NFlast)rpopitemrr_ deliver_nextrrrr$Qs  z9SignalReceiver._redeliver_remaining..deliver_next)rrrr#r_redeliver_remainingIs z#SignalReceiver._redeliver_remainingcCs t|jSr)lenrrrrr_pending_signal_count\s z$SignalReceiver._pending_signal_countcCs|Srrrrrr __aiter___szSignalReceiver.__aiter__csx|jrtd|j'|js|jIdHntjIdH|jj dd\}}|WdS1s5wYdS)Nz)open_signal_receiver block already exitedFr) r RuntimeErrorrrrZparkrr checkpointr )rrr"rrr __anext__bs$zSignalReceiver.__anext__N) __name__ __module__ __qualname__rrr%r'r(r+rrrrr8s  rc's|stdtstdtjtfdd}z&t|| VWdn1s1wYWdSWdSw)aA context manager for catching signals. Entering this context manager starts listening for the given signals and returns an async iterator; exiting the context manager stops listening. The async iterator blocks until a signal arrives, and then yields it. Note that if you leave the ``with`` block while the iterator has unextracted signals still pending inside it, then they will be re-delivered using Python's regular signal handling logic. This avoids a race condition when signals arrives just before we exit the ``with`` block. Args: signals: the signals to listen for. Raises: TypeError: if no signals were provided. RuntimeError: if you try to use this anywhere except Python's main thread. (This is a Python limitation.) Example: A common convention for Unix daemons is that they should reload their configuration when they receive a ``SIGHUP``. Here's a sketch of what that might look like using :func:`open_signal_receiver`:: with trio.open_signal_receiver(signal.SIGHUP) as signal_aiter: async for signum in signal_aiter: assert signum == signal.SIGHUP reload_configuration() zNo signals were providedz]Sorry, open_signal_receiver is only possible when running in Python interpreter's main threadcsjj|dddS)NT)Z idempotent)Z run_sync_soonrr!queuetokenrrr sz%open_signal_receiver..handlerN) TypeErrorrr)rrZcurrent_trio_tokenrrr%)r r rr/ropen_signal_receiverqs"$   r3) r contextlibr collectionsrrZ_utilrrrrrr3rrrrs  &  9