B
p!=p @ s d Z dZddlZddlZddlZddlZddlmZ ddlm Z ddlm
Z
ej Z ej Z ej
Z
ejZejZejZejZejd ZG dd d ZeZd
d Zd d
Zdd Zdd Zdd ZddddZy ddl Z W n e k
r Y n X e j ZZ dS )z.A Future class similar to the one in PEP 3148.)CancelledError TimeoutErrorInvalidStateErrorFuture wrap_futureisfuture N ) base_futures)events)format_helpersc @ s e Zd ZdZeZdZdZdZdZ dZ
dZ ddddZ e
jZdd Zd d
Zed d Zejd
d Zdd Zdd Zdd Zdd Zdd Zdd Zdd Zdd d d Z d d Z d!d" Z d#d$ Z d%d& Z e Z!dS )'r a, This class is *almost* compatible with concurrent.futures.Future.
Differences:
- This class is not thread-safe.
- result() and exception() do not take a timeout argument and
raise an exception when the future isn't done yet.
- Callbacks registered with add_done_callback() are always called
via the event loop's call_soon().
- This class is not compatible with the wait() and as_completed()
methods in the concurrent.futures package.
(In Python 3.4 or later we may be able to unify the implementations.)
NF)loopc C s@ |dkrt | _n|| _g | _| j r )format __class____name__join
_repr_info)r r r r __repr__Y 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__] s
zFuture.__del__c C s | j S )N)r" )r r r r _log_tracebackm 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' q s c C s | j }|dkrtd|S )z-Return the event loop the Future is bound to.Nz!Future object is not initialized.)r RuntimeError)r r r r r get_loopw s zFuture.get_loopc C s& d| _ | jtkrdS t| _| dS )zCancel the future and schedule callbacks.
If the future is already done or cancelled, return False. Otherwise,
change the future's state to cancelled, schedule the callbacks and
return True.
FT)r" _state_PENDING
_CANCELLED_Future__schedule_callbacks)r r r r cancel~ s
z
Future.cancelc C sL | j dd }|sdS g | j dd< x"|D ]\}}| jj|| |d q*W dS )zInternal: Ask the event loop to call all callbacks.
The callbacks are scheduled to be called as soon as possible. Also
clears the callback list.
N)r% )r r call_soon)r Z callbackscallbackctxr r r Z__schedule_callbacks s zFuture.__schedule_callbacksc C s
| j tkS )z(Return True if the future was cancelled.)r- r/ )r r r r cancelled s zFuture.cancelledc C s
| j tkS )zReturn True if the future is done.
Done means either that a result / exception are available, or that the
future was cancelled.
)r- r. )r r r r done s z Future.donec C s< | j tkrt| j tkr tdd| _| jdk r6| j| jS )a Return the result this future represents.
If the future has been cancelled, raises CancelledError. If the
future's result isn't yet available, raises InvalidStateError. If
the future is done and has an exception set, this exception is raised.
zResult is not ready.FN)r- r/ r _FINISHEDr r" r# _result)r r r r result s
z
Future.resultc C s, | j tkrt| j tkr tdd| _| jS )a& Return the exception that was set on this future.
The exception (or None if no exception was set) is returned only if
the future is done. If the future has been cancelled, raises
CancelledError. If the future isn't done yet, raises
InvalidStateError.
zException is not set.F)r- r/ r r7 r r" r# )r r r r r s
zFuture.exception)r% c C sB | j tkr | jj|| |d n |dkr.t }| j||f dS )zAdd a callback to be run when the future becomes done.
The callback is called with a single argument - the future object. If
the future is already done when this is called, the callback is
scheduled with call_soon.
)r% N)r- r. r r2 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 dd< |S )z}Remove all instances of a callback from the "call when done" list.
Returns the number of callbacks removed.
c s g | ]\}}| kr||fqS r r ).0fr4 )r< r r
s z/Future.remove_done_callback..N)r len)r r< Zfiltered_callbacksZ
removed_countr )r< r remove_done_callback s
zFuture.remove_done_callbackc C s4 | j tkr td| j | || _t| _ | dS )zMark the future done and set its result.
If the future is already done when this method is called, raises
InvalidStateError.
z{}: {!r}N)r- r. r r r8 r7 r0 )r r9 r r r
set_result s
zFuture.set_resultc C s^ | j tkr td| j | t|tr,| }t|tkr@td|| _t | _ |
d| _ dS )zMark the future done and set an exception.
If the future is already done when this method is called, raises
InvalidStateError.
z{}: {!r}zPStopIteration interacts badly with generators and cannot be raised into a FutureTN) r- r. r r
isinstancetype
StopIteration TypeErrorr# r7 r0 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)r6 _asyncio_future_blockingr+ r9 )r r r r __await__ s zFuture.__await__)"r
__module__ __qualname____doc__r. r- r8 r# r r rI r" r r Z_future_repr_infor r r& propertyr' setterr, r1 r0 r5 r6 r9 r r= rB rC rH rJ __iter__r r r r r s6 r c C s, y
| j }W n tk
r Y nX | S | jS )N)r, AttributeErrorr )futr, r r r _get_loop s
rS c C s | r dS | | dS )z?Helper setting the result only if the future was not cancelled.N)r5 rC )rR r9 r r r _set_result_unless_cancelled s rT c C sZ | s t| r | | s(dS | }|dk rD| | n| }| | dS )z8Copy state from a future to a concurrent.futures.Future.N) r6 AssertionErrorr5 r1 Z set_running_or_notify_cancelr rH r9 rC )
concurrentsourcer r9 r r r _set_concurrent_future_state& s rX c C sh | s t| rdS | r$t| r6| n.| }|dk rR|| n| }|| dS )zqInternal helper to copy state from another Future.
The other Future may be a concurrent.futures.Future.
N)r6 rU r5 r1 r rH r9 rC )rW destr r9 r r r _copy_future_state5 s
rZ c s t s ttjjs tdt s._set_statec s2 | r.d ks kr" n j d S )N)r5 r1 call_soon_threadsafe) destination) dest_looprW source_loopr r _call_check_cancel_ s
z)_chain_future.._call_check_cancelc sJ r d k r r d S d ks,kr8 | n | d S )N)r5 Z is_closedr] )rW )r\ r_ r^ r` r r _call_set_statef s z&_chain_future.._call_set_state)r rD rV futuresr rG rS r= )rW r^ ra rb r )r\ r_ r^ rW r` r
_chain_futureI s
rd )r c C sN t | r | S t| tjjs(td| |dkr8t }| }t | | |S )z&Wrap concurrent.futures.Future object.z+concurrent.futures.Future is expected, got N)
r rD rV rc r rU r
r
Z
create_futurerd )r! r Z
new_futurer r r r s s
r ) rM __all__Zconcurrent.futuresrV r: Zloggingr r r
r r r r r r. r/ r7 DEBUGZ STACK_DEBUGr Z _PyFuturerS rT rX rZ rd r Z_asyncio ImportErrorZ_CFuturer r r r s<
q *