B
§ @ s< d Z ddlZddlZddlZddlZddlZddlZddlZddl m
Z
ddl m Z ddl m Z ddl m
Z
ddl mZ d=d d
Zd>d d Zd?d
dZG dd de
jZeZy ddlZW n ek
r Y n X ej ZZdd Zej
jZej
jZej
jZddedddZ dd Z ddddZ dd Z d d Z ddd d d!Z!ejd"d# Z"d@ddd$d%Z#ddd&d'Z$ed(d) Z%G d*d+ d+e
j&Z'dd,d-d.d/Z(ddd0d1Z)d2d3 Z*e+ Z,i Z-d4d5 Z.d6d7 Z/d8d9 Z0d:d; Z1e.Z2e1Z3e/Z4e0Z5y$dd5 s z all_tasks..)r r list
_all_tasks RuntimeError)r itasksr )r r r " s r c sb d krt d}x:y tt}W n& tk
rH |d7 }|dkrD Y qX P qW fdd|D S )Nr r i c s h | ]}t | kr|qS r )r r! )r# r$ )r r r r% N s z$_all_tasks_compat..)r get_event_loopr& r' r( )r r) r* r )r r _all_tasks_compat9 s r, c s e Zd ZdZed ddZed ddZdd fdd
Z fd
d Zd d
Z dd Z
dd Z ddddZ dddddZ
dd Zd fdd Zd d Z ZS )!r TNc C s( t jdtdd |d kr t }t|S )NzETask.current_task() is deprecated, use asyncio.current_task() instead )
stacklevel)warningswarnPendingDeprecationWarningr r+ r )clsr r r r r c s zTask.current_taskc C s t jdtdd t|S )Nz?Task.all_tasks() is deprecated, use asyncio.all_tasks() insteadr- )r. )r/ r0 r1 r, )r2 r r r r r s s zTask.all_tasks)r c sv t j|d | jr | jd= t|s:d| _td|d| _d | _|| _ t
| _ | j
j| j| j d t| d S )N)r Fz a coroutine was expected, got )context)super__init___source_tracebackr iscoroutine_log_destroy_pending TypeError _must_cancel _fut_waiter_coro contextvarsZ copy_context_context_loop call_soon _Task__stepr )selfcoror ) __class__r r r6 s
z
Task.__init__c sF | j tjkr8| jr8| dd}| jr,| j|d< | j| t d S )Nz%Task was destroyed but it is pending!)taskmessageZsource_traceback) Z_stater Z_PENDINGr9 r7 r@ Zcall_exception_handlerr5 __del__)rC r4 )rE r r rH s
z Task.__del__c C s
t | S )N)r Z_task_repr_info)rC r r r
_repr_info s zTask._repr_infoc C s t dd S )Nz*Task does not support set_result operation)r( )rC resultr r r
set_result s zTask.set_resultc C s t dd S )Nz-Task does not support set_exception operation)r( )rC exceptionr r r
set_exception s zTask.set_exception)limitc C s t | |S )N)r Z_task_get_stack)rC rN r r r get_stack s zTask.get_stack)rN filec C s t | ||S )N)r Z_task_print_stack)rC rN rP r r r print_stack s zTask.print_stackc C s4 d| _ | rdS | jd k r*| j r*dS d| _dS )NFT)Z_log_tracebackr" r< cancelr; )rC r r r rR s
z Task.cancelc
s | r td| d|| jr>t|tjs8t }d| _| j}d | _t| j | zhy"|d krp|
d }n
| |}W n t k
r } z0| jrd| _t
t nt
|j W d d }~X Y n tjk
r t
Y n tk
r } zt
| W d d }~X Y n tk
rN } zt
| W d d }~X Y npX t|dd }|d k rBt|| j k rtd| d|d}| j j| j|| jd n|r|| krtd | }| j j| j|| jd n8d|_|j| j | jd || _| jr@| j r@d| _n*td
| d |}| j j| j|| jd n||d krb| j j| j| jd n\t |rtd | d |}| j j| j|| jd n$td
|}| j j| j|| jd W d t | j | d } X d S )Nz_step(): already done: z, F_asyncio_future_blockingzTask z got Future z attached to a different loop)r4 z Task cannot await on itself: z-yield was used instead of yield from in task z with z;yield was used instead of yield from for generator in task zTask got bad yield: ) r" r ZInvalidStateErrorr;
isinstanceCancelledErrorr= r< r r@ sendthrow
StopIterationr5 rM rK valuerR Exception
BaseExceptiongetattrr! r( rA rB r? rS add_done_callback
_Task__wakeupinspectZ isgeneratorr )rC excrD rJ Zblockingnew_exc)rE r r Z__step s|
z Task.__stepc
C sJ y | W n, tk
r8 } z| | W d d }~X Y n
X | d } d S )N)rJ rZ rB )rC futurer` r r r Z__wakeup< s z
Task.__wakeup)N)N)N)__name__
__module__ __qualname__r9 classmethodr r r6 rH rI rK rM rO rQ rR rB r^
__classcell__r r )rE r r Q s !Tr c C s t }|| S )N)r r r )rD r r r r r Y s r )r timeout return_whenc s t | st| r(tdt| j | s4td|tt t
fkrPtd| d kr`t fddt
| D } t| || I d H S )Nz expect a list of futures, not z#Set of coroutines/Futures is empty.zInvalid return_when value: c s h | ]}t | d qS ))r )r )r# f)r r r r% s zwait..)r isfuturer r8 r: typerc
ValueErrorr r r r r+ set_wait)fsr rh ri r )r r r i s r c G s | s| d d S )N)r" rK )waiterargsr r r _release_waiter s rs )r c s |d krt }|d kr"| I d H S |dkrVt| |d} | rF| S | t | }| |t
|}t t
|}t| |d} |
| zry|I d H W n* tjk
r | | | Y nX | r| S | | t| |dI d H t W d | X d S )Nr )r )r r+ r r" rJ rR r TimeoutError
create_future
call_laterrs functoolspartialr] rU remove_done_callback_cancel_and_wait)futrh r rq timeout_handlecbr r r r s8
r c
s | d |d k r"||tt| fdd}x| D ]}|| qBW zI d H W d d k rt x| D ]}|| qzW X t t }}x*| D ]"}| r| | q| | qW ||fS )Nc sZ d8 dks4t ks4tkrV| sV| d k rVd k rD sVd d S )Nr r )r r cancelledrL rR r" rK )rj )counterri r| rq r r _on_completion s z _wait.._on_completion)
ru rv rs lenr] rR ry rn r" add)rp rh ri r r rj r" Zpendingr )r ri r| rq r ro s(
ro c sF | }tt|}| | z| |I d H W d | | X d S )N)ru rw rx rs r] rR ry )r{ r rq r} r r r rz s
rz )r rh c # s t | st| r(tdt| j d k r4nt fddt | D ddl
m } |dd fdd}fd d
fd d }xD ]}| qW r|d k rȈ
||x ttD ] }| V qW d S )
Nz expect a list of futures, not c s h | ]}t | d qS ))r )r )r# rj )r r r r% s z as_completed..r )Queue)r c s. x D ]} | d qW d S )N)ry
put_nowaitclear)rj )r r" todor r _on_timeout s
z!as_completed.._on_timeoutc s4 sd S | | s0d k r0 d S )N)remover rR )rj )r" r| r r r r # s
z$as_completed.._on_completionc s$ I d H } | d kr tj| S )N)r r rt rJ )rj )r" r r
_wait_for_one+ s z#as_completed.._wait_for_one)r rk r r8 r: rl rc r r+ rn Zqueuesr r] rv ranger )rp r rh r r r rj _r )r r" r r| r r r s
r c c s
d V d S )Nr r r r r __sleep0: s r c s\ | dkrt I d H |S |d kr(t }| }|| tj||}z
|I d H S | X d S )Nr )r r r+ ru rv r Z _set_result_unless_cancelledrR )ZdelayrJ r rb hr r r r F s
r c C s t | r6|d krt }|| }|jr2|jd= |S t| rb|d k r^|t| k r^t d| S t
| r|t t
| |dS tdd S )Nr3 z$loop argument must agree with Future)r z:An asyncio.Future, a coroutine or an awaitable is required)r r8 r r+ r r7 r rk r! rm r_ Z isawaitabler _wrap_awaitabler: )Zcoro_or_futurer rF r r r r X s
r c c s | E d H S )N) __await__)Z awaitabler r r r o s r c s* e Zd Zdd fdd
Zdd Z ZS )_GatheringFutureN)r c s t j|d || _d| _d S )N)r F)r5 r6 _children_cancel_requested)rC childrenr )rE r r r6 s z_GatheringFuture.__init__c C s: | r dS d}x| jD ]}| rd}qW |r6d| _|S )NFT)r" r rR r )rC retZchildr r r rR s z_GatheringFuture.cancel)rc rd re r6 rR rg r r )rE r r y s r F)r return_exceptionsc s |s*| d krt } | g S fdd}i }g ddxn|D ]f}||krt|| d}| d kr~t|} ||k rd|_d7 |||< || n|| } | qTW t
| dS )Nc s d7 r$| s | d S sd| rFt }| d S | }|d k rd| d S krg }x@ D ]8} | rt }n| }|d kr| }|| qvW jrȈt n
| d S )Nr )
r" r~ rL r rU rM rJ appendr rK )r{ r` resultsZres)r nfinishednfutsouterr r r _done_callback s4
z gather.._done_callbackr )r Fr ) r r+ ru rK r r r! r9 r] r r )r r Zcoros_or_futuresr Z
arg_to_futargr{ r )r r r r r r r
s2
1
r
c s\ t | |d rS t}| fdd fdd} | S )N)r c s\ r | s| d S | r. n*| }|d k rJ | n | d S )N)r~ rL rR rM rK rJ )innerr` )r r r _inner_done_callback# s
z$shield.._inner_done_callbackc s s d S )N)r" ry )r )r r r r _outer_done_callback4 s z$shield.._outer_done_callback)r r" r r! ru r] )r r r r )r r r r r s
r c s: t stdtj fdd}| S )Nz A coroutine object is requiredc
sT yt t d W n6 tk
rN } z r<| W d d } ~ X Y nX d S )N)r )r Z
_chain_futurer rZ Z set_running_or_notify_cancelrM )r` )rD rb r r r callbackF s
z*run_coroutine_threadsafe..callback)r r8 r:
concurrentr FutureZcall_soon_threadsafe)rD r r r )rD rb r r r
= s
r
c C s t | d S )N)r' r )rF r r r r Z s r c C s4 t | }|d k r(td|d|d|t | < d S )NzCannot enter into task z while another task z is being executed.)r r r( )r rF r r r r r _ s
r c C s2 t | }||k r(td|d|dt | = d S )Nz
Leaving task z! does not match the current task .)r r r( )r rF r r r r r g s
r c C s t | d S )N)r' discard)rF r r r r o s r )r r r r r' r )N)N)N)N):__all__concurrent.futuresr r> rw r_ typesr/ weakref r r r r r r r r, Z _PyFuturer Z_PyTaskZ_asyncio ImportErrorZ_CTaskr r r r r rs r ro rz r r r r r r r r
r r
ZWeakSetr' r r r r r Z_py_register_taskZ_py_unregister_taskZ_py_enter_taskZ_py_leave_taskZ_c_register_taskZ_c_unregister_taskZ
_c_enter_taskZ
_c_leave_taskr r r r sv
}
9,7
m;$