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
diZe�d�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd � d e�ZG d d � d e�ZG d d � d e�ZG d!d"� d"e�Z d#d$� Z d%d&� Z d3d'd(�Z e� d)d*�Z!defd+d,�Z"G d-d.� d.e�Z#G d/d0� d0e�Z$G d1d2� d2e%�Z&dS )4z"Brian Quinlan ([email protected])� N�FIRST_COMPLETED�FIRST_EXCEPTION�
ALL_COMPLETED�
_AS_COMPLETED�PENDING�RUNNING� CANCELLED�CANCELLED_AND_NOTIFIED�FINISHED�pending�running� cancelled�finishedzconcurrent.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)� threadingZEvent�event�finished_futures��selfr r r �__init__> s
z_Waiter.__init__c C s | j �|� d S r �r �append�r �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� fdd�Z� fdd�Z� fdd�Z� fdd�Z� ZS ) �_AsCompletedWaiterc s t t| ��� t�� | _d S r )�superr( r r �Lock�lockr �� __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� fdd�Z� fdd�Z� fdd�Z� 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� fdd�Zdd� Z� fdd�Z� fdd�Z� fd d
�Z� ZS ) �_AllCompletedWaiterc s$ || _ || _t�� | _t� �� d S r )�num_pending_calls�stop_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)�sorted�id�futures)r r: r r r r � s z_AcquireFutures.__init__c C s | j D ]}|j�� qd S r )r: �
_condition�acquirer# 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
��.0�fr 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_when�waiterZ
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|�}z�t ||| fd�E d H |�r|d kr�d }n(|t �� }|dk r�t dt|�|f ��|j
�|� |j� |j}g |_|j
�� W 5 Q R X |�� t ||| |fd�E 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 �clear�reverse) rJ �timeout�end_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| �kr�t||�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 dd�Z d dd�Z
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 t�d| � Y qX qd S �Nz!exception calling callback for %r)re � Exception�LOGGERr^ )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 � TypeError�warnings�warn�DeprecationWarning�NotImplementedError)r? �kwargsr~ r r r �submit. s
�
�zExecutor.submitz ($self, fn, /, *args, **kwargs)Nr4 )rZ � chunksizec sB �d k r�t �� � ��fdd�t|� 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]