U
e5db3 � @ s� d Z ddlZddlZddlZddlZddlmZ ddlmZ ddlm Z ddlm
Z
ej Z ej Z ej
Z
ejZejd ZG dd � d �ZeZd
d � Zd d
� Zdd� Zdd� Zdd� Zdd� Zdd�dd�Zz ddlZW n ek
r� Y n X ej ZZ dS ))�Future� wrap_future�isfuture� N� )� base_futures)�events)�
exceptions)�format_helpersc @ s� e Zd ZeZdZdZdZdZdZ dZ
dd�dd�Z e j
Zdd� Zdd � Zed
d � �Zejd d � �Zd
d� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd�d d �Zd d � Z d d!� Z d"d#� Z d$d%� Z e Z dS )&r NF��loopc C s@ |d krt �� | _n|| _g | _| j�� r� )�format� __class__�__name__�join�
_repr_info�r r r r �__repr__V s
�zFuture.__repr__c C sF | j s
d S | j}| jj� d�|| d�}| jr6| j|d< | j�|� d S )Nz exception was never retrieved)�message� exception�futureZsource_traceback)�_Future__log_traceback�
_exceptionr r r r
Zcall_exception_handler)r �exc�contextr r r �__del__Z s �
zFuture.__del__c C s | j S �N)r# r r r r �_log_tracebackj s zFuture._log_tracebackc C s t |�rtd��d| _d S )Nz'_log_traceback can only be set to FalseF)�bool�
ValueErrorr# )r �valr r r r) n s c C s | j }|d krtd��|S )Nz!Future object is not initialized.)r
� RuntimeErrorr r r r �get_loopt s zFuture.get_loopc C s& d| _ | jtkrdS t| _| �� dS )NFT)r# �_state�_PENDING�
_CANCELLED�_Future__schedule_callbacksr r r r �cancel{ s
z
Future.cancelc C sH | j d d � }|sd S g | j d d �< |D ]\}}| jj|| |d� q(d S �N�r& )r r
� call_soon)r Z callbacks�callback�ctxr r r Z__schedule_callbacks� s zFuture.__schedule_callbacksc C s
| j tkS r( )r/ r1 r r r r � cancelled� s zFuture.cancelledc C s
| j tkS r( )r/ r0 r r r r �done� s z Future.donec C s@ | j tkrtj�| j tkr$t�d��d| _| jd k r:| j�| jS )NzResult is not ready.F) r/ r1 r �CancelledError� _FINISHED�InvalidStateErrorr# r$ �_resultr r r r �result� s
z
Future.resultc C s0 | j tkrtj�| j tkr$t�d��d| _| jS )NzException is not set.F)r/ r1 r r; r< r= r# r$ r r r r r! � s
zFuture.exceptionr5 c C sB | j tkr | jj|| |d� n |d kr.t�� }| j�||f� d S r4 )r/ r0 r
r6 � contextvarsZ copy_contextr �append)r �fnr& r 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||f�qS r r )�.0�fr8 �rB r r �
� s �z/Future.remove_done_callback..)r �len)r rB Zfiltered_callbacksZ
removed_countr rF r �remove_done_callback� s
�zFuture.remove_done_callbackc C s8 | j tkr t�| j � d| ����|| _t| _ | �� d S )N�: )r/ r0 r r= r> r< r2 )r r? r r r �
set_result� s
zFuture.set_resultc C sb | j tkr t�| j � d| ����t|t�r0|� }t|�tkrDtd��|| _t | _ | �
� d| _ d S )NrJ zPStopIteration interacts badly with generators and cannot be raised into a FutureT) r/ r0 r r= �
isinstance�type�
StopIteration� TypeErrorr$ r< r2 r# )r r! r r r �
set_exception� s
zFuture.set_exceptionc c s, | � � sd| _| V | � � s$td��| �� S )NTz await wasn't used with future)r: �_asyncio_future_blockingr- r? r r r r � __await__ s zFuture.__await__)!r �
__module__� __qualname__r0 r/ r> r$ r
r rQ r# r r Z_future_repr_infor r r' �propertyr) �setterr. r3 r2 r9 r: r? r! rC rI rK rP rR �__iter__r r r r r s8
r c C s, z
| j }W n tk
r Y nX |� S | jS r( )r. �AttributeErrorr
)�futr. r r r � _get_loop s
rZ c C s | � � r d S | �|� d S r( )r9 rK )rY r? r r r � _set_result_unless_cancelled s r[ c C sX t | �}|tjjkr tj| j� S |tjjkr8tj| j� S |tjjkrPtj| j� S | S d S r( )rM �
concurrent�futuresr; r �args� TimeoutErrorr= )r% Z exc_classr r r �_convert_future_exc# s r` c C sR |� � r| �� | �� s d S |�� }|d k r<| �t|�� n|�� }| �|� d S r( )r9 r3 Z set_running_or_notify_cancelr! rP r` r? rK )r\ �sourcer! r? r r r � _set_concurrent_future_state/ s rb c C sT |� � r d S | � � r |�� n2| �� }|d k r>|�t|�� n| �� }|�|� d S r( )r9 r3 r! rP r` r? rK )ra �destr! r? r r r �_copy_future_state> s
rd c s� t ��s t�tjj�s td��t ��s._set_statec s2 | � � r.�d ks�� kr"��� n ���j� d S r( )r9 r3 �call_soon_threadsafe)� destination)� dest_loopra � source_loopr r �_call_check_cancelh s
z)_chain_future.._call_check_cancelc sJ �� � r �d k r ��� r d S �d ks,��kr8� �| � n��� �| � d S r( )r9 Z is_closedrg )ra )rf ri rh rj r r �_call_set_stateo s �� z&_chain_future.._call_set_state)r rL r\ r] r rO rZ rC )ra rh rk rl r )rf ri rh ra rj r �
_chain_futureR s � �
rm r
c C s2 t | �r | S |d kr t�� }|�� }t| |� |S r( )r r r Z
create_futurerm )r" r Z
new_futurer r r r | s
r ) �__all__Zconcurrent.futuresr\ r@ Zloggingr � r r r r r r0 r1 r< �DEBUGZ STACK_DEBUGr Z _PyFuturerZ r[ r` rb rd rm r Z_asyncio� ImportErrorZ_CFuturer r r r � s8
q *