U
e5dY
@ sp d Z ddlZddlZddlZddlZdZdZdZdZdZ dZ
d Z d
Z d Z
e e
e e e
gZe d e
d
e de de
diZedZG dd deZG dd deZG dd deZG dd deZG dd deZG dd d eZG d d d eZG d d d eZG d!d" d"eZ d#d$ Z d%d& Z d3d'd(Z e d)d*Z!defd+d,Z"G d-d. d.eZ#G d/d0 d0eZ$G d1d2 d2e%Z&dS )4z"Brian Quinlan (brian@sweetapp.com) NFIRST_COMPLETEDFIRST_EXCEPTION
ALL_COMPLETED
_AS_COMPLETEDPENDINGRUNNING CANCELLEDCANCELLED_AND_NOTIFIEDFINISHEDpendingrunning cancelledfinishedzconcurrent.futuresc @ s e Zd ZdS )ErrorN__name__
__module__ __qualname__ r r 0/usr/lib64/python3.8/concurrent/futures/_base.pyr , s r c @ s e Zd ZdS )CancelledErrorNr r r r r r 0 s r c @ s e Zd ZdS ) TimeoutErrorNr r r r r r 4 s r c @ s e Zd ZdS )InvalidStateErrorNr r r r r r 8 s r c @ s, e Zd Zdd Zdd Zdd Zdd Zd S )
_Waiterc C s t | _g | _d S N) threadingZEventeventfinished_futuresselfr r r __init__> s
z_Waiter.__init__c C s | j | d S r r appendr futurer r r
add_resultB s z_Waiter.add_resultc C s | j | d S r r! r# r r r
add_exceptionE s z_Waiter.add_exceptionc C s | j | d S r r! r# r r r
add_cancelledH s z_Waiter.add_cancelledN)r r r r r% r& r' r r r r r < s r c s@ e Zd Z fddZ fddZ fddZ fddZ ZS ) _AsCompletedWaiterc s t t| t | _d S r )superr( r r Locklockr __class__r r r N s z_AsCompletedWaiter.__init__c s0 | j tt| | | j W 5 Q R X d S r )r+ r) r( r% r setr# r, r r r% R s z _AsCompletedWaiter.add_resultc s0 | j tt| | | j W 5 Q R X d S r )r+ r) r( r& r r. r# r, r r r& W s z _AsCompletedWaiter.add_exceptionc s0 | j tt| | | j W 5 Q R X d S r )r+ r) r( r' r r. r# r, r r r' \ s z _AsCompletedWaiter.add_cancelled)r r r r r% r& r'
__classcell__r r r, r r( K s r( c s4 e Zd Z fddZ fddZ fddZ ZS )_FirstCompletedWaiterc s t | | j d S r )r) r% r r. r# r, r r r% d s z _FirstCompletedWaiter.add_resultc s t | | j d S r )r) r& r r. r# r, r r r& h s z#_FirstCompletedWaiter.add_exceptionc s t | | j d S r )r) r' r r. r# r, r r r' l s z#_FirstCompletedWaiter.add_cancelled)r r r r% r& r' r/ r r r, r r0 a s r0 c sH e Zd Z fddZdd Z fddZ fddZ fd d
Z ZS ) _AllCompletedWaiterc s$ || _ || _t | _t d S r )num_pending_callsstop_on_exceptionr r* r+ r) r )r r2 r3 r, r r r s s
z _AllCompletedWaiter.__init__c C s4 | j $ | jd8 _| js&| j W 5 Q R X d S )N )r+ r2 r r. r r r r _decrement_pending_callsy s z,_AllCompletedWaiter._decrement_pending_callsc s t | | d S r )r) r% r5 r# r, r r r% s z _AllCompletedWaiter.add_resultc s* t | | jr | j n| d S r )r) r& r3 r r. r5 r# r, r r r& s z!_AllCompletedWaiter.add_exceptionc s t | | d S r )r) r' r5 r# r, r r r' s z!_AllCompletedWaiter.add_cancelled) r r r r r5 r% r& r' r/ r r r, r r1 p s
r1 c @ s$ e Zd Zdd Zdd Zdd ZdS )_AcquireFuturesc C s t |td| _d S )N)key)sortedidfutures)r r: r r r r s z_AcquireFutures.__init__c C s | j D ]}|j qd S r )r:
_conditionacquirer# r r r __enter__ s
z_AcquireFutures.__enter__c G s | j D ]}|j qd S r )r: r; release)r argsr$ r r r __exit__ s
z_AcquireFutures.__exit__N)r r r r r= r@ r r r r r6 s r6 c C s |t krt }nZ|tkr t }nJtdd | D }|tkrHt|dd}n"|tkr^t|dd}n td| | D ]}|j
| qn|S )Nc s s | ]}|j ttfkV qd S r _stater r
.0fr r r s z._create_and_install_waiters..T)r3 Fz Invalid return condition: %r) r r( r r0 sumr r1 r
ValueError_waitersr" )fs return_whenwaiterZ
pending_countrE r r r _create_and_install_waiters s rM c c sP | rL| d }|D ]}| | q|j |j | W 5 Q R X ~| V q d S )N)remover; rI pop)rJ rL ref_collectrE Z futures_setr r r _yield_finished_futures s rR c c sB |d k r|t }t| } t| }t| * tdd | D }| | }t| t}W 5 Q R X t|}zt ||| fdE d H |r|d krd }n(|t }|dk rt dt||f |j
| |j |j}g |_|j
W 5 Q R X | t ||| |fdE d H q|W 5 | D ]$}|j |j
| W 5 Q R X qX d S )Nc s s | ]}|j ttfkr|V qd S r rA rC r r r rF s z as_completed..)rQ r z %d (of %d) futures unfinished)time monotonicr. lenr6 rM r listr; rI rO rR r r waitr+ r clearreverse) rJ timeoutend_timeZ
total_futuresr r rL rE Z wait_timeoutr r r as_completed sL
r\ DoneAndNotDoneFuturesz
done not_donec
C s
t | tdd | D }t| | }|tkrJ|rJt||W 5 Q R S |tkr~|r~tdd |D r~t||W 5 Q R S t|t| krt||W 5 Q R S t| |}W 5 Q R X |j | | D ]"}|j
|j | W 5 Q R X q|
|j t|t| | S )Nc s s | ]}|j ttfkr|V qd S r rA rC r r r rF ! s zwait..c s s& | ] }| s| d k r|V qd S r )r
exceptionrC r r r rF ( s )r6 r. r r] r anyrU rM r rW r; rI rO updater )rJ rZ rK doneZnot_donerL rE r r r rW s"
rW c @ s e Zd Zdd Zdd Zdd Zdd Zd d
Zd d Zd
d Z dd Z
dd Z d ddZ d ddZ
dd Zdd Zd d ZdS ) Futurec C s, t | _t| _d | _d | _g | _g | _d S r ) r Z Conditionr; r rB _result
_exceptionrI _done_callbacksr r r r r < s
zFuture.__init__c C s> | j D ]2}z || W q tk
r6 td| Y qX qd S Nz!exception calling callback for %r)re ExceptionLOGGERr^ )r callbackr r r _invoke_callbacksE s
zFuture._invoke_callbacksc
C s | j | jtkrx| jrHd| jjt| t| j | jjjf W 5 Q R S d| jjt| t| j | jjjf W 5 Q R S d| jjt| t| j f W 5 Q R S Q R X d S )Nz <%s at %#x state=%s raised %s>z <%s at %#x state=%s returned %s>z<%s at %#x state=%s>) r; rB r
rd r- r r9 _STATE_TO_DESCRIPTION_MAPrc r r r r __repr__L s(
zFuture.__repr__c C sf | j N | jttfkr$W 5 Q R dS | jttfkr@W 5 Q R dS t| _| j W 5 Q R X | dS )NFT)r; rB r r
r r
notify_allrj r r r r cancel` s z
Future.cancelc
C s, | j | jttfkW 5 Q R S Q R X d S r )r; rB r r r r r r r
s s zFuture.cancelledc
C s( | j | jtkW 5 Q R S Q R X d S r )r; rB r r r r r r x s zFuture.runningc
C s. | j | jtttfkW 5 Q R S Q R X d S r )r; rB r r r
r r r r ra } s z Future.donec C s$ | j rz
| j W 5 d } X n| jS d S r )rd rc r r r r Z __get_result s
zFuture.__get_resultc C sn | j 0 | jtttfkr2| j| W 5 Q R d S W 5 Q R X z || W n tk
rh t d| Y nX d S rf )
r; rB r r r
re r" rg rh r^ )r fnr r r add_done_callback s zFuture.add_done_callbackNc
C s z| j | jttfkr t n"| jtkrB| W 5 Q R W ZS | j | | jttfkrdt n(| jtkr| W 5 Q R W S t W 5 Q R X W 5 d } X d S r ) r; rB r r r r
_Future__get_resultrW r r rZ r r r result s
z
Future.resultc
C s | j | | jttfkr t n | jtkr:| jW 5 Q R S | j | | jttfkr\t n"| jtkrx| jW 5 Q R S t W 5 Q R X d S r ) r; rB r r r r
rd rW r rr r r r r^ s
zFuture.exceptionc C s | j t | jtkrdd l}|jdtdd ntdt | d t d S ) N z:descriptor 'submit' of 'Executor' object needs an argumentro r z.Passing 'fn' as keyword argument is deprecated)
stacklevelz6submit expected at least 1 positional argument, got %dr4 )rU TypeErrorwarningswarnDeprecationWarningNotImplementedError)r? kwargsr~ r r r submit. s
zExecutor.submitz ($self, fn, /, *args, **kwargs)Nr4 )rZ chunksizec sB d k rt fddt| D fdd}| S )Nc s g | ]}j f| qS r )r )rD r? )ro r r r
` s z Executor.map..c 3 s\ zB r@d kr& V q
t V q
W 5 D ] } | qHX d S r )rn rY rP rs rS rT )r$ )r[ rJ rZ r r result_iteratord s z%Executor.map..result_iterator)rS rT zip)r ro rZ r iterablesr r )r[ ro rJ r rZ r mapG s
z Executor.mapTc C s d S r r )r rW r r r shutdowns s zExecutor.shutdownc C s | S r r r r r r r= s zExecutor.__enter__c C s | j dd dS )NT)rW F)r )r exc_typeZexc_valZexc_tbr r r r@ s zExecutor.__exit__)T) r r r r __text_signature__r r r= r@ r r r r rz + s ,
rz c @ s e Zd ZdS )BrokenExecutorNr r r r r r s r )N)'
__author__ collectionsZloggingr rS r r r r r r r r r
Z_FUTURE_STATESrk Z getLoggerrh rg r r r r objectr r( r0 r1 r6 rM rR r\
namedtupler] rW rb rz rt r r r r r sh
> 1 s]