3
\ > � @ s d dddddgZ ddlZddlZddlZddlZdd lmZ dd
lmZ dd lm Z ej
Z
ej Z ej Z ej
Z
ejZejZejZejd ZG d d
� d
�ZG dd� d�ZeZdd� Zdd� Zdd� Zdd� Zdd�dd�Zy ddlZW n e k
r� Y n X ej ZZ dS )�CancelledError� TimeoutError�InvalidStateError�Future� wrap_future�isfuture� N� )� base_futures)�compat)�eventsc @ s0 e Zd ZdZdd� Zdd� Zd d
� Zd d � Zd
S )�_TracebackLogger�loop�source_traceback�exc�tbc C s |j | _|j| _|| _d | _d S )N)�_loopr
�_source_tracebackr r r )�self�futurer � r �'/usr/lib64/python3.6/asyncio/futures.py�__init__R s z_TracebackLogger.__init__c C s, | j }|d k r(d | _ tj|j||j�| _d S )N)r � traceback�format_exception� __class__�
__traceback__r )r r r r r �activateX s
z_TracebackLogger.activatec C s d | _ d | _d S )N)r r )r r r r �clear_ s z_TracebackLogger.clearc C sb | j r^d}| jr:djtj| j��}|d7 }|d|j� 7 }|dj| j �j� 7 }| jjd|i� d S )Nz*Future/Task exception was never retrieved
� z0Future/Task created at (most recent call last):
z%s
�message)r r �joinr � format_list�rstripr
�call_exception_handler)r �msg�srcr r r �__del__c s z_TracebackLogger.__del__N)r
r r r )�__name__�
__module__� __qualname__� __slots__r r r r& r r r r r s
2r c @ s� e Zd ZeZdZdZdZdZdZ dZ
dd�dd�Z e j
Zdd� ZejrNdd � Zd
d � Zd d
� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zd d � Z d d � Z ej r�e Z dS ) r NF)r
c C s@ |d krt j� | _n|| _g | _| jj� r� )r r' r �
_repr_info)r r r r �__repr__� s zFuture.__repr__c C sD | j s
d S | j}d| jj || d�}| jr4| j|d< | jj|� d S )Nz %s exception was never retrieved)r � exceptionr r )�_log_traceback�
_exceptionr r' r r r# )r r �contextr r r r& � s
zFuture.__del__c C s& d| _ | jtkrdS t| _| j� dS )NFT)r4 �_state�_PENDING�
_CANCELLED�_schedule_callbacks)r r r r �cancel� s
z
Future.cancelc C sD | j d d � }|sd S g | j d d �< x|D ]}| jj|| � q*W d S )N)r, r � call_soon)r Z callbacks�callbackr r r r: � s
zFuture._schedule_callbacksc C s
| j tkS )N)r7 r9 )r r r r � cancelled� s zFuture.cancelledc C s
| j tkS )N)r7 r8 )r r r r �done� s z Future.donec C s< | j tkrt�| j tkr td��d| _| jd k r6| j�| jS )NzResult is not ready.F)r7 r9 r � _FINISHEDr r4 r5 �_result)r r r r �result� s
z
Future.resultc C s, | j tkrt�| j tkr td��d| _| jS )NzException is not set.F)r7 r9 r r@ r r4 r5 )r r r r r3 � s
zFuture.exceptionc C s* | j tkr| jj|| � n | jj|� d S )N)r7 r8 r r< r, �append)r �fnr r r �add_done_callback s
zFuture.add_done_callbackc s< � fdd�| j D �}t| j �t|� }|r8|| j d d �< |S )Nc s g | ]}|� kr|�qS r r )�.0�f)rD r r �
s z/Future.remove_done_callback..)r, �len)r rD Zfiltered_callbacksZ
removed_countr )rD r �remove_done_callback s
zFuture.remove_done_callbackc C s4 | j tkr tdj| j | ���|| _t| _ | j� d S )Nz{}: {!r})r7 r8 r �formatrA r@ r: )r rB r r r �
set_result s
zFuture.set_resultc C s� | j tkr tdj| j | ���t|t�r,|� }t|�tkr@td��|| _t | _ | j
� t j rbd| _
n t| |�| _| jj| jj� d S )Nz{}: {!r}zPStopIteration interacts badly with generators and cannot be raised into a FutureT)r7 r8 r rK �
isinstance�type�
StopIteration� TypeErrorr5 r@ r: r
�PY34r4 r Z
_tb_loggerr r< r )r r3 r r r �
set_exception, s
zFuture.set_exceptionc c s | j � sd| _| V | j� S )NT)r? �_asyncio_future_blockingrB )r r r r �__iter__D s zFuture.__iter__) r' r( r) r8 r7 rA r5 r r rS r4 r r Z_future_repr_infor1 r2 r
rQ r& r; r: r> r? rB r3 rE rJ rL rR rT ZPY35� __await__r r r r r n s2
c C s | j � r d S | j|� d S )N)r> rL )ZfutrB r r r � _set_result_unless_cancelledS s rV c C sN |j � r| j� | j� s d S |j� }|d k r8| j|� n|j� }| j|� d S )N)r> r; Z set_running_or_notify_cancelr3 rR rB rL )�
concurrent�sourcer3 rB r r r � _set_concurrent_future_stateZ s rY c C sP |j � r d S | j � r |j� n.| j� }|d k r:|j|� n| j� }|j|� d S )N)r> r; r3 rR rB rL )rX �destr3 rB r r r �_copy_future_statei s
r[ c s� t �� r"t�tjj� r"td��t �� rDt�tjj� rDtd��t ��rR�jnd �t ��rd�jnd �dd� � ���fdd�}� ���fdd�}�j|� �j|� d S ) Nz(A future is required for source argumentz-A future is required for destination argumentc S s" t | �rt|| � n
t| |� d S )N)r r[ rY )r �otherr r r �
_set_state� s z!_chain_future.._set_statec s2 | j � r.�d ks�� kr"�j� n �j�j� d S )N)r> r; �call_soon_threadsafe)� destination)� dest_looprX � source_loopr r �_call_check_cancel� s
z)_chain_future.._call_check_cancelc sJ �j � r �d k r �j� r d S �d ks,��kr8� �| � n�j� �| � d S )N)r> Z is_closedr^ )rX )r] r` r_ ra r r �_call_set_state� s z&_chain_future.._call_set_state)r rM rW Zfuturesr rP r rE )rX r_ rb rc r )r] r` r_ rX ra r �
_chain_future} s
rd )r
c C s2 t | �r | S |d kr tj� }|j� }t| |� |S )N)r r r+ Z
create_futurerd )r r
Z
new_futurer r r r � s
) �__all__Zconcurrent.futuresrW Zloggingr. r r r r
r r r r r r8 r9 r@ �DEBUGZ STACK_DEBUGr r Z _PyFuturerV rY r[ rd r Z_asyncio� ImportErrorZ_CFuturer r r r � s<
P c*