U
|
Z ±dO ã @ sf d dl Z d dlmZ d dlmZ d dlZddlmZmZm Z edd Z
|
G dd d Z ed
|
d Z dS ) é N)Úcontextmanager)Ú OrderedDicté )Ú signal_raiseÚis_main_threadÚConflictDetectorc c sR i }z(t| D ]}t ||¡||< qd V W 5 | ¡ D ]\}}t ||¡ q6X d S ©N)ÚitemsÚsignalÚset)ÚsignalsÚhandlerZoriginal_handlersÚsignumÚoriginal_handler© r úDd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\trio/_signals.pyÚ_signal_handler, s
|
r c @ s< e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd S )ÚSignalReceiverc C s( t | _tj ¡ | _td| _d| _d S )Nz8only one task can iterate on a signal receiver at a timeF) r Ú_pendingÚtrioÚlowlevelZ
|
ParkingLotÚ_lotr Ú_conflict_detectorÚ_closed©Úselfr r r Ú__init__9 s ÿzSignalReceiver.__init__c C s( | j rt| nd | j|< | j ¡ d S r )r r r r Zunpark)r r r r r Ú_addB s
|
|
zSignalReceiver._addc s d_ fdd d S )NTc s2 j r.j jdd\} }z t| W 5 X d S )NF©Úlast)r Úpopitemr ©r Ú_©Ú deliver_nextr r r r$ Q s
|
z9SignalReceiver._redeliver_remaining.<locals>.deliver_next)r r r r# r Ú_redeliver_remainingI s z#SignalReceiver._redeliver_remainingc C s
|
t | jS r )Úlenr r r r r Ú_pending_signal_count\ s z$SignalReceiver._pending_signal_countc C s | S r r r r r r Ú __aiter___ s zSignalReceiver.__aiter__c
|
à sj | j rtd| jL | js.| j ¡ I d H ntj ¡ I d H | jj dd\}}|W 5 Q R £ S Q R X d S )Nz)open_signal_receiver block already exitedFr )
|
r Ú RuntimeErrorr r r Zparkr r Ú
|
checkpointr )r r r" r r r Ú __anext__b s zSignalReceiver.__anext__N) Ú__name__Ú
|
__module__Ú __qualname__r r r% r' r( r+ r r r r r 8 s r c ' sh | s t dt stdtj ¡ t fdd}z t| | V W 5 Q R X W 5 ¡ X dS )aÞ A 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 threadc s j j| dd d S )NT)Z
|
idempotent)Z run_sync_soonr r! ©ÚqueueÚtokenr r r s z%open_signal_receiver.<locals>.handlerN) Ú TypeErrorr r) r r Zcurrent_trio_tokenr r% r )r r r r/ r Úopen_signal_receiverq s $ÿ
|
r3 ) r
|
Ú
|
contextlibr Ú collectionsr r Z_utilr r r r r r3 r r r r Ú<module> s &
|
9
|
