U
��,ab3�@s�dZdZddlZddlZddlZddlZddlmZddlm Z ddlm
Z
ddlm Z ej Z ej
Z
ejZejZejdZGd d
�d
�ZeZd d �Zd
d�Zdd�Zdd�Zdd�Zdd�Zdd�dd�Zz ddlZWne k
r�Yn XejZZ dS)z.A Future class similar to the one in PEP 3148.)�Future� wrap_future�isfuture�N�)� base_futures)�events)�
exceptions)�format_helpersc@s�eZdZdZeZdZdZdZdZ dZ
dZ dd�dd�Z 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)'ra,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��loopcCs@|dkrt��|_n||_g|_|j��r<t�t�d��|_ dS)z�Initialize the future.
The optional event_loop argument allows explicitly setting the event
loop object used by the future. If it's not provided, the future uses
the default event loop.
Nr)
r�get_event_loop�_loop�
_callbacksZ get_debugr �
extract_stack�sys� _getframe�_source_traceback��selfr �r�4/opt/alt/python38/lib64/python3.8/asyncio/futures.py�__init__Ds
�zFuture.__init__cCsd�|jjd�|����S)Nz<{} {}>� )�format� __class__�__name__�join�
_repr_info�rrrr�__repr__Vs
�zFuture.__repr__cCsF|js
dS|j}|jj�d�||d�}|jr6|j|d<|j�|�dS)Nz exception was never retrieved)�message� exception�futureZsource_traceback)�_Future__log_traceback�
_exceptionrrrr
Zcall_exception_handler)r�exc�contextrrr�__del__Zs �
zFuture.__del__cCs|jS�N)r#r rrr�_log_tracebackjszFuture._log_tracebackcCst|�rtd��d|_dS)Nz'_log_traceback can only be set to FalseF)�bool�
ValueErrorr#)r�valrrrr)nscCs|j}|dkrtd��|S)z-Return the event loop the Future is bound to.Nz!Future object is not initialized.)r
� RuntimeErrorrrrr�get_looptszFuture.get_loopcCs&d|_|jtkrdSt|_|��dS)z�Cancel 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_callbacksr rrr�cancel{s
z
Future.cancelcCsH|jdd�}|sdSg|jdd�<|D]\}}|jj|||d�q(dS)z�Internal: 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&)rr
� call_soon)rZ callbacks�callback�ctxrrrZ__schedule_callbacks�s zFuture.__schedule_callbackscCs
|jtkS)z(Return True if the future was cancelled.)r/r1r rrr� cancelled�szFuture.cancelledcCs
|jtkS)z�Return True if the future is done.
Done means either that a result / exception are available, or that the
future was cancelled.
)r/r0r rrr�done�sz Future.donecCs@|jtkrtj�|jtkr$t�d��d|_|jdk r:|j�|jS)aReturn 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/r1r�CancelledError� _FINISHED�InvalidStateErrorr#r$�_resultr rrr�result�s
z
Future.resultcCs0|jtkrtj�|jtkr$t�d��d|_|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/r1rr:r;r<r#r$r rrrr!�s
zFuture.exceptionr4cCsB|jtkr |jj|||d�n |dkr.t��}|j�||f�dS)z�Add 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.
r4N)r/r0r
r5� contextvarsZ copy_contextr�append)r�fnr&rrr�add_done_callback�s
zFuture.add_done_callbackcs<�fdd�|jD�}t|j�t|�}|r8||jdd�<|S)z}Remove all instances of a callback from the "call when done" list.
Returns the number of callbacks removed.
cs g|]\}}|�kr||f�qSrr)�.0�fr7�rArr�
<listcomp>�s�z/Future.remove_done_callback.<locals>.<listcomp>N)r�len)rrAZfiltered_callbacksZ
removed_countrrEr�remove_done_callback�s
�zFuture.remove_done_callbackcCs8|jtkr t�|j�d|����||_t|_|��dS)z�Mark the future done and set its result.
If the future is already done when this method is called, raises
InvalidStateError.
�: N)r/r0rr<r=r;r2)rr>rrr�
set_result�s
zFuture.set_resultcCsb|jtkr t�|j�d|����t|t�r0|�}t|�tkrDtd��||_t |_|�
�d|_ dS)z�Mark the future done and set an exception.
If the future is already done when this method is called, raises
InvalidStateError.
rIzPStopIteration interacts badly with generators and cannot be raised into a FutureTN) r/r0rr<�
isinstance�type�
StopIteration� TypeErrorr$r;r2r#)rr!rrr�
set_exception�s
zFuture.set_exceptionccs,|��sd|_|V|��s$td��|��S)NTz await wasn't used with future)r9�_asyncio_future_blockingr-r>r rrr� __await__s zFuture.__await__)"r�
__module__� __qualname__�__doc__r0r/r=r$r
rrPr#rrZ_future_repr_infor r r'�propertyr)�setterr.r3r2r8r9r>r!rBrHrJrOrQ�__iter__rrrrr s:
rcCs,z
|j}Wntk
r YnX|�S|jSr()r.�AttributeErrorr
)�futr.rrr� _get_loops
rZcCs|��r dS|�|�dS)z?Helper setting the result only if the future was not cancelled.N)r8rJ)rYr>rrr� _set_result_unless_cancelled sr[cCsXt|�}|tjjkr tj|j�S|tjjkr8tj|j�S|tjjkrPtj|j�S|SdSr()rL�
concurrent�futuresr:r�args� TimeoutErrorr<)r%Z exc_classrrr�_convert_future_exc#s r`cCs^|��s t�|��r |��|��s(dS|��}|dk rH|�t|��n|��}|� |�dS)z8Copy state from a future to a concurrent.futures.Future.N)
r9�AssertionErrorr8r3Z set_running_or_notify_cancelr!rOr`r>rJ)r\�sourcer!r>rrr� _set_concurrent_future_state/s rccCsl|��s t�|��rdS|��r$t�|��r6|��n2|��}|dk rV|�t|��n|��}|�|�dS)zqInternal helper to copy state from another Future.
The other Future may be a concurrent.futures.Future.
N) r9rar8r3r!rOr`r>rJ)rb�destr!r>rrr�_copy_future_state>s
recs�t��s t�tjj�s td��t��s<t�tjj�s<td��t��rLt��nd�t��r`t��nd�dd�����fdd�}����fdd �}��|���|�dS)
aChain two futures so that when one completes, so does the other.
The result (or exception) of source will be copied to destination.
If destination is cancelled, source gets cancelled too.
Compatible with both asyncio.Future and concurrent.futures.Future.
z(A future is required for source argumentz-A future is required for destination argumentNcSs"t|�rt||�n
t||�dSr()rrerc)r"�otherrrr�
_set_statebs z!_chain_future.<locals>._set_statecs2|��r.�dks��kr"���n ���j�dSr()r8r3�call_soon_threadsafe)� destination)� dest_looprb� source_looprr�_call_check_cancelhs
z)_chain_future.<locals>._call_check_cancelcsJ���r �dk r ���r dS�dks,��kr8��|�n����|�dSr()r8Z is_closedrh)rb)rgrjrirkrr�_call_set_stateos�� z&_chain_future.<locals>._call_set_state)rrKr\r]rrNrZrB)rbrirlrmr)rgrjrirbrkr�
_chain_futureRs � �
rnr
cCsNt|�r |St|tjj�s(td|����|dkr8t��}|��}t ||�|S)z&Wrap concurrent.futures.Future object.z+concurrent.futures.Future is expected, got N)
rrKr\r]rrarr Z
create_futurern)r"r Z
new_futurerrrr|s�
r) rT�__all__Zconcurrent.futuresr\r?Zloggingr�rrrr rr0r1r;�DEBUGZ STACK_DEBUGrZ _PyFuturerZr[r`rcrernrZ_asyncio� ImportErrorZ_CFuturerrrr�<module>s:
q *
|