a
XC?hs @ s d Z ddlZddlZddlZddlZddlZddlZddlZddl Z ddlm
Z
ddl m Z ddl m
Z
ddl mZ ddl mZ dd l mZ dd
l
mZ edjZdDd d ZdEd
dZdFddZdd ZG dd dejZeZz ddl Z W n e y Y n 0 e j 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*dGdd d)d*Z+dd d+d,Z,ej)d-d. Z-ee-_G d/d0 d0ej.Z/dd1d2d3d4Z0dd1d2d5d6Z1dd d7d8Z2d9d: Z3e 4 Z5i Z6d;d< Z7d=d> Z8d?d@ Z9dAdB Z:e7Z;e:Zz$ddCl m7Z7m:Z:m8Z8m9Z9m5Z5m6Z6 W n e yv Y n0 e7Z?e:Z@e8ZAe9ZBdS )H)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) GenericAlias )
base_tasks)
coroutines)events)
exceptions)futures)
_is_coroutinec C s | d u rt } t| S N)r get_running_loop_current_tasksgetloop r# 2/opt/alt/python39/lib64/python3.9/asyncio/tasks.pyr # s r c s\ d u rt d}z tt}W qJ tyD |d7 }|dkr@ Y q0 qJq fdd|D S )Nr r c s& h | ] }t | u r| 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 u rt d}z tt}W qJ tyD |d7 }|dkr@ Y q0 qJq fdd|D S )Nr r r% c s h | ]}t | u r|qS r# )r r& r( r! r# r$ r+ V z$_all_tasks_compat..)r get_event_loopr, r- r. r/ r# r! r$ _all_tasks_compatA s r3 c C s2 |d ur.z
| j }W n ty$ Y n
0 || d S r )set_nameAttributeError)tasknamer4 r# r# r$ _set_task_nameY s
r8 c s e Zd ZdZddd fdd
Z fddZeeZdd Z d
d Z
d d
Z dd Z dd Z
dd ZddddZdddddZd ddZd! fd d Zd d Z ZS )"r TN)r" r7 c s t j|d | jr | jd= t|s:d| _td||d u rRdt | _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" r7 __class__r# r$ r= u 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!)r6 messageZsource_traceback) Z_stater Z_PENDINGr@ r> rJ Zcall_exception_handlerr< __del__)rM r; rO r# r$ rR s
z Task.__del__c C s
t | S r )r Z_task_repr_inforM r# r# r$
_repr_info s zTask._repr_infoc C s | j S r )rG rS r# r# r$ get_coro s z
Task.get_coroc C s | j S r )rC rS r# r# r$ get_name s z
Task.get_namec C s t || _d S r )rD rC )rM valuer# r# r$ r4 s z
Task.set_namec C s t dd S )Nz*Task does not support set_result operationr. )rM resultr# r# r$
set_result s zTask.set_resultc C s t dd S )Nz-Task does not support set_exception operationrX )rM exceptionr# r# r$
set_exception s zTask.set_exception)limitc C s t | |S r )r Z_task_get_stack)rM r] r# r# r$ get_stack s zTask.get_stack)r] filec C s t | ||S r )r Z_task_print_stack)rM r] r_ r# r# r$ print_stack s zTask.print_stackc C s> d| _ | rdS | jd ur.| jj|dr.dS d| _|| _dS NFmsgT)Z_log_tracebackr' rF cancelrE _cancel_message)rM rc r# r# r$ rd s
z Task.cancelc
s | r td| d|| jr>t|tjs8| }d| _| j}d | _t | j
| zz"|d u rp| d }n
| |}W n t
y } z:| jrd| _t j| jd nt |j W Y d }~n d }~0 tjy } z || _t W Y d }~nd }~0 ttfy> } z t | W Y d }~nd }~0 typ } zt | W Y d }~n~d }~0 0 t|dd }|d urjt|| j
urtd| d|d}| j
j | j || j d n|r>|| u rtd
| }| j
j | j || j d n>d|_ |j | j!| j d || _| jrh| jj| jdrhd| _n*td | d |}| j
j | j || j d n||d u r| j
j | j | j d n\t"#|rtd
| d |}| j
j | j || j d n$td|}| j
j | j || j d W t$| j
| d } nt$| j
| d } 0 d S )Nz_step(): already done: z, Frb _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 ZInvalidStateErrorrE
isinstanceCancelledError_make_cancelled_errorrG rF r rJ sendthrow
StopIterationr< rd re rZ rW Z_cancelled_excKeyboardInterrupt
SystemExitr\
BaseExceptiongetattrr r& r. rK rL rI rf add_done_callback
_Task__wakeupinspectZ isgeneratorr )rM excrN rY Zblockingnew_excrO r# r$ Z__step s $ $
z Task.__stepc
C sL z | W n. ty: } z| | W Y d }~nd }~0 0 | d } d S r )rY ro rL )rM futurert r# r# r$ Z__wakeupF s z
Task.__wakeup)N)N)__name__
__module__ __qualname__r@ r= rR classmethodr __class_getitem__rT rU rV r4 rZ r\ r^ r` rd rL rr
__classcell__r# r# rO r$ r c s
"Wr )r7 c C s t }|| }t|| |S r )r r r r8 )rN r7 r" r6 r# r# r$ r c s
r )r" timeout return_whenc s t | st| r(tdt| j | s4td|tt t
fvrPtd| d u rbt nt
jdtdd t| } tdd | D rt
jd tdd fd
d | 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.
stacklevelc s s | ]}t |V qd S r )r r? r) fr# r# r$ r1 zwait..zThe explicit passing of coroutine objects to asyncio.wait() is deprecated since Python 3.8, and scheduled for removal in Python 3.11.c s h | ]}t | d qS r! r r r! r# r$ r+ r1 zwait..)r isfuturer r? rA typerw
ValueErrorr r r r r warningswarnDeprecationWarningsetany_wait)fsr" r} r~ r# r! r$ r u s$
r c G s | s| d d S r )r' rZ )waiterargsr# r# r$ _release_waiter s r r! c
s |d u rt }ntjdtdd |d u r4| I d H S |dkrt| |d} | rX| S t| |dI d H z
| W S t j
y } zt |W Y d }~n
d }~0 0 | }|
|t|}tt|}t| |d} | | zz|I d H W nT t j
yB | r | Y W | S | | t| |dI d H Y n0 | r`| W | S | | t| |dI d H z| W W | S t j
y } zt |W Y d }~n
d }~0 0 W | n
| 0 d S )Nr r r r r! )r r r r r r r' rY _cancel_and_waitr rh TimeoutError
create_future
call_laterr functoolspartialrq rd remove_done_callback)futr} r" rt r timeout_handlecbr# r# r$ r sZ
"r c
s | d |d ur"||tt| fdd}| D ]}|| q@z2I d H W d urn | D ]}|| qrn&d ur | D ]}|| q0 t t }}| D ]"}| r| | q| | q||fS )Nc sZ d8 dks4t ks4tkrV| sV| d urVd urD sVd d S )Nr r )r r cancelledr[ rd r' rZ r Zcounterr~ r r r# r$ _on_completion s
z _wait.._on_completion)
r r r lenrq rd r r r' add)r r} r~ r" r r r' Zpendingr# r r$ r s0 r c sP | }tt|}| | z | |I d H W | | n | | 0 d S r )r r r r rq rd r )r r" r r r# r# r$ r s
r )r" r} c # s t | st| r(tdt| j d ur@tjdt dd ddl
m } |dd u rft
fdd t| D d fd
d }fd d
fdd}D ]}| qr|d ur؈||ttD ] }| V qd S )Nz#expect an iterable of futures, not r r r r )Queuer! c s h | ]}t | d qS r r r r! r# r$ r+ M r1 z as_completed..c s* D ]} | d q d S r )r
put_nowaitclearr )r r' todor# r$ _on_timeoutP s
z!as_completed.._on_timeoutc s4 sd S | | s0d ur0 d S r )remover rd r )r' r r r# r$ r V s
z$as_completed.._on_completionc s$ I d H } | d u r tj| S r )r r r rY r )r' r# r$
_wait_for_one^ s z#as_completed.._wait_for_one)r r r r? rA r rw r r r Zqueuesr r r2 r rq r ranger )r r" r} r r r r _r# )r r' r" r r r$ r . s*
r c c s
d V d S r r# r# r# r# r$ __sleep0m s r c s~ |d urt jdtdd | dkr0t I d H |S |d u r@t }| }|| tj ||}z|I d H W |
S |
0 d S )Nr r r r ) r r r r r r r r r Z _set_result_unless_cancelledrd )ZdelayrY r" rv hr# r# r$ r y s&
r c C s t | r6|d u rt }|| }|jr2|jd= |S t| rb|d ur^|t| ur^t d| S t
| r|t t
| |dS tdd S )Nr9 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 r? r r2 r r> r r r& r rs Z isawaitabler _wrap_awaitablerA )Zcoro_or_futurer" r6 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
ZdddZ ZS )_GatheringFutureNr! c s t j|d || _d| _d S )Nr! F)r< r= _children_cancel_requested)rM childrenr" rO r# r$ r= s z_GatheringFuture.__init__c C s: | r dS d}| jD ]}|j|drd}q|r6d| _|S ra )r' r rd r )rM rc Zretchildr# r# r$ rd s
z_GatheringFuture.cancel)N)rw rx ry r= rd r| r# r# rO r$ r s r Fr" return_exceptionsc G s( | d urt jdtdd t|| |dS )Nr r r r )r r r _gather)r" r coros_or_futuresr# r# r$ r
s
r
c s |s*| d u rt } | g S fdd}i }g ddd |D ]f}||vrt|| d}| d u rt|} ||urd|_d7 |||< || n|| } | qVt
| dS )Nc s d7 d u s r,| s(| d S sl| rN| }| d S | }|d url| d S krg } D ]J} | rt| jd u rdn| j}n| }|d u r| }| | q|j
r| }| n
| d S )Nr ) r' r r[ ri r\ r rh re rY appendr rZ )r rt Zresultsresr Z nfinishedZnfutsouterr r# r$ _done_callback s<
z _gather.._done_callbackr r! Fr ) r r2 r rZ r r r& r@ rq r r )r" r r r Z
arg_to_futargr r# r r$ r s4
7
r c st |d urt jdtdd t| |d r0S t}| fdd fdd} | S ) Nr r r r! c s\ r | s| d S | r. n*| }|d urJ | n | d S r )r r[ rd r\ rZ rY )innerrt r r# r$ _inner_done_callbackz 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 rq )r r" r r# )r r r r$ r U s
r c s: t stdtj fdd}| S )Nz A coroutine object is requiredc
sl zt t d W nN ttfy0 Y n8 tyf } z rP| W Y d } ~ n
d } ~ 0 0 d S )Nr! )r Z
_chain_futurer rn rm ro Z set_running_or_notify_cancelr\ )rt rN rv r" r# r$ callback s
z*run_coroutine_threadsafe..callback)r r? rA
concurrentr FutureZcall_soon_threadsafe)rN r" r r# r r$ r
s
r
c C s t | d S r )r- r r6 r# r# r$ r s r c C s4 t | }|d ur(td|d|d|t | < d S )NzCannot enter into task z while another task z is being executed.r r r. r" r6 r r# r# r$ r s
r c C s2 t | }||ur(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)C__all__Zconcurrent.futuresr rH r rs itertoolstypesr weakrefr r r r r r r r count__next__rB r r r3 r8 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 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
u
+L,?
&^?$