3
\a
@ s d ddddddddd d
d d g
Z d
dlZd
dlZd
dlZd
dlZd
dlZddlmZ ddlm Z ddlm
Z
ddlm Z ddlm Z ddl
m
Z
G dd d e jZeZy d
dlZW n ek
r Y n X ej ZZej jZej jZej jZe
ddedddZdd Ze
ddd dZe
d d Zddd d dZe
d.ddd!dZ ddd"d#Z e e d< de _ [ ddd$d Z e
d%d& Z!G d'd( d(e jZ"dd)d*d+d Z#ddd,d
Z$d-d Z%dS )/TaskFIRST_COMPLETEDFIRST_EXCEPTION
ALL_COMPLETEDwaitwait_for as_completedsleepasyncgathershield
ensure_futurerun_coroutine_threadsafe N )
base_tasks)compat)
coroutines)events)futures) coroutinec s e Zd Zej Zi ZdZedddZ edddZ
dd fdd
Z e j
rTd
d Zd d
ZddddZdddddZdd Zd fdd Zdd Z ZS ) r TNc C s |d krt j }| jj|S )N)r get_event_loop_current_tasksget)clsloop r %/usr/lib64/python3.6/asyncio/tasks.py current_task. s zTask.current_taskc s$ d krt j fdd| jD S )Nc s h | ]}|j kr|qS r )_loop).0t)r r r B s z!Task.all_tasks..)r r
_all_tasks)r r r )r r all_tasks: s zTask.all_tasks)r c sN t j|d | jr | jd= || _d | _d| _| jj| j | j j
j | d S )N)r r F) super__init___source_traceback_coro _fut_waiter _must_cancelr call_soon_step __class__r" add)selfcoror )r- r r r& D s z
Task.__init__c C sH | j tjkr8| jr8| dd}| jr,| j|d< | jj| tjj| d S )Nz%Task was destroyed but it is pending!)taskmessageZsource_traceback) Z_stater Z_PENDING_log_destroy_pendingr' r Zcall_exception_handlerFuture__del__)r/ contextr r r r5 S s
z Task.__del__c C s
t j| S )N)r Z_task_repr_info)r/ r r r
_repr_info^ s zTask._repr_info)limitc C s t j| |S )N)r Z_task_get_stack)r/ r8 r r r get_stacka s zTask.get_stack)r8 filec C s t j| ||S )N)r Z_task_print_stack)r/ r8 r: r r r print_stackx s zTask.print_stackc C s4 d| _ | j rdS | jd k r*| jj r*dS d| _dS )NFT)Z_log_tracebackdoner) cancelr* )r/ r r r r= s
z Task.cancelc sf | j r t|tjstj }d| _ | j}d | _| | jj| j< zy"|d krT|j d }n
|j
|}W n t k
r } z.| j rd| _ | j tj n | j
|j W Y d d }~X n tjk
r t j Y n| tk
r } z| j | W Y d d }~X nP tk
r( } z| j | W Y d d }~X n X t|dd }|d k r|j| jk rl| jj| jtdj| | n||r|| kr| jj| jtdj| n2d|_|j| j || _| j r| jj rd| _ n | jj| jtdj| | n^|d kr| jj| j nDtj |r.| jj| jtdj| | n| jj| jtdj| W d | jjj | j d } X d S )NF_asyncio_future_blockingz6Task {!r} got Future {!r} attached to a different loopz!Task cannot await on itself: {!r}z;yield was used instead of yield from in task {!r} with {!r}zIyield was used instead of yield from for generator in task {!r} with {!r}zTask got bad yield: {!r}) r*
isinstancer CancelledErrorr( r) r- r r sendthrow
StopIteration
set_exception
set_resultvaluer% r= Exception
BaseExceptiongetattrr+ r, RuntimeErrorformatr> add_done_callback_wakeupinspectZ isgeneratorpop)r/ excr0 resultZblocking)r- r r r, s~
z
Task._stepc C sJ y |j W n, tk
r8 } z| j| W Y d d }~X n
X | j d } d S )N)rQ rG r, )r/ futurerP r r r rM s z Task._wakeup)N)N)N)__name__
__module__ __qualname__weakrefWeakSetr" r r3 classmethodr r# r& r ZPY34r5 r7 r9 r; r= r, rM
__classcell__r r )r- r r s
!T)r timeout return_whenc # s t j| stj| r&tdt| j | s2td|tt t
fkrNtdj | d kr^t j
fddt| D } t| || E d H S )Nz expect a list of futures, not %sz#Set of coroutines/Futures is empty.z Invalid return_when value: {}c s h | ]}t | d qS ))r )r )r f)r r r r! 7 s zwait..)r isfuturer iscoroutine TypeErrortyperS
ValueErrorr r r rK r r set_wait)fsr rZ r[ r )r r r s c G s | j s| jd d S )N)r< rE )waiterargsr r r _release_waiter< s rg )r c c s |d krt j }|d kr"| E d H S |j }|j|t|}tjt|}t| |d} | j| zhy|E d H W n* t j
k
r | j | | j Y nX | j
r| j S | j | | j t j W d |j X d S )N)r )r r
create_future
call_laterrg functoolspartialr rL r r@ remove_done_callbackr= r< rQ TimeoutError)futrZ r re timeout_handlecbr r r r A s,
c # s |j d |d k r"|j|tt| fdd}x| D ]}|j| qBW zE d H W d d k rtj X t t }}x4| D ],}|j| |j r|j | q|j | qW ||fS )Nc s\ d8 dks6t ks6tkrX| j rX| j d k rXd k rFj j sXjd d S )Nr r )r r cancelled exceptionr= r< rE )r\ )counterr[ ro re r r _on_completion| s z _wait.._on_completion)
rh ri rg lenrL r= rb rl r< r. )rd rZ r[ r rt r\ r< pendingr )rs r[ ro re r rc o s&
rc )r rZ c # s t j| stj| r&tdt| j d k r2ntj fddt | D ddl
m } |dd fdd}fd d
t fd d }xD ]}|j
qW r|d k rʈj||x ttD ] }| V qW d S )
Nz expect a list of futures, not %sc s h | ]}t | d qS ))r )r )r r\ )r r r r! s z as_completed..r )Queue)r c s. x D ]} | j jd qW j d S )N)rl
put_nowaitclear)r\ )rt r< todor r _on_timeout s
z!as_completed.._on_timeoutc s6 sd S j | j| r2d k r2j d S )N)removerx r= )r\ )r< ro rz r r rt s
z$as_completed.._on_completionc 3 s$ j E d H } | d kr tj| j S )N)r r rm rQ )r\ )r< r r
_wait_for_one s z#as_completed.._wait_for_one)r r] r r^ r_ r` rS r r rb Zqueuesrw r rL ri rangeru )rd r rZ rw r{ r} r\ _r )rt r< r ro rz r r s
c
c sX | dkrd V |S |d kr"t j }|j }|jj| tj||}z
|E d H S |j X d S )Nr )r r rh r ri r Z _set_result_unless_cancelledr= )ZdelayrQ r rR hr r r r s
c C s t jdtdd t| |dS )Nz;asyncio.async() function is deprecated, use ensure_future() )
stacklevel)r )warningswarnDeprecationWarningr )coro_or_futurer r r r async_ s r c C s t j| r(|d k r$|| jk r$td| S tj| r^|d krBtj }|j| }|j rZ|j d= |S t
j r~t j
| r~tt| |dS tdd S )Nz$loop argument must agree with Futurer )r z:An asyncio.Future, a coroutine or an awaitable is requiredr$ )r r] r ra r r^ r r Z create_taskr' r ZPY35rN Z isawaitabler _wrap_awaitabler_ )r r r1 r r r r s
c c s | j E d H S )N) __await__)Z awaitabler r r r s r c s* e Zd Zdd fdd
Zdd Z ZS )_GatheringFutureN)r c s t j|d || _d| _d S )N)r F)r% r& _children_cancel_requested)r/ childrenr )r- r r r& $ s z_GatheringFuture.__init__c C s: | j r dS d}x| jD ]}|j rd}qW |r6d| _|S )NFT)r< r r= r )r/ ZretZchildr r r r= ) s z_GatheringFuture.cancel)rS rT rU r& r= rY r r )r- r r s r F)r return_exceptionsc s |s*| d krt j } | j jg S i xjt|D ]^}tj|sht|| d}| d kr`|j} d|_ n&|}| d kr||j} n|j| k rt
d| |< q8W fdd|D }t |t || ddd g fdd}x&t
|D ]\}}|jtj|| qW S ) N)r Fz)futures are tied to different event loopsc s g | ] } | qS r r )r arg)
arg_to_futr r
h s zgather..r c s j r |j s|j d S |j r@tj }slj| d S n,|jd k rf|j }slj| d S n|j}|| < d7 krjrjtj n
j d S )Nr )
r< rq rr r r@ rD Z
_exceptionZ_resultr rE )irn res) nchildren nfinishedouterresultsr r r _done_callbackn s*
z gather.._done_callback)r r rh rE rb r r] r r r3 ra ru r enumeraterL rj rk )r r Zcoros_or_futuresr rn r r r r )r r r r r r r r
8 s8
c s@ t | |d}|j r|S |j}|j fdd}|j| S )N)r c s\ j r | j s| j d S | j r. j n*| j }|d k rJ j| n j| j d S )N)rq rr r= rD rE rQ )innerrP )r r r r s
z shield.._done_callback)r r< r rh rL )r r r r r )r r r s
c s: t j stdtjj fdd}j| S )Nz A coroutine object is requiredc sT yt jt d W n6 tk
rN } zj r<j| W Y d d } ~ X nX d S )N)r )r Z
_chain_futurer rG Z set_running_or_notify_cancelrD )rP )r0 rR r r r callback s
z*run_coroutine_threadsafe..callback)r r^ r_
concurrentr r4 Zcall_soon_threadsafe)r0 r r r )r0 rR r r r
s
)N)&__all__Zconcurrent.futuresr rj rN r rV r r r r r r r4 r Z_PyTaskZ_asyncio ImportErrorZ_CTaskr r r r rg r rc r r r globalsrS r r r r
r r
r r r r sX
s
- -8
W5