U
e5d @ sr d Z ddlZddlZddlZddlZddlZddlZddlZddl Z ddl
m Z ddl
m Z ddl
m
Z
ddl
mZ ddl
mZ dd l mZ edjZdAd
d ZdBd d
ZdCddZdd ZG dd dejZeZz ddlZW n e k
r Y n X ej ZZ ddddZ ejj Z ejj Z ejj!Z!dde!dddZ"dd Z#dd d d Z$d d Z%d!d" Z&ddd#d$d%Z'ej(d&d' Z)dDdd d(d)Z*dd d*d+Z+ej(d,d- Z,ee,_G d.d/ d/ej-Z.dd0d1d2d3Z/dd d4d5Z0d6d7 Z1e 2 Z3i Z4d8d9 Z5d:d; Z6dd? Z8e5Z9e8Z:e6Z;e7Ze6Z?e7Z@dS )E)Task create_taskFIRST_COMPLETEDFIRST_EXCEPTION
ALL_COMPLETEDwaitwait_for as_completedsleepgathershield
ensure_futurerun_coroutine_threadsafe current_task all_tasks_register_task_unregister_task _enter_task _leave_task N )
base_tasks)
coroutines)events)
exceptions)futures)
_is_coroutinec C s | d krt } t| S N)r get_running_loop_current_tasksgetloop r" %/usr/lib64/python3.8/asyncio/tasks.pyr " s r c s^ d krt d}z tt}W qL tk
rF |d7 }|dkrB Y qX qLq fdd|D S )Nr r c s& h | ] }t | kr| s|qS r" )r _get_loopdone.0tr r" r# < s z all_tasks..)r r list
_all_tasks RuntimeErrorr! iZtasksr" r r# r ) s r c s^ d krt d}z tt}W qL tk
rF |d7 }|dkrB Y qX qLq fdd|D S )Nr r r$ c s h | ]}t | kr|qS r" )r r% r' r r" r# r* U s z$_all_tasks_compat..)r get_event_loopr+ r, r- r. r" r r# _all_tasks_compat@ s r1 c C s4 |d k r0z
| j }W n tk
r& Y n
X || d S r )set_nameAttributeError)tasknamer2 r" r" r# _set_task_nameX s
r6 c s e Zd ZdZed$ddZed%ddZddd fdd
Z fd
d Zd d
Z dd Z
dd Z dd Z dd Z
dd ZddddZdddd d Zd d Zd& fd d! Zd"d# Z ZS )'r TNc C s( t jdtdd |d kr t }t|S )NzVTask.current_task() is deprecated since Python 3.7, use asyncio.current_task() instead
stacklevel)warningswarnDeprecationWarningr r0 r clsr! r" r" r# r t s zTask.current_taskc C s t jdtdd t|S )NzPTask.all_tasks() is deprecated since Python 3.7, use asyncio.all_tasks() insteadr7 r8 )r: r; r< r1 r= r" r" r# r s
zTask.all_tasks)r! r5 c s t j|d | jr | jd= t|s:d| _td||d krRdt | _n
t || _d| _
d | _ || _ t
| _| jj| j| jd t| d S )Nr Fz a coroutine was expected, got zTask-context)super__init___source_tracebackr iscoroutine_log_destroy_pending TypeError_task_name_counter_namestr _must_cancel _fut_waiter_coro contextvarsZ copy_context_context_loop call_soon _Task__stepr )selfcoror! r5 __class__r" r# rC 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!)r4 messageZsource_traceback) Z_stater Z_PENDINGrF rD rP Zcall_exception_handlerrB __del__)rS rA rU r" r# rX s
z Task.__del__c C s
t | S r )r Z_task_repr_inforS r" r" r#
_repr_info s zTask._repr_infoc C s | j S r )rM rY r" r" r# get_coro s z
Task.get_coroc C s | j S r )rI rY r" r" r# get_name s z
Task.get_namec C s t || _d S r )rJ rI )rS valuer" r" r# r2 s z
Task.set_namec C s t dd S )Nz*Task does not support set_result operationr- )rS resultr" r" r#
set_result s zTask.set_resultc C s t dd S )Nz-Task does not support set_exception operationr^ )rS exceptionr" r" r#
set_exception s zTask.set_exception)limitc C s t | |S r )r Z_task_get_stack)rS rc r" r" r# get_stack s zTask.get_stack)rc filec C s t | ||S r )r Z_task_print_stack)rS rc re 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& rL cancelrK rY r" r" r# rh s
z Task.cancelc
s | r td| d|| jr>t|tjs8t }d| _| j}d | _t| j | zfz"|d krp| d }n
| |}W n t
k
r } z*| jrd| _t nt |j W 5 d }~X Y n tjk
r t Y n ttfk
r } zt | W 5 d }~X Y n tk
rL } zt | W 5 d }~X Y npX t|dd }|d k r@t|| j k rtd| d|d}| j j| j|| j d n|r|| krtd | }| j j| j|| j d n8d|_ |j | j | j d || _| jr>| j r>d| _n*td
| d |}| j j| j|| j d n||d kr`| j j| j| j d n\t !|rtd | d |}| j j| j|| j d n$td
|}| j j| j|| j d W 5 t
| j | d } X d S )Nz_step(): already done: z, F_asyncio_future_blockingzTask z got Future z attached to a different loopr@ 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 ZInvalidStateErrorrK
isinstanceCancelledErrorrM rL r rP r sendthrow
StopIterationrB rh r` r] KeyboardInterrupt
SystemExitrb
BaseExceptiongetattrr r% r- rQ rR rO ri add_done_callback
_Task__wakeupinspectZ isgenerator)rS excrT r_ Zblockingnew_excrU r" r# Z__step s
z Task.__stepc
C sJ z | W n, tk
r8 } z| | W 5 d }~X Y n
X | d } d S r )r_ rq rR )rS futurerv r" r" r# Z__wakeup[ s z
Task.__wakeup)N)N)N)__name__
__module__ __qualname__rF classmethodr r rC rX rZ r[ r\ r2 r` rb rd rf rh rR rt
__classcell__r" r" rU r# r b s$ !Tr )r5 c C s t }|| }t|| |S r )r r r r6 )rT r5 r! r4 r" r" r# r x s
r )r! timeout return_whenc s t | st| r(tdt| j | s4td|tt t
fkrPtd| d krbt nt
jdtdd 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: [The loop argument is deprecated since Python 3.8, and scheduled for removal in Python 3.10.r7 r8 c s h | ]}t | d qS r r r( fr r" r# r* s zwait..)r isfuturer rE rG typery
ValueErrorr r r r r r: r; r< set_wait)fsr! r~ r r" r r# r s
r c G s | s| d d S r )r& r` )waiterargsr" r" r# _release_waiter s r r c
s |d krt }ntjdtdd |d kr4| I d H S |dkrt| |d} | rX| S t| |dI d H z | W n. t j
k
r } zt |W 5 d }~X Y n
X t | }|
|t|}tt|}t| |d} | | zz|I d H W nP t j
k
rF | r$| Y W dS | | t| |dI d H Y nX | r^| W *S | | t| |dI d H t W 5 | X d S )Nr r7 r8 r r )r r r: r; r< r r& r_ _cancel_and_waitr rk TimeoutError
create_future
call_laterr functoolspartialrs rh remove_done_callback)futr~ r! rv r timeout_handlecbr" r" r# r sL
r c
s | d |d k r"||tt| fdd}| D ]}|| q@zI d H W 5 d k rp | D ]}|| qtX t t }}| D ]"}| r| | q| | q||fS )Nc sZ d8 dks4t ks4tkrV| sV| d k rVd k rD sVd d S )Nr r )r r cancelledra rh r& r` r Zcounterr r r r" r# _on_completion s
z _wait.._on_completion)
r r r lenrs rh r r r& add)r r~ r r! r r r& Zpendingr" r r# r s( r c sF | }tt|}| | z| |I d H W 5 | | X d S r )r r r r rs r rh )r r! r r r" r" r# r & s
r )r! r~ c # s t | st| r(tdt| j ddlm} |dd krPt
nt j dt
dd fdd t| D d fd
d }fd d
fdd}D ]}| qr|d k r҈||ttD ] }| V qd S )Nz#expect an iterable of futures, not r )Queuer r r7 r8 c s h | ]}t | d qS r r r r r" r# r* U s z as_completed..c s* D ]} | d q d S r )r
put_nowaitclearr )r r& todor" r# _on_timeoutX s
z!as_completed.._on_timeoutc s4 sd S | | s0d k r0 d S r )remover rh r )r& r r r" r# r ^ s
z$as_completed.._on_completionc s$ I d H } | d kr tj| S r )r r r r_ r )r& r" r#
_wait_for_onef s z#as_completed.._wait_for_one)r r r rE rG r ry Zqueuesr r r0 r: r; r< r rs r ranger )r r! r~ r r r r _r" )r r& r! r r r# r 7 s*
r c c s
d V d S r r" r" r" r" r# __sleep0u s r c sr | dkrt I d H |S |d kr*t }ntjdtdd | }|| tj ||}z|I d H W S |
X d S )Nr r r7 r8 ) r r r r: r; r< r r r Z _set_result_unless_cancelledrh )Zdelayr_ r! rx hr" r" r# r 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 )Nr? zRThe future belongs to a different loop than the one specified as the loop argumentr z:An asyncio.Future, a coroutine or an awaitable is required)r rE r r0 r rD r r r% r ru Z isawaitabler _wrap_awaitablerG )Zcoro_or_futurer! r4 r" r" r# r s
r c c s | E d H S r ) __await__)Z awaitabler" r" r# r s r c s* e Zd Zdd fdd
Zdd Z ZS )_GatheringFutureNr c s t j|d || _d| _d S )Nr F)rB rC _children_cancel_requested)rS childrenr! rU r" r# rC s z_GatheringFuture.__init__c C s6 | r dS d}| jD ]}| rd}q|r2d| _|S rg )r& r rh r )rS ZretZchildr" r" r# rh s
z_GatheringFuture.cancel)ry rz r{ rC rh r} r" r" rU r# r s r F)r! return_exceptionsc s |s<| d krt } ntjdtdd | g S fdd}i }g dd|D ]f}||krt|| d}| d krt |} ||k rd|_
d 7 |||< | | n|| } | qdt
| dS )
Nr r7 r8 c s d7 r$| s | d S sd| rFt }| d S | }|d k rd| d S krg } D ]8} | rt }n| }|d kr| }|| qtjrĈt n
| d S )Nr )
r& r ra r rk rb r_ appendr r` )r rv Zresultsresr Z nfinishedZnfutsouterr r" r# _done_callback s4
z gather.._done_callbackr r Fr )r r0 r: r; r< r r` r r r% rF rs r r )r! r Zcoros_or_futuresr Z
arg_to_futargr r" r r# r
s:
1
r
c st |d k rt jdtdd t| |d r0S t}| fdd fdd} | S ) Nr r7 r8 r c s\ r | s| d S | r. n*| }|d k rJ | n | d S r )r ra rh rb r` r_ )innerrv r r" r# _inner_done_callbacku s
z$shield.._inner_done_callbackc s s d S r )r& r r )r r r" r# _outer_done_callback s z$shield.._outer_done_callback) r: r; r< r r& r r% r rs )r r! r r" )r r r r# r P s
r c s: t stdtj fdd}| S )Nz A coroutine object is requiredc
sl zt t d W nN ttfk
r2 Y n6 tk
rf } z rT| W 5 d } ~ X Y nX d S )Nr )r Z
_chain_futurer rp ro rq Z set_running_or_notify_cancelrb )rv rT rx r! r" r# callback s
z*run_coroutine_threadsafe..callback)r rE rG
concurrentr FutureZcall_soon_threadsafe)rT r! r r" r r# r
s
r
c C s t | d S r )r, r r4 r" r" r# r 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! r4 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" r# r s
r c C s t | d S r )r, discardr r" r" r# r s r )r r r r r, r )N)N)N)N)A__all__Zconcurrent.futuresr rN r ru itertoolstypesr: weakref r r r r r r count__next__rH r r r1 r6 Z _PyFuturer Z_PyTaskZ_asyncio ImportErrorZ_CTaskr r r r r r r r r r coroutiner 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# s
#H,>
x?$