U
ifb3 @ 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 ZddddZz ddlZW n ek
r Y n X ej ZZ dS ))Future wrap_futureisfuture N ) base_futures)events)
exceptions)format_helpersc @ s e Zd ZeZdZdZdZdZdZ dZ
ddddZ e j
Zdd Zdd Zed
d Zejd d Zd
d Zdd Zdd Zdd Zdd Zdd Zdd Zddd d Zd d Z d d! Z d"d# Z d$d% Z e Z dS )&r NFloopc C s@ |d krt | _n|| _g | _| j r )format __class____name__join
_repr_infor 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 exceptionfutureZsource_traceback)_Future__log_traceback
_exceptionr r r r
Zcall_exception_handler)r exccontextr 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 |rtdd| _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 Nr& )r r
call_soon)r Z callbackscallbackctxr 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$tdd| _| jd k r:| j| jS )NzResult is not ready.F) r/ r1 r CancelledError _FINISHEDInvalidStateErrorr# r$ _resultr r r r result s
z
Future.resultc C s0 | j tkrtj| j tkr$tdd| _| 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||fqS r r ).0fr8 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|tr0| }t|tkrDtd|| _t | _ |
d| _ d S )NrJ zPStopIteration interacts badly with generators and cannot be raised into a FutureT) r/ r0 r r=
isinstancetype
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
concurrentfuturesr; 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 ttjjs tdt 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 *