o
0i~ @ sj d Z ddlZddlZddlZddlZddlZddlZddlZddl Z ddlm
Z
ddl m Z ddl m
Z
ddl mZ ddl mZ dd l mZ dd
l
mZ edjZdBd d ZdBd
dZdd ZG dd dejZeZzddlZW n e y{ Y nw ej ZZ ddddZ ejj Z ejj Z ejj!Z!de!dddZ"dd Z#dd Z$d d Z%d d Z&dd!d"d#Z'ej(d$d% Z)dBd&d'Z*dd(d)d*Z+dd(d+d,Z,ej(d-d. Z-ee-_G d/d0 d0ej.Z/d1d2d3d4Z0d5d6 Z1d7d8 Z2e 3 Z4i Z5d9d: Z6d;d< Z7d=d> Z8d?d@ Z9e6Z:e9Z;e7Ze9Z?e7Z@e8ZAdS )C)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 r t } t| S N)r get_running_loop_current_tasksgetloop r# 4/opt/alt/python310/lib64/python3.10/asyncio/tasks.pyr # s
r c s` d u r t d} ztt}W n ty$ |d7 }|dkr" Y nw nq fdd|D S )Nr Tr i c s& h | ]}t | u r| s|qS r# )r _get_loopdone).0tr! r# r$ = s z all_tasks..)r r list
_all_tasks RuntimeError)r" iZtasksr# r! r$ r * s r c C s8 |d urz| j }W n
ty Y d S w || d S d S r )set_nameAttributeError)tasknamer. r# r# r$ _set_task_nameA s
r2 c s e Zd Z dZddd 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" r1 c s t j|d | jr| jd= t|s d| _td||d u r)dt | _nt || _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" r1 __class__r# r$ r7 ] s
z
Task.__init__c sF | j tjkr | jr | dd}| jr| j|d< | j| t d S )Nz%Task was destroyed but it is pending!)r0 messageZsource_traceback) Z_stater Z_PENDINGr: r8 rD Zcall_exception_handlerr6 __del__)rG r5 rI r# r$ rL t s
z Task.__del__c C s
t | S r )r Z_task_repr_inforG r# r# r$
_repr_info s
zTask._repr_infoc C | j S r )rA rM r# r# r$ get_coro z
Task.get_coroc C rO r )r= rM r# r# r$ get_name rQ z
Task.get_namec C s t || _d S r )r> r= )rG valuer# r# r$ r. s z
Task.set_namec C t d)Nz*Task does not support set_result operationr, )rG resultr# r# r$
set_result zTask.set_resultc C rT )Nz-Task does not support set_exception operationrU )rG exceptionr# r# r$
set_exception rX zTask.set_exception)limitc C s t | |S r )r Z_task_get_stack)rG r[ r# r# r$ get_stack s zTask.get_stack)r[ filec C s t | ||S r )r Z_task_print_stack)rG r[ r] r# r# r$ print_stack s zTask.print_stackc C s@ d| _ | r
dS | jd ur| jj|drdS d| _|| _dS NFmsgT)Z_log_tracebackr& r@ cancelr? _cancel_message)rG ra r# r# r$ rb s
z Task.cancelc
s | rtd| d|| jr t|tjs | }d| _| j}d | _t | j
| zz|d u r8| d }n| |}W nn t
yg } z | jrUd| _t j| jd nt |j W Y d }~n
d }~w tjy } z|| _t W Y d }~nd }~w ttfy } zt | d }~w ty } z t | W Y d }~nd }~ww t|dd }|d ur't|| j
urtd| d|d}| j
j | j || j d n|r|| u rtd
| }| j
j | j || j d nd|_ |j | j!| j d || _| jr| jj| jdrd| _ntd | d |}| j
j | j || j d nj|d u r7| j
j | j | j d nt"#|rStd
| d |}| j
j | j || j d nutd|}| j
j | j || j d W t$| j
| d } d S W t$| j
| d } d S W t$| j
| d } d S W t$| j
| d } d S W t$| j
| d } d S W t$| j
| d } d S W t$| j
| d } d S W t$| j
| d } d S W t$| j
| d } d S W t$| j
| d } d S t$| j
| d } w )Nz_step(): already done: z, Fr` _asyncio_future_blockingzTask z got Future z attached to a different loopr4 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?
isinstanceCancelledError_make_cancelled_errorrA r@ r rD sendthrow
StopIterationr6 rb rc rW rS Z_cancelled_excKeyboardInterrupt
SystemExitrZ
BaseExceptiongetattrr r% r, rE rF rC rd add_done_callback
_Task__wakeupinspectZ isgeneratorr )rG excrH rV Zblockingnew_excrI r# r$ Z__step s
6 1 ( " z Task.__stepc
C sR z| W n ty } z| | W Y d }~d } d S d }~ww | d } d S r )rV rm rF )rG futurerr r# r# r$ Z__wakeup. s
z
Task.__wakeupr )__name__
__module__ __qualname__r: r7 rL classmethodr __class_getitem__rN rP rR r. rW rZ r\ r^ rb rF rp
__classcell__r# r# rI r$ r K s"
"Wr )r1 c C s" t }|| }t|| |S r )r r r r2 )rH r1 r" r0 r# r# r$ r K s
r )timeout return_whenc s t | s t| rtdt| j | s td|tt t
fvr*td| t t
| } tdd | D rCtjdtdd 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: c s s | ]}t |V qd S r )r r9 r' fr# r# r$ x s zwait..zThe explicit passing of coroutine objects to asyncio.wait() is deprecated since Python 3.8, and scheduled for removal in Python 3.11.
stacklevelc h | ]}t | d qS r! r r} r! r# r$ r) ~ zwait..)r isfuturer r9 r; typeru
ValueErrorr r r r r setanywarningswarnDeprecationWarning_wait)fsr{ r| r# r! r$ r ] s r c G s | s | d d S d S r )r& rW )waiterargsr# r# r$ _release_waiter r c
s t }|d u r| I d H S |dkrBt| |d} | r!| S t| |dI d H z| W S tjyA } zt |d }~ww | }|
|t |}t
t |}t| |d} | | zfz|I d H W n& tjy | r}| Y W | S | | t| |dI d H w | r| W | S | | t| |dI d H z
| W W | S tjy } zt |d }~ww | w )Nr r! )r r r r& rV _cancel_and_waitr rf TimeoutError
create_future
call_laterr functoolspartialro rb remove_done_callback)futr{ r" rr r timeout_handlecbr# r# r$ r sZ
r c
s | d |d ur||tt| fdd}| D ]}|| q"zI d H W d ur9 | D ]}|| q;nd urL | D ]}|| qNw t t }}| D ]}| rl| | q`| | q`||fS )Nc sj d8 dkst kstkr-| s/| d ur1d ur" s3d d S d S d S d S d S )Nr r )r r cancelledrY rb r& rW r~ Zcounterr| r r r# r$ _on_completion s z _wait.._on_completion)
r r r lenro rb r r r& add)r r{ r| r" r r~ r& Zpendingr# r r$ r s8 r c sR | }tt|}| | z| |I d H W | | d S | | w r )r r r r ro rb r )r r" r r r# r# r$ r s
r )r{ c # s t | s t| rtdt| j ddlm} | t
fddt | D d fdd}fdd fd
d }D ]}| qHr\|d ur\
||ttD ]}| V qbd S ) Nz#expect an iterable of futures, not r )Queuec r r r r} r! r# r$ r) % r z as_completed..c s* D ] } | d q d S r )r
put_nowaitclearr )r r& todor# r$ _on_timeout( s
z!as_completed.._on_timeoutc s< sd S | | sd ur d S d S d S r )remover rb r )r& r r r# r$ r . s
z$as_completed.._on_completionc s& I d H } | d u rtj| S r )r r r rV r )r& r# r$
_wait_for_one6 s
z#as_completed.._wait_for_one)r r r r9 r; r ru Zqueuesr r _get_event_loopr ro r ranger )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$ __sleep0E s
r c s^ | dkrt I d H |S t }| }|| tj||}z
|I d H W | S | w )Nr )r r r r r r Z _set_result_unless_cancelledrb )ZdelayrV r" rt hr# r# r$ r Q s
r r! c C s t | |dS Nr! )_ensure_future)coro_or_futurer" r# r# r$ r b s r c C s t | r|d ur|t | urtd| S d}t| s-t| r)t| } d}nt d|d u r7t
j dd}z| | W S t
yK |sJ| w )NzRThe future belongs to a different loop than the one specified as the loop argumentFTz:An asyncio.Future, a coroutine or an awaitable is required r )r r r% r r r9 rq Z isawaitable_wrap_awaitabler; r r r r, close)r r" Zcalled_wrap_awaitabler# r# r$ r j s&
r c c s | E d H S r ) __await__)Z awaitabler# r# r$ r s r c s( e Zd Z fddZdddZ ZS )_GatheringFuturec s t j|d || _d| _d S )Nr! F)r6 r7 _children_cancel_requested)rG childrenr" rI r# r$ r7 s
z_GatheringFuture.__init__Nc C s: | rdS d}| jD ]
}|j|drd}q |rd| _|S r_ )r& r rb r )rG ra Zretchildr# r# r$ rb s
z_GatheringFuture.cancelr )ru rv rw r7 rb rz r# r# rI r$ r s r F)return_exceptionsc s |st }| g S fdd}i }g ddd }d |D ]3}||vrTt||d}|d u r?t|}||urFd|_d7 |||< || n|| } | q*t
|dS )Nc s d7 d u s r| s| d S s6| r'| }| d S | }|d ur6| d S kryg } D ]%} | rRt| jd u rMdn| j}n | }|d u r^| }| | q>j
rr| }| d S | d S d S )Nr ) r& r rY rg rZ r rf rc rV appendr rW )r rr Zresultsresr Z nfinishedZnfutsouterr r# r$ _done_callback s>
z gather.._done_callbackr r! Fr ) r r r rW r r r% r: ro r r )r Zcoros_or_futuresr" r Z
arg_to_futargr r# r r$ r
s6
7
r
c sZ t | r S t}| fdd fdd} | S )Nc s` r| s | d S | r d S | }|d ur' | d S | d S r )r rY rb rZ rW rV )innerrr r r# r$ _inner_done_callbackO s z$shield.._inner_done_callbackc s s d S d S r )r& r r )r r r# r$ _outer_done_callback` r z$shield.._outer_done_callback)r r& r r% r ro )r r" r r# )r r r r$ r ' s
r c s< t s
tdtj fdd}| S )Nz A coroutine object is requiredc
sZ z
t t d W d S ttfy ty, } z
r'| d } ~ ww r )r Z
_chain_futurer rl rk rm Z set_running_or_notify_cancelrZ )rr rH rt r" r# r$ callbackr s
z*run_coroutine_threadsafe..callback)r r9 r;
concurrentr FutureZcall_soon_threadsafe)rH r" r r# r r$ r
i s
r
c C t | d S r )r+ r r0 r# r# r$ r r c C s4 t | }|d urtd|d|d|t | < d S )NzCannot enter into task z while another task z is being executed.r r r, r" r0 r r# r# r$ r s
r c C s2 t | }||urtd|d|dt | = d S )Nz
Leaving task z! does not match the current task .r r r# r# r$ r s
r c C r r )r+ discardr r# r# r$ r r r )r r r r r+ r r )B__all__Zconcurrent.futuresr rB r rq itertoolstypesr weakrefr r r r r r r r count__next__r< r r r2 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
&G,9
{B$