�
{ fc @ sT d g Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l m Z m Z m Z d d l
m Z m Z d Z
d Z d Z e j � Z d � Z d e f d
� � YZ e f e d � Z d e f d � � YZ d
e f d � � YZ e Z d e f d � � YZ d e f d � � YZ d e f d � � YZ d e f d � � YZ d S( t Pooli����N( t Processt cpu_countt TimeoutError( t Finalizet debugi i i c C s
t | � S( N( t map( t args( ( s, /usr/lib64/python2.7/multiprocessing/pool.pyt mapstar@ s t MaybeEncodingErrorc B s) e Z d Z d � Z d � Z d � Z RS( sV Wraps possible unpickleable errors, so they can be
safely sent through the socket.c C sA t | � | _ t | � | _ t t | � j | j | j � d S( N( t reprt exct valuet superR t __init__( t selfR R ( ( s, /usr/lib64/python2.7/multiprocessing/pool.pyR K s c C s d | j | j f S( Ns( Error sending result: '%s'. Reason: '%s'( R R ( R ( ( s, /usr/lib64/python2.7/multiprocessing/pool.pyt __str__P s c C s d t | � S( Ns <MaybeEncodingError: %s>( t str( R ( ( s, /usr/lib64/python2.7/multiprocessing/pool.pyt __repr__T s ( t __name__t
__module__t __doc__R R R ( ( ( s, /usr/lib64/python2.7/multiprocessing/pool.pyR G s c C s� | d k s6 t | � t t f k r0 | d k s6 t � | j } | j } t | d � rt | j j � | j
j � n | d k r� | | � n d } xI| d k s� | r�| | k r�y
| � } Wn" t t f k
r� t
d � Pn X| d k r t
d � Pn | \ } }
} } }
y t | | |
� f } Wn t k
rO} t | f } n Xy | | |
| f � WnM t k
r�} t | | d � } t
d | � | | |
t | f f � n Xd } } } } } }
| d 7} q� Wt
d | � d S( Ni t _writers) worker got EOFError or IOError -- exitings worker got sentinel -- exitingi s0 Possible encoding error while sending result: %ss worker exiting after %d tasks( t Nonet typet intt longt AssertionErrort putt gett hasattrR t closet _readert EOFErrort IOErrorR t Truet Exceptiont FalseR ( t inqueuet outqueuet initializert initargst maxtasksR R t completedt taskt jobt it funcR t kwdst resultt et wrapped( ( s, /usr/lib64/python2.7/multiprocessing/pool.pyt workerX sB 6
!
c B s e Z d Z e Z d d d d d � Z d � Z d � Z d � Z d � Z d i d � Z
d d � Z d d � Z d d
� Z
d i d d � Z d d d � Z e d
� � Z e d � � Z e d � � Z e d � � Z d � Z d � Z d � Z d � Z e d � � Z e d � � Z RS( sH
Class which supports an async version of the `apply()` builtin
c C s/ | j � t j � | _ i | _ t | _ | | _ | | _ | | _ | d k r| y
t
� } Wq| t k
rx d } q| Xn | d k r� t d � � n | d k r� t
| d � r� t d � � n | | _ g | _ | j � t j d t j d | f � | _ t | j _ t | j _ | j j � t j d t j d | j | j | j | j | j f � | _ t | j _ t | j _ | j j � t j d t j d | j | j | j f � | _ t | j _ t | j _ | j j � t! | | j" d | j | j# | j | j | j | j | j | j f d d �| _$ d S( Ni s& Number of processes must be at least 1t __call__s initializer must be a callablet targetR t exitpriorityi (% t
_setup_queuest Queuet
_taskqueuet _cachet RUNt _statet _maxtasksperchildt _initializert _initargsR R t NotImplementedErrort
ValueErrorR t TypeErrort
_processest _poolt _repopulate_poolt threadingt ThreadR t _handle_workerst _worker_handlerR# t daemont startt
_handle_taskst
_quick_putt _outqueuet
_task_handlert _handle_resultst
_quick_gett _result_handlerR t _terminate_poolt _inqueuet
_terminate( R t processesR( R) t maxtasksperchild( ( s, /usr/lib64/python2.7/multiprocessing/pool.pyR � sX
c C sw t } xj t t t | j � � � D]M } | j | } | j d k r" t d | � | j � t } | j | =q" q" W| S( s� Cleanup after any worker processes which have exited due to reaching
their specified lifetime. Returns True if any workers were cleaned up.
s cleaning up worker %dN(
R% t reversedt ranget lenRE t exitcodeR R t joinR# ( R t cleanedR. R4 ( ( s, /usr/lib64/python2.7/multiprocessing/pool.pyt _join_exited_workers� s "
c
C s� x� t | j t | j � � D]� } | j d t d | j | j | j | j | j
f � } | j j | � | j j
d d � | _ t | _ | j � t d � q Wd S( s� Bring the number of pool processes up to the specified number,
for use after reaping workers which have exited.
R6 R R t
PoolWorkers added workerN( RZ RD R[ RE R R4 RU RO R? R@ R> t appendt namet replaceR# RK RL R ( R R. t w( ( s, /usr/lib64/python2.7/multiprocessing/pool.pyRF � s #
c C s | j � r | j � n d S( sE Clean up any exited workers and start replacements for them.
N( R_ RF ( R ( ( s, /usr/lib64/python2.7/multiprocessing/pool.pyt _maintain_pool� s c C sP d d l m } | � | _ | � | _ | j j j | _ | j j j | _ d S( Ni ( t SimpleQueue(
t queuesRf RU RO R t sendRN R t recvRR ( R Rf ( ( s, /usr/lib64/python2.7/multiprocessing/pool.pyR8 � s
c C s. | j t k s t � | j | | | � j � S( s1
Equivalent of `apply()` builtin
( R= R< R t apply_asyncR ( R R/ R R0 ( ( s, /usr/lib64/python2.7/multiprocessing/pool.pyt apply� s c C s. | j t k s t � | j | | | � j � S( s/
Equivalent of `map()` builtin
( R= R< R t map_asyncR ( R R/ t iterablet chunksize( ( s, /usr/lib64/python2.7/multiprocessing/pool.pyR � s i c s� | j t k s t � | d k rf t | j � � | j j � � f d � t | � D� � j f � � S| d k sx t � t j
� | | � } t | j � � | j j � f d � t | � D� � j f � d � � D� Sd S( sZ
Equivalent of `itertools.imap()` -- can be MUCH slower than `Pool.map()`
i c 3 s0 | ]& \ } } � j | � | f i f Vq d S( N( t _job( t .0R. t x( R/ R1 ( s, /usr/lib64/python2.7/multiprocessing/pool.pys <genexpr> s c 3 s0 | ]& \ } } � j | t | f i f Vq d S( N( Ro R ( Rp R. Rq ( R1 ( s, /usr/lib64/python2.7/multiprocessing/pool.pys <genexpr>
s c s s" | ] } | D] } | Vq
q d S( N( ( Rp t chunkt item( ( s, /usr/lib64/python2.7/multiprocessing/pool.pys <genexpr> s N( R= R< R t IMapIteratorR; R: R t enumeratet _set_lengthR t
_get_tasks( R R/ Rm Rn t task_batches( ( R/ R1 s, /usr/lib64/python2.7/multiprocessing/pool.pyt imap� s c s� | j t k s t � | d k rf t | j � � | j j � � f d � t | � D� � j f � � S| d k sx t � t j
� | | � } t | j � � | j j � f d � t | � D� � j f � d � � D� Sd S( sK
Like `imap()` method but ordering of results is arbitrary
i c 3 s0 | ]& \ } } � j | � | f i f Vq d S( N( Ro ( Rp R. Rq ( R/ R1 ( s, /usr/lib64/python2.7/multiprocessing/pool.pys <genexpr> s c 3 s0 | ]& \ } } � j | t | f i f Vq d S( N( Ro R ( Rp R. Rq ( R1 ( s, /usr/lib64/python2.7/multiprocessing/pool.pys <genexpr> s c s s" | ] } | D] } | Vq
q d S( N( ( Rp Rr Rs ( ( s, /usr/lib64/python2.7/multiprocessing/pool.pys <genexpr>! s N( R= R< R t IMapUnorderedIteratorR; R: R Ru Rv R Rw ( R R/ Rm Rn Rx ( ( R/ R1 s, /usr/lib64/python2.7/multiprocessing/pool.pyt imap_unordered s c C sV | j t k s t � t | j | � } | j j | j d | | | f g d f � | S( s>
Asynchronous equivalent of `apply()` builtin
N( R= R< R t ApplyResultR; R: R Ro R ( R R/ R R0 t callbackR1 ( ( s, /usr/lib64/python2.7/multiprocessing/pool.pyRj # s +c s� | j t k s t � t | d � s3 t | � } n | d k r} t t | � t | j � d � \ } } | r} | d 7} q} n t | � d k r� d } n t j
| | | � } t | j | t | � | � � | j
j � f d � t | � D� d f � � S( s<
Asynchronous equivalent of `map()` builtin
t __len__i i i c 3 s0 | ]& \ } } � j | t | f i f Vq d S( N( Ro R ( Rp R. Rq ( R1 ( s, /usr/lib64/python2.7/multiprocessing/pool.pys <genexpr>= s N( R= R< R R t listR t divmodR[ RE R Rw t MapResultR; R: R Ru ( R R/ Rm Rn R} t extraRx ( ( R1 s, /usr/lib64/python2.7/multiprocessing/pool.pyRl , s ( c C so t j � } xB | j t k s6 | j rP | j t k rP | j � t j d � q W| j j
d � t d � d S( Ng�������?s worker handler exiting(
RG t current_threadR= R< R; t TERMINATERe t timet sleepR: R R R ( t poolt thread( ( s, /usr/lib64/python2.7/multiprocessing/pool.pyRI A s *
c C s
t j � } x�t | j d � D]q\ } } d } d } zGy� x� t | � D]� \ } } | j rm t d � Pn y | | � WqJ t k
r� }
| d \ } } y | | j | t
|
f � Wq� t k
r� q� XqJ XqJ W| r� t d � | | d � n w PWn} t k
r|}
| r | d n d \ } } | | k rX| | j | d t
|
f � n | r}t d � | | d � q}n XWd d } } } Xq Wt d � y@ t d � | j d � t d � x | D] } | d � q�WWn t
k
r�t d
� n Xt d � d S(
Ni����s' task handler found thread._state != RUNi s doing set_length()i i s task handler got sentinels/ task handler sending sentinel to result handlers( task handler sending sentinel to workerss/ task handler got IOError when sending sentinelss task handler exiting( i i ( RG R� t iterR R Ru R= R R$ t _setR% t KeyErrorR R" ( t taskqueueR R' R� t cacheR� t taskseqt
set_lengthR, R. R2 R- t indt ext p( ( s, /usr/lib64/python2.7/multiprocessing/pool.pyRM N sR
!
c C s t j � } x� y
| � } Wn" t t f k
r@ t d � d SX| j rm | j t k s_ t � t d � Pn | d k r� t d � Pn | \ } } } y | | j | | � Wn t
k
r� n Xd } } } q Wx� | r�| j t k r�y
| � } Wn" t t f k
r t d � d SX| d k r:t d � q� n | \ } } } y | | j | | � Wn t
k
rtn Xd } } } q� Wt | d � r�t d � y5 x. t d � D] } | j
j � s�Pn | � q�WWq�t t f k
r�q�Xn t d t | � | j � d S( Ns. result handler got EOFError/IOError -- exitings, result handler found thread._state=TERMINATEs result handler got sentinels& result handler ignoring extra sentinelR s" ensuring that outqueue is not fulli
s7 result handler exiting: len(cache)=%s, thread._state=%s( RG R� R" R! R R= R� R R R� R� R RZ R t pollR[ ( R' R R� R� R, R- R. t obj( ( s, /usr/lib64/python2.7/multiprocessing/pool.pyRQ � s\
c c sD t | � } x1 t t j | | � � } | s1 d S| | f Vq Wd S( N( R� t tuplet itertoolst islice( R/ t itt sizeRq ( ( s, /usr/lib64/python2.7/multiprocessing/pool.pyRw � s c C s t d � � d S( Ns: pool objects cannot be passed between processes or pickled( RA ( R ( ( s, /usr/lib64/python2.7/multiprocessing/pool.pyt
__reduce__� s c C s5 t d � | j t k r1 t | _ t | j _ n d S( Ns closing pool( R R= R< t CLOSERJ ( R ( ( s, /usr/lib64/python2.7/multiprocessing/pool.pyR � s
c C s- t d � t | _ t | j _ | j � d S( Ns terminating pool( R R� R= RJ RV ( R ( ( s, /usr/lib64/python2.7/multiprocessing/pool.pyt terminate� s
c C sn t d � | j t t f k s% t � | j j � | j j � | j j � x | j D] } | j � qV Wd S( Ns joining pool(
R R= R� R� R RJ R] RP RS RE ( R R� ( ( s, /usr/lib64/python2.7/multiprocessing/pool.pyR] � s
c C sW t d � | j j � x9 | j � rR | j j � rR | j j � t j d � q Wd S( Ns7 removing tasks from inqueue until task handler finishedi ( R t _rlockt acquiret is_aliveR R� Ri R� R� ( R& t task_handlerR� ( ( s, /usr/lib64/python2.7/multiprocessing/pool.pyt _help_stuff_finish� s
c
C s� t d � t | _ t | _ t d � | j | | t | � � | j � sc t | � d k sc t � t | _ | j d � t d � t j
� | k r� | j d � n | r� t | d d � r� t d � x- | D]" } | j
d k r� | j � q� q� Wn t d � t j
� | k r$| j d � n t d � t j
� | k rP| j d � n | r�t | d d � r�t d
� x; | D]0 } | j � rzt d | j � | j � qzqzWn d S( Ns finalizing pools& helping task handler/workers to finishi s joining worker handlerg}Ô%�I�TR� s terminating workerss joining task handlers joining result handlers joining pool workerss cleaning up worker %d( R R� R= R� R[ R� R R R RG R� R] R R\ R� t pid(
t clsR� R& R' R� t worker_handlerR� t result_handlerR� R� ( ( s, /usr/lib64/python2.7/multiprocessing/pool.pyRT � s8
$
N( ( ( ( R R R R R R R_ RF Re R8 Rk R Ry R{ Rj Rl t staticmethodRI RM RQ Rw R� R R� R] R� t classmethodRT ( ( ( s, /usr/lib64/python2.7/multiprocessing/pool.pyR � s0 :
4< R| c B sD e Z d � Z d � Z d � Z d d � Z d d � Z d � Z RS( c C sS t j t j � � | _ t j � | _ | | _ t | _ | | _
| | | j <d S( N( RG t Conditiont Lockt _condt job_countert nextRo R; R% t _readyt _callback( R R� R} ( ( s, /usr/lib64/python2.7/multiprocessing/pool.pyR s c C s | j S( N( R� ( R ( ( s, /usr/lib64/python2.7/multiprocessing/pool.pyt ready& s c C s | j s t � | j S( N( R� R t _success( R ( ( s, /usr/lib64/python2.7/multiprocessing/pool.pyt
successful) s c C sB | j j � z | j s, | j j | � n Wd | j j � Xd S( N( R� R� R� t waitt release( R t timeout( ( s, /usr/lib64/python2.7/multiprocessing/pool.pyR� - s
c C s< | j | � | j s t � n | j r/ | j S| j � d S( N( R� R� R R� t _value( R R� ( ( s, /usr/lib64/python2.7/multiprocessing/pool.pyR 5 s
c C s� | \ | _ | _ | j r7 | j r7 | j | j � n | j j � z t | _ | j j � Wd | j j � X| j | j
=d S( N( R� R� R� R� R� R# R� t notifyR� R; Ro ( R R. R� ( ( s, /usr/lib64/python2.7/multiprocessing/pool.pyR� > s
N( R R R R� R� R R� R R� ( ( ( s, /usr/lib64/python2.7/multiprocessing/pool.pyR| s R� c B s e Z d � Z d � Z RS( c C s t j | | | � t | _ d g | | _ | | _ | d k r` d | _ t | _ | | j =n | | t
| | � | _ d S( Ni ( R| R R# R� R R� t
_chunksizet _number_leftR� Ro t bool( R R� Rn t lengthR} ( ( s, /usr/lib64/python2.7/multiprocessing/pool.pyR R s
c C s | \ } } | r� | | j | | j | d | j +| j d 8_ | j d k r| j rn | j | j � n | j | j =| j j � z t | _ | j j
� Wd | j j � XqnW t | _
| | _ | j | j =| j j � z t | _ | j j
� Wd | j j � Xd S( Ni i ( R� R� R� R� R; Ro R� R� R# R� R� R� R% R� ( R R. t success_resultt successR1 ( ( s, /usr/lib64/python2.7/multiprocessing/pool.pyR� ^ s* "
( R R R R� ( ( ( s, /usr/lib64/python2.7/multiprocessing/pool.pyR� P s Rt c B s> e Z d � Z d � Z d d � Z e Z d � Z d � Z RS( c C sk t j t j � � | _ t j � | _ | | _ t j � | _
d | _ d | _
i | _ | | | j <d S( Ni ( RG R� R� R� R� R� Ro R; t collectionst dequet _itemst _indexR t _lengtht _unsorted( R R� ( ( s, /usr/lib64/python2.7/multiprocessing/pool.pyR s c C s | S( N( ( R ( ( s, /usr/lib64/python2.7/multiprocessing/pool.pyt __iter__� s c C s� | j j � z� y | j j � } Wn� t k
r� | j | j k rN t � n | j j | � y | j j � } Wq� t k
r� | j | j k r� t � n t � q� Xn XWd | j j
� X| \ } } | r� | S| � d S( N( R� R� R� t popleftt
IndexErrorR� R� t
StopIterationR� R R� ( R R� Rs R� R ( ( s, /usr/lib64/python2.7/multiprocessing/pool.pyR� � s&
c C s� | j j � z� | j | k r� | j j | � | j d 7_ xJ | j | j k r� | j j | j � } | j j | � | j d 7_ qA W| j j � n
| | j | <| j | j k r� | j | j
=n Wd | j j � Xd S( Ni ( R� R� R� R� Ra R� t popR� R� R; Ro R� ( R R. R� ( ( s, /usr/lib64/python2.7/multiprocessing/pool.pyR� � s
c C s^ | j j � z< | | _ | j | j k rH | j j � | j | j =n Wd | j j � Xd S( N( R� R� R� R� R� R; Ro R� ( R R� ( ( s, /usr/lib64/python2.7/multiprocessing/pool.pyRv � s
N( R R R R� R R� t __next__R� Rv ( ( ( s, /usr/lib64/python2.7/multiprocessing/pool.pyRt } s
Rz c B s e Z d � Z RS( c C st | j j � zR | j j | � | j d 7_ | j j � | j | j k r^ | j | j =n Wd | j j � Xd S( Ni (
R� R� R� Ra R� R� R� R; Ro R� ( R R. R� ( ( s, /usr/lib64/python2.7/multiprocessing/pool.pyR� � s
( R R R� ( ( ( s, /usr/lib64/python2.7/multiprocessing/pool.pyRz � s t
ThreadPoolc B sB e Z d d l m Z d d d d � Z d � Z e d � � Z RS( i ( R c C s t j | | | | � d S( N( R R ( R RW R( R) ( ( s, /usr/lib64/python2.7/multiprocessing/pool.pyR � s c C s@ t j � | _ t j � | _ | j j | _ | j j | _ d S( N( R9 RU RO R RN R RR ( R ( ( s, /usr/lib64/python2.7/multiprocessing/pool.pyR8 � s c C sW | j j � z5 | j j � | j j d g | � | j j � Wd | j j � Xd S( N( t not_emptyR� t queuet cleart extendR t
notify_allR� ( R& R� R� ( ( s, /usr/lib64/python2.7/multiprocessing/pool.pyR� � s
N( ( R R t dummyR R R R8 R� R� ( ( ( s, /usr/lib64/python2.7/multiprocessing/pool.pyR� � s ( t __all__RG R9 R� R� R� t multiprocessingR R R t multiprocessing.utilR R R< R� R� t countR� R R$ R R R4 t objectR R| t AsyncResultR� Rt Rz R� ( ( ( s, /usr/lib64/python2.7/multiprocessing/pool.pyt <module># s, ,� �.-I