← Back
Editing: pool.cpython-34.pyo
� e f�a � @ s� d 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 Z d d l Z d d l m Z d d l m Z m Z d Z d Z d Z e j � Z d d � Z d d � Z Gd d � d e � Z Gd d � d � Z d d � Z Gd d � d e � Z d f d d d d � Z Gd d � d e � Z Gd d � d e � Z e Z Gd d � d e � Z Gd d � d e � Z Gd d � d e � Z Gd d � d e � Z d S)!�Pool� ThreadPool� N� )�util)� get_context� TimeoutError� c C s t t | � � S)N)�list�map)�args� r �9/opt/alt/python34/lib64/python3.4/multiprocessing/pool.py�mapstar+ s r c C s t t j | d | d � � S)Nr r )r � itertools�starmap)r r r r � starmapstar. s r c @ s( e Z d Z d d � Z d d � Z d S)�RemoteTracebackc C s | | _ d S)N)�tb)�selfr r r r �__init__6 s zRemoteTraceback.__init__c C s | j S)N)r )r r r r �__str__8 s zRemoteTraceback.__str__N)�__name__� __module__� __qualname__r r r r r r r 5 s r c @ s( e Z d Z d d � Z d d � Z d S)�ExceptionWithTracebackc C sD t j t | � | | � } d j | � } | | _ d | | _ d S)N� z """ %s""")� traceback�format_exception�type�join�excr )r r r r r r r < s z ExceptionWithTraceback.__init__c C s t | j | j f f S)N)� rebuild_excr r )r r r r � __reduce__A s z!ExceptionWithTraceback.__reduce__N)r r r r r" r r r r r ; s r c C s t | � | _ | S)N)r � __cause__)r r r r r r! D s r! c s@ e Z d Z d Z � f d d � Z d d � Z d d � Z � S)�MaybeEncodingErrorzVWraps possible unpickleable errors, so they can be safely sent through the socket.c sA t | � | _ t | � | _ t t | � j | j | j � d S)N)�reprr �value�superr$ r )r r r& )� __class__r r r P s zMaybeEncodingError.__init__c C s d | j | j f S)Nz(Error sending result: '%s'. Reason: '%s')r& r )r r r r r U s zMaybeEncodingError.__str__c C s d t | � S)Nz<MaybeEncodingError: %s>)�str)r r r r �__repr__Y s zMaybeEncodingError.__repr__)r r r �__doc__r r r* r r )r( r r$ L s r$ Fc ' C s� | j } | j } t | d � r> | j j � | j j � n | d k rW | | � n d } xx| d k s~ | r�| | k r�y | � } Wn& t t f k r� t j d � PYn X| d k r� t j d � Pn | \ } } } } } y d | | | � f } WnL t k rM} z, | r/t | | j � } n d | f } WYd d } ~ Xn Xy | | | | f � Wnb t k r�} zB t | | d � } t j d | � | | | d | f f � WYd d } ~ Xn X| d 7} q` Wt j d | � d S) N�_writerr z)worker got EOFError or OSError -- exitingz worker got sentinel -- exitingTFr z0Possible encoding error while sending result: %sz worker exiting after %d tasks)�put�get�hasattrr, �close�_reader�EOFError�OSErrorr �debug� Exceptionr � __traceback__r$ )�inqueue�outqueue� initializer�initargsZmaxtasksZwrap_exceptionr- r. Z completed�task�job�i�funcr �kwds�result�e�wrappedr r r �worker] sB ! ,rC c @ s� e Z d Z d Z d Z d d � Z d d f d d d d � Z d d � Z d d � Z d d � Z d d � Z f i d d � Z d d d � Z d d d � Z d d d d d � Z d d d � Z d d d � Z f i d d d d � Z d d d d d � Z d d d d! d" � Z e d# d$ � � Z e d% d& � � Z e d' d( � � Z e d) d* � � Z d+ d, � Z d- d. � Z d/ d0 � Z d1 d2 � Z e d3 d4 � � Z e d5 d6 � � Z d7 d8 � Z d9 d: � Z! d S);r zS Class which supports an async version of applying functions to arguments. Tc O s | j j | | � S)N)�_ctx�Process)r r r? r r r rE � s z Pool.ProcessNc C s, | p t � | _ | j � t j � | _ i | _ t | _ | | _ | | _ | | _ | d k ry t j � ps d } n | d k r� t d � � n | d k r� t | � r� t d � � n | | _ g | _ | j � t j d t j d | f � | _ d | j _ t | j _ | j j � t j d t j d | j | j | j | j | j f � | _ d | j _ t | j _ | j j � t j d t j d | j | j | j f � | _! d | j! _ t | j! _ | j! j � t" j# | | j$ d | j | j% | j | j | j | j | j! | j f d d �| _& d S) Nr z&Number of processes must be at least 1z initializer must be a callable�targetr TZ exitpriority� )'r rD � _setup_queues�queue�Queue� _taskqueue�_cache�RUN�_state�_maxtasksperchild� _initializer� _initargs�os� cpu_count� ValueError�callable� TypeError� _processes�_pool�_repopulate_pool� threadingZThreadr �_handle_workers�_worker_handler�daemon�start� _handle_tasks� _quick_put� _outqueue� _task_handler�_handle_results� _quick_get�_result_handlerr ZFinalize�_terminate_pool�_inqueue� _terminate)r � processesr9 r: Zmaxtasksperchild�contextr r r r � sT z Pool.__init__c C sz d } xm t t t | j � � � D]P } | j | } | j d k r" t j d | � | j � d } | j | =q" q" W| S)z�Cleanup after any worker processes which have exited due to reaching their specified lifetime. Returns True if any workers were cleaned up. FNzcleaning up worker %dT)�reversed�range�lenrX �exitcoder r4 r )r Zcleanedr= rC r r r �_join_exited_workers� s " zPool._join_exited_workersc C s� x� t | j t | j � � D]� } | j d t d | j | j | j | j | j | j f � } | j j | � | j j d d � | _ d | _ | j � t j d � q Wd S)z�Bring the number of pool processes up to the specified number, for use after reaping workers which have exited. rF r rE Z PoolWorkerTz added workerN)rl rW rm rX rE rC rg ra rP rQ rO �_wrap_exception�append�name�replacer] r^ r r4 )r r= �wr r r rY � s # zPool._repopulate_poolc C s | j � r | j � n d S)zEClean up any exited workers and start replacements for them. N)ro rY )r r r r �_maintain_pool� s zPool._maintain_poolc C sL | j j � | _ | j j � | _ | j j j | _ | j j j | _ d S)N) rD Z SimpleQueuerg ra r, �sendr` r1 �recvrd )r r r r rH � s zPool._setup_queuesc C s | j | | | � j � S)z6 Equivalent of `func(*args, **kwds)`. )� apply_asyncr. )r r> r r? r r r �apply� s z Pool.applyc C s | j | | t | � j � S)zx Apply `func` to each element in `iterable`, collecting the results in a list that is returned. )� _map_asyncr r. )r r> �iterable� chunksizer r r r � s zPool.mapc C s | j | | t | � j � S)z� Like `map()` method but the elements of the `iterable` are expected to be iterables as well and will be unpacked as arguments. Hence `func` and (a, b) becomes func(a, b). )rz r r. )r r> r{ r| r r r r s z Pool.starmapc C s | j | | t | | | � S)z= Asynchronous version of `starmap()` method. )rz r )r r> r{ r| �callback�error_callbackr r r � starmap_async s zPool.starmap_asyncr c s� | j t k r t d � � n | d k rr t | j � � | j j � � f d d � t | � D� � j f � � St j � | | � } t | j � � | j j � f d d � t | � D� � j f � d d � � D� Sd S)zP Equivalent of `map()` -- can be MUCH slower than `Pool.map()`. zPool not runningr c 3 s0 | ]& \ } } � j | � | f i f Vq d S)N)�_job)�.0r= �x)r> r@ r r � <genexpr> s z Pool.imap.<locals>.<genexpr>c 3 s0 | ]& \ } } � j | t | f i f Vq d S)N)r� r )r� r= r� )r@ r r r� % s c s s" | ] } | D] } | Vq q d S)Nr )r� �chunk�itemr r r r� ' s N) rN rM rT � IMapIteratorrL rK r- � enumerate� _set_lengthr � _get_tasks)r r> r{ r| � task_batchesr )r> r@ r �imap s z Pool.imapc s� | j t k r t d � � n | d k rr t | j � � | j j � � f d d � t | � D� � j f � � St j � | | � } t | j � � | j j � f d d � t | � D� � j f � d d � � D� Sd S)zL Like `imap()` method but ordering of results is arbitrary. zPool not runningr c 3 s0 | ]& \ } } � j | � | f i f Vq d S)N)r� )r� r= r� )r> r@ r r r� 1 s z&Pool.imap_unordered.<locals>.<genexpr>c 3 s0 | ]& \ } } � j | t | f i f Vq d S)N)r� r )r� r= r� )r@ r r r� 8 s c s s" | ] } | D] } | Vq q d S)Nr )r� r� r� r r r r� : s N) rN rM rT �IMapUnorderedIteratorrL rK r- r� r� r r� )r r> r{ r| r� r )r> r@ r �imap_unordered) s zPool.imap_unorderedc C sb | j t k r t d � � n t | j | | � } | j j | j d | | | f g d f � | S)z; Asynchronous version of `apply()` method. zPool not runningN)rN rM rT � ApplyResultrL rK r- r� )r r> r r? r} r~ r@ r r r rx < s +zPool.apply_asyncc C s | j | | t | | | � S)z9 Asynchronous version of `map()` method. )rz r )r r> r{ r| r} r~ r r r � map_asyncG s zPool.map_asyncc s | j t k r t d � � n t | d � s< 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 | � | d | �� | j j � � f d d � t | � D� d f � � S) zY Helper function to implement map, starmap and their async counterparts. zPool not running�__len__N� r r r~ c 3 s0 | ]& \ } } � j | � | f i f Vq d S)N)r� )r� r= r� )�mapperr@ r r r� c s z"Pool._map_async.<locals>.<genexpr>)rN rM rT r/ r �divmodrm rX r r� � MapResultrL rK r- r� ) r r> r{ r� r| r} r~ Zextrar� r )r� r@ r rz O s ( zPool._map_asyncc C sr t j � } xB | j t k s6 | j rP | j t k rP | j � t j d � q W| j j d � t j d � d S)Ng�������?zworker handler exiting) rZ �current_threadrN rM rL � TERMINATEru �time�sleeprK r- r r4 )�pool�threadr r r r[ g s * zPool._handle_workersc C s> t j � } x�t | j d � D]�\ } } d } d } y� x� t | � D]� \ } } | j rm t j d � Pn y | | � WqG t k r� } zN | d d � \ } } y | | j | d | f � Wn t k r� Yn XWYd d } ~ XqG XqG W| rt j d � | | d � n w PWq t k r�} zx | rD| d d � n d \ } } | | k r}| | j | d d | f � n | r�t j d � | | d � n WYd d } ~ Xq Xq Wt j d � yF t j d � | j d � t j d � x | D] } | d � q�WWn t k r,t j d � Yn Xt j d � d S)Nr z'task handler found thread._state != RUNr Fzdoing set_length()r ztask handler got sentinelz/task handler sending sentinel to result handlerz(task handler sending sentinel to workersz/task handler got OSError when sending sentinelsztask handler exiting���)r r ) rZ r� �iterr. r� rN r r4 r5 �_set�KeyErrorr- r3 )� taskqueuer- r8 r� �cacher� ZtaskseqZ set_lengthr; r= rA r< ZindZex�pr r r r_ t sN " ! ( zPool._handle_tasksc C s� t j � } x� y | � } Wn) t t f k rG t j d � d SYn X| j rb t j d � Pn | d k r t j d � Pn | \ } } } y | | j | | � Wq t k r� Yq Xq Wx� | rn| j t k rny | � } Wn) t t f k rt j d � d SYn X| d k r/t j d � q� n | \ } } } y | | j | | � Wq� t k rjYq� Xq� Wt | d � r�t j d � y5 x. t d � D] } | j j � s�Pn | � q�WWq�t t f k r�Yq�Xn t j d t | � | j � d S) Nz.result handler got EOFError/OSError -- exitingz,result handler found thread._state=TERMINATEzresult handler got sentinelz&result handler ignoring extra sentinelr1 z"ensuring that outqueue is not full� z7result handler exiting: len(cache)=%s, thread._state=%s)rZ r� r3 r2 r r4 rN r� r� r� r/ rl r1 �pollrm )r8 r. r� r� r; r< r= �objr r r rc � sV zPool._handle_resultsc c sD t | � } x1 t t j | | � � } | s1 d S| | f Vq Wd S)N)r� �tupler �islice)r> �it�sizer� r r r r� � s zPool._get_tasksc C s t d � � d S)Nz:pool objects cannot be passed between processes or pickled)�NotImplementedError)r r r r r" � s zPool.__reduce__c C s8 t j d � | j t k r4 t | _ t | j _ n d S)Nz closing pool)r r4 rN rM �CLOSEr\ )r r r r r0 � s z Pool.closec C s0 t j d � t | _ t | j _ | j � d S)Nzterminating pool)r r4 r� rN r\ rh )r r r r � terminate� s zPool.terminatec C sV t j d � | j j � | j j � | j j � x | j D] } | j � q> Wd S)Nz joining pool)r r4 r\ r rb re rX )r r� r r r r � s z Pool.joinc C sZ t j d � | j j � x9 | j � rU | j j � rU | j j � t j d � q Wd S)Nz7removing tasks from inqueue until task handler finishedr ) r r4 Z_rlock�acquire�is_aliver1 r� rw r� r� )r7 � task_handlerr� r r r �_help_stuff_finish s zPool._help_stuff_finishc C s� t j d � t | _ t | _ t j d � | j | | t | � � t | _ | j d � t j d � t j � | k r� | j � n | r� t | d d � r� t j d � x- | D]" } | j d k r� | j � q� q� Wn t j d � t j � | k r | j � n t j d � t j � | k r5| j � n | r�t | d d � r�t j d � x> | D]3 } | j � rbt j d | j � | j � qbqbWn d S) Nzfinalizing poolz&helping task handler/workers to finishzjoining worker handlerr r� zterminating workerszjoining task handlerzjoining result handlerzjoining pool workerszcleaning up worker %d)r r4 r� rN r� rm r- rZ r� r r/ rn r� r� �pid) �clsr� r7 r8 r� Zworker_handlerr� Zresult_handlerr� r� r r r rf s6 zPool._terminate_poolc C s | S)Nr )r r r r � __enter__; s zPool.__enter__c C s | j � d S)N)r� )r �exc_typeZexc_valZexc_tbr r r �__exit__> s z Pool.__exit__)"r r r r+ rp rE r ro rY ru rH ry r r r r� r� rx r� rz � staticmethodr[ r_ rc r� r" r0 r� r r� � classmethodrf r� r� r r r r r � sD 8 3: . c @ s^ e Z d Z d d � Z d d � Z d d � Z d d d � Z d d d � Z d d � Z d S)r� c C sJ t j � | _ t t � | _ | | _ | | _ | | _ | | | j <d S)N) rZ ZEvent�_event�next� job_counterr� rL � _callback�_error_callback)r r� r} r~ r r r r G s zApplyResult.__init__c C s | j j � S)N)r� Zis_set)r r r r �readyO s zApplyResult.readyc C s | j S)N)�_success)r r r r � successfulR s zApplyResult.successfulNc C s | j j | � d S)N)r� �wait)r �timeoutr r r r� V s zApplyResult.waitc C s? | j | � | j � s" t � n | j r2 | j S| j � d S)N)r� r� r r� �_value)r r� r r r r. Y s zApplyResult.getc C s{ | \ | _ | _ | j r7 | j r7 | j | j � n | j r] | j r] | j | j � n | j j � | j | j =d S)N)r� r� r� r� r� �setrL r� )r r= r� r r r r� b s zApplyResult._set) r r r r r� r� r� r. r� r r r r r� E s r� c @ s( e Z d Z d d � Z d d � Z d S)r� c C s� t j | | | d | �d | _ d g | | _ | | _ | d k rj d | _ | j j � | | j =n | | t | | � | _ d S)Nr~ Tr ) r� r r� r� � _chunksize� _number_leftr� r� r� �bool)r r� r| �lengthr} r~ r r r r s s zMapResult.__init__c C s� | \ } } | r� | | j | | j | d | j � <| j d 8_ | j d k r� | j rq | j | j � n | j | j =| j j � q� nH d | _ | | _ | j r� | j | j � n | j | j =| j j � d S)Nr r F) r� r� r� r� rL r� r� r� r� r� )r r= Zsuccess_result�successr@ r r r r� � s % zMapResult._setN)r r r r r� r r r r r� q s r� c @ sU e Z d Z d d � Z d d � Z d d d � Z e Z d d � Z d d � Z d S) r� c C sk t j t j � � | _ t t � | _ | | _ t j � | _ d | _ d | _ i | _ | | | j <d S)Nr )rZ Z ConditionZLock�_condr� r� r� rL � collections�deque�_items�_index�_length� _unsorted)r r� r r r r � s zIMapIterator.__init__c C s | S)Nr )r r r r �__iter__� s zIMapIterator.__iter__Nc C s� | j j � z� y | j j � } Wn� t k r� | j | j k rN t � n | j j | � y | j j � } Wn3 t k r� | j | j k r� t � n t � Yn XYn XWd | j j � X| \ } } | r� | S| � d S)N) r� r� r� �popleft� IndexErrorr� r� � StopIterationr� r �release)r r� r� r� r& r r r r� � s& zIMapIterator.nextc 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)Nr ) r� r� r� r� rq r� �pop�notifyr� rL r� r� )r r= r� r r r r� � s zIMapIterator._setc 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� rL r� r� )r r� r r r r� � s zIMapIterator._set_length) r r r r r� r� �__next__r� r� r r r r r� � s r� c @ s e Z d Z d d � Z d S)r� 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)Nr ) r� r� r� rq r� r� r� rL r� r� )r r= r� r r r r� � s zIMapUnorderedIterator._setN)r r r r� r r r r r� � s r� c @ s[ e Z d Z d Z e d d � � Z d d f d d � Z d d � Z e d d � � Z d S) r Fc O s d d l m } | | | � S)Nr )rE )ZdummyrE )r r? rE r r r rE � s zThreadPool.ProcessNc C s t j | | | | � d S)N)r r )r ri r9 r: r r r r � s zThreadPool.__init__c C s@ t j � | _ t j � | _ | j j | _ | j j | _ d S)N)rI rJ rg ra r- r` r. rd )r r r r rH � s zThreadPool._setup_queuesc C sW | j j � z5 | j j � | j j d g | � | j j � Wd | j j � Xd S)N)Z not_emptyr� rI �clear�extendZ notify_allr� )r7 r� r� r r r r� s z ThreadPool._help_stuff_finish) r r r rp r� rE r rH r� r r r r r � s )!�__all__rZ rI r r� rR r� r r r r r rM r� r� �countr� r r r5 r r r! r$ rC �objectr r� Z AsyncResultr� r� r� r r r r r �<module> s: ,� �&%I
Save File
Cancel