B
z‹dñ ã @ sf d Z ddlmZmZmZmZ dgZddlZddlZddl m
Z
ddl m Z dd„ Z
G d d„ de
ƒZdS )
a×
Python's multiprocessing.Pool class doesn't interact well with
``KeyboardInterrupt`` signals, as documented in places such as:
* ``_
* ``_
* ``_
Various workarounds have been shared. Here, we adapt the one proposed in the
last link above, by John Reese, and shared as
* ``_
Our version is a drop-in replacement for multiprocessing.Pool ... as long as
the map() method is the only one that needs to be interrupt-friendly.
Contributed by Peter K. G. Williams .
*Added in version 2.1.0*
é )ÚdivisionÚprint_functionÚabsolute_importÚunicode_literalsÚInterruptiblePoolN)ÚPool)Ú TimeoutErrorc G s$ t t jt j¡ | dk r | |Ž dS )zÚ
We ignore SIGINT. It's up to our parent to kill us in the typical
condition of this arising from ``^C`` on a terminal. If someone is
manually killing us with that signal, well... nothing will happen.
N)ÚsignalÚSIGINTÚSIG_IGN)Zactual_initializerÚrest© r
úg/work/yifan.wang/ringdown/master-ringdown-env/lib/python3.7/site-packages/ptemcee/interruptible_pool.pyÚ_initializer_wrapper$ s r c s0 e Zd ZdZdZd ‡ fdd„ Zd
dd„Z‡ ZS ) r aŽ
A modified version of :class:`multiprocessing.pool.Pool` that has better
behavior with regard to ``KeyboardInterrupts`` in the :func:`map` method.
:param processes: (optional)
The number of worker processes to use; defaults to the number of CPUs.
:param initializer: (optional)
Either ``None``, or a callable that will be invoked by each worker
process when it starts.
:param initargs: (optional)
Arguments for *initializer*; it will be called as
``initializer(*initargs)``.
:param kwargs: (optional)
Extra arguments. Python 2.7 supports a ``maxtasksperchild`` parameter.
i Nr
c s( t t|¡}tt| ƒj|||f|Ž d S )N)Ú functoolsÚpartialr Úsuperr Ú__init__)ÚselfZ processesZ initializerZinitargsÚkwargsZnew_initializer)Ú __class__r
r r F s zInterruptiblePool.__init__c C s^ | |||¡}xJy | | j¡S tk
r0 Y q tk
rT | ¡ | ¡ ‚ Y qX qW dS )a
Equivalent of ``map()`` built-in, without swallowing
``KeyboardInterrupt``.
:param func:
The function to apply to the items.
:param iterable:
An iterable of items that will have `func` applied to them.
N)Z map_asyncÚgetÚ wait_timeoutr ÚKeyboardInterruptÚ terminateÚjoin)r ÚfuncÚiterableÚ chunksizeÚrr
r
r ÚmapL s zInterruptiblePool.map)NNr
)N)Ú__name__Ú
__module__Ú __qualname__Ú__doc__r r r Ú
__classcell__r
r
)r r r 0 s )r$ Ú
__future__r r r r Ú__all__r r Zmultiprocessing.poolr Úmultiprocessingr r r r
r
r
r Ú s