֦i d Z dZddlZddlZddlZddlZddlZddlZddl Z ddl
Z
ddlm Z ddl m
Z
ddl mZ ddl mZ dd l mZ dd
l mZ dd l mZ ej& d j( Zd/d Zd/d
Zd Z G d dej2 ZeZ ddl Z e j4 xZZ ddddZ ej" j@ Z ej" jB Z!ej" jD Z"de"ddZ#d Z$d Z%d Z&d Z'dddZ(ejR d Z*d/d Z+dd d Z, G d d ejZ Z.d!d"d#Z/d$ Z0d% Z1d& Z2 e2e Z3 e
jh Z5 e6 Z7i Z8d' Z9d( Z:d) Z;d* Zd- Z?eZ@e9ZAe:ZBe>ZCe?ZDe;ZEeZ>m?Z?m;Z;mZKe?ZLe;ZMeD! #t+AFFH
D > >
FADy >s B 0B"B B c | | j } || y y # t $ r t j dt d Y y w xY w)Nz~Task.set_name() was added in Python 3.8, the method support will be mandatory for third-party task implementations since 3.13. )
stacklevel)set_nameAttributeErrorwarningswarnDeprecationWarning)tasknamer8 s r& _set_task_namer? F sM }}H
TN 8 MM 9 )Q
8 8s %AAc e Zd ZdZdZddddd fd
Z fdZ ee Z d Z
d Z d
Z d Z
d Zd
Zd ZdddZddddZddZd Zd Zd ZddZ fdZd Z xZS )r z A coroutine wrapped in a Future.TNFr% r> context eager_startc F t | | | j r
| j d= t j | sd| _ t
d| |dt | _ nt | | _ d| _
d| _ d | _ || _
|t j | _ n|| _ |r+| j" j% r| j' y | j" j) | j* | j t- | y )Nr$ Fz a coroutine was expected, got zTask-r rB )super__init___source_tracebackr iscoroutine_log_destroy_pending TypeError_task_name_counter_namestr_num_cancels_requested _must_cancel _fut_waiter_coro contextvars copy_context_context_loop
is_running_Task__eager_start call_soon _Task__stepr )selfcoror% r> rB rC __class__s r& rH z
Task.__init__o s
d# ! !&&r*%%d+ ).D % | >r' c d| _ | j ry| xj dz
c_ | j | j j | ryd| _ || _ y)a Request that this task cancel itself.
This arranges for a CancelledError to be thrown into the
wrapped coroutine on the next cycle through the event loop.
The coroutine then has a chance to clean up or even deny
the request using try/except/finally.
Unlike Future.cancel, this does not guarantee that the
task will be cancelled: the exception might be caught and
acted upon, delaying cancellation of the task or preventing
cancellation completely. The task may also return a value or
raise a different exception.
Immediately after this method is called, Task.cancelled() will
not return True (unless the task was already cancelled). A
task will be marked as cancelled when the wrapped coroutine
terminates with a CancelledError exception (even if cancel()
was not called).
This also increases the task's count of cancellation requests.
Fr msgT)_log_tracebackr0 rP rR cancelrQ _cancel_message)r\ r s r& r z Task.cancel sf , $ 99; ##q(# ' &&3&/ " r' c | j S )zReturn the count of the task's cancellation requests.
This count is incremented when .cancel() is called
and may be decremented using .uncancel().
rP ri s r&
cancellingzTask.cancelling s ***r' c b | j dkD r| xj dz c_ | j S )zDecrement the task's count of cancellation requests.
This should be called by the party that called `cancel()` on the task
beforehand.
Returns the remaining number of cancellation requests.
r r r ri s r& uncancelz
Task.uncancel s/ & & * ' '1 , '***r' c t | j | } t | | j j | j
d t
| t | j | }|| u sJ | j r
d | _ d } y t | y # t
| w xY w# | j r
d | _ d } w t | w xY w# t | j | }|| u sJ | j r
d | _ d } w t | w # | j r
d | _ d } w t | w xY wxY wrg )
_swap_current_taskrW _register_eager_taskrV run!_Task__step_run_and_handle_result_unregister_eager_taskr0 rS r )r\ prev_taskcurtasks r&
__eager_startzTask.__eager_start s &tzz48 ) &
-
!!$"C"CTJ&t,
),TZZC $&99;!%DJ D"4( 't, 99;!%DJ D"4(
),TZZC $&99;!%DJ D"4( 99;!%DJ D"4(sF C &B
C B*
B''C *'CE D3&E'E Ec | j rt j d| d| | j r1t |t j
s| j
}d| _ d | _ t | j | | j | t | j | d } y # t | j | d } w xY w)Nz_step(): already done: z, F) r0 r InvalidStateErrorrQ
isinstanceCancelledError_make_cancelled_errorrR r rW r r )r\ excs r& __stepz Task.__step# s 99; ..)$C7;=
= c:#<#<=002 %D DJJ% - -c 2
D )D
D )Ds B1 1C c | j } ||j d }n|j | }t |dd }|lt j
| | j urGt d| d|d }| j j | j || j d } y |r|| u rCt d| }| j j | j || j d } y d|_ |j | j | j || _ | j rN| j j! | j" r'd| _ d } y t d | d
| }| j j | j || j d } y |4| j j | j | j d } y t% j& | rFt d | d
| }| j j | j || j d } y t d | }| j j | j || j d } y d } y # t( $ rS}| j r"d| _ t* | A | j" nt* | Y |j. Y d }~d } y d }~wt0 j2 $ r!}|| _ t* | A Y d }~d } y d }~wt6 t8 f$ r}t* | u | d }~wt< $ r}t* | u | Y d }~d } y d }~ww xY w# d } w xY w)
N_asyncio_future_blockingzTask z got Future z attached to a different looprF z Task cannot await on itself: Fr 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: ) rS sendthrowgetattrr r/ rW r, rZ r[ rV r add_done_callback
_Task__wakeuprR rQ r r inspect isgenerator
StopIterationrG rw rs r r _cancelled_excKeyboardInterrupt
SystemExitrz
BaseException)r\ r r] rv blockingnew_excr^ s r& __step_run_and_handle_resultz!Task.__step_run_and_handle_result4 se zzG { 4 C $ v'A4HH#$$V,DJJ> * x|!*$AC DG JJ(( Wdmm ) EP DM ~". ;D8 D#F
,, KK$-- - ID D? ;@7 00 MM4== 1 B+1( ,, # / / 6 6(,(<(< 7 >49 10 D- +##'(&
< =G JJ(( Wdmm ) E& D!
$$T[[$--$H D $$V,&))-vjBC
$$KK$-- % A D ')=fZ'HI
$$KK$-- % A D4DA . $)!4#7#78"399-t Ds (( "%D GN l Dk ":. G !# & ' G !# & &b De 'd Ds %J A5M ,AM 5A0M )AM 03M &AM A M MAK M M5L
M M#L33 M?MM MM M!c |j | j d } y # t $ r }| j | Y d }~d } y d }~ww xY wrg )rv r[ r )r\ futurer s r& __wakeupz
Task.__wakeup sH MMO
KKM KK s % A AA rg )__name__
__module__ __qualname____doc__rK rH re classmethodr __class_getitem__rj rl ro rq r8 rw rz r~ r r r r rY r[ r r
__classcell__r^ s @r& r r S s +. %)d " !> $L1+ IL "& 7. $(d ?(T+
+)&"IVr' r r> rB c t j }||j | }n|j | | }t || |S )z]Schedule the execution of a coroutine object in a spawn task.
Return a Task object.
rF )r r! r r? )r] r> rB r% r= s r& r r sK
" " $D % g 64 Kr' )timeout return_whenc K t j | st j | r!t dt | j | s t d |t t t fvrt d| t | } t d | D r t d t j }t | ||| d{ S 7 w)a} Wait for the Futures or Tasks given by fs to complete.
The fs iterable must not be empty.
Coroutines will be wrapped in Tasks.
Returns two sets of Future: (done, pending).
Usage:
done, pending = await asyncio.wait(fs)
Note: This does not raise TimeoutError! Futures that aren't done
when the timeout occurs are returned in the second set.
z expect a list of futures, not z Set of Tasks/Futures is empty.zInvalid return_when value: c 3 F K | ] }t j | y wrg )r rJ ).0fs r& zwait.. s
1b: ! !! $bs !z6Passing coroutines is forbidden, use tasks explicitly.N)r isfuturer rJ rL typer
ValueErrorr r r setanyr r! _wait)fsr r r% s r& r r s z 5 5b 98b9J9J8KLMM
9::?O]KK6{mDEE RB
1b
11PQQ " " $Dr7K6 66 6s CC C
Cc H | j s| j d y y rg )r0 rw )waiterargss r& _release_waiterr s ;;=$ r' c K |T|dk rOt | } | j r| j S t | d{ | j S t j | 4 d{ | d{ cddd d{ S 7 N# t j
$ r }t |d}~ww xY w7 C7 ;7 -# 1 d{ 7 sw Y yxY ww)a Wait for the single Future or coroutine to complete, with timeout.
Coroutine will be wrapped in Task.
Returns result of the Future or coroutine. When a timeout occurs,
it cancels the task and raises TimeoutError. To avoid the task
cancellation, wrap it in shield().
If the wait is cancelled, the task is also cancelled.
If the task suppresses the cancellation and returns a value instead,
that value is returned.
This function is a coroutine.
Nr ) r
r0 rv _cancel_and_waitr r TimeoutErrorr r )futr r s r& r r s F w!|C 88:::< s### (::< ((y )((
$ (( ( C ' ( ) )(((s ACBC B C2B63C6B<<B8=B< C B:
CB3'B..B33C8B<:C<CC C
Cc B
K | sJ d |j
d ||j |t
t |
fd}| D ] }|j |
d{ j | D ] }|j
| t t }}| D ]5 }|j r|j | %|j | 7 ||fS 7 # j | D ] }|j
| w xY ww)zVInternal helper for wait().
The fs argument must be a collection of Futures.
zSet of Futures is empty.Nc dz dk s2t k( s)t k( rW| j sF| j 5j j sj
d y y y y y )Nr r )r r cancelledry r r0 rw )r counterr timeout_handler s r& _on_completionz _wait.._on_completion st 1 qL ? * ? *AKKM01
0I ) %%';;=!!$' ! 1J 5B *r' )
create_future
call_laterr lenr r remove_done_callbackr r0 add) r r r r% r r r0 pendingr r r s ` @@@r& r r
s
)))2
!FN /6J"gG
( N+ 3 % ! ! #A
" "> 2 E35'D
668 HHQK KKN
= % ! ! #A
" "> 2 s1 A D 'C0 ,C.-C0 1A=D .C0 0,D D c 2 K t j }|j }t j t
| }| j
| | j | d{ | j | y7 # | j | w xY ww)z._on_timeoutd s2 A
" "> 2 OOD !
r' c | sy j | j | sj y y y rg )remover r )r r0 r r s r& r z$as_completed.._on_completionj s; A 2 ! ! # 3tr' c K j d { } | t j | j S 7 &wrg )r# r r rv )r r0 s r&
_wait_for_onez#as_completed.._wait_for_oner s7 ((* 9 )) )xxz
s A >'A )r r r rJ rL r r queuesr r get_event_loopr r
r r ranger ) r r r r% r r r _r r0 r r s @@@@r& r r H s $ z 5 5b 9=d2h>O>O=PQRR 7D "D14R 9AM!$ ' 9DN $ N+ # +>
3t9
o 9 :s A:D C=A.Dc # K d yw)zSkip one event loop run cycle.
This is a private helper for 'asyncio.sleep()', used
when the 'delay' is set to 0. It uses a bare 'yield'
expression (which Task.__step knows how to handle)
instead of creating a Future object.
N r r' r& __sleep0r s
s c 0 K | dk rt d{ |S t j }|j }|j | t
j || } | d{ |j S 7 g7 # |j w xY ww)z9Coroutine that completes after a given time (in seconds).r N)r r r! r r r _set_result_unless_cancelledr )delayrv r% r hs r& r
r
s zj
" " $D
!F << (A|
s: BA=A B#B (A?)B ,B?B BBr$ c t j | r&|"|t j | ur t d | S d}t j
| s.t
j | rd } || } d}n t d |t j } |j | S # t $ r |r| j w xY w)zmWrap a coroutine or an awaitable in a future.
If the argument is a Future, it is returned directly.
zRThe future belongs to a different loop than the one specified as the loop argumentTc " K | d { S 7 wrg r ) awaitables r& _wrap_awaitablez&ensure_future.._wrap_awaitable s &&s
Fz:An asyncio.Future, a coroutine or an awaitable is required)r r r/ r r rJ r isawaitablerL r r r r, close)coro_or_futurer% should_closer s r& r
r
s
' G,=,=n,M M E F
F L ! !. 1 ~ .
' -^ ._done_callback s, Q =EJJL==?
}} //1##C(mmo?'', G ==? %33 ! 1 1 9++-C --/C{ !jjls# " && //1##C( ); r' r Nr$ Fr ) r r r rw r
r r/ rK r0 r
r r ) r coros_or_futuresr% r
arg_to_fut done_futsargr r r r r s ` @@@@r& r r s < $$&""$
5* 5*n JH
EII D E j $/C| ((-#~
,1( QJE !JsOxxz %%%n5 S/C / 2
XD 1E s Lr' c t | j rS t j }|j fdfd}j j | S )a Wait for a future, shielding it from cancellation.
The statement
task = asyncio.create_task(something())
res = await shield(task)
is exactly equivalent to the statement
res = await something()
*except* that if the coroutine containing it is cancelled, the
task running in something() is not cancelled. From the POV of
something(), the cancellation did not happen. But its caller is
still cancelled, so the yield-from expression still raises
CancelledError. Note: If something() is cancelled by other means
this will still cancel shield().
If you want to completely ignore cancellation (not recommended)
you can combine shield() with a try/except clause, as follows:
task = asyncio.create_task(something())
try:
res = await shield(task)
except CancelledError:
res = None
Save a reference to tasks passed to this function, to avoid
a task disappearing mid-execution. The event loop only keeps
weak references to tasks. A task that isn't referenced elsewhere
may get garbage collected at any time, even before it's done.
c 0 j r!| j s| j y | j rj y | j }|j | y j | j y rg )r ry r rz rw rv )innerr r s r& _inner_done_callbackz$shield.._inner_done_callback sj ?? ??$! ?? LLN//#C##C( 0r' c J j sj y y rg )r0 r )r r r s r& _outer_done_callbackz$shield.._outer_done_callback s zz| & &'; < r' )r
r0 r r/ r r )r r% r r r r s @@@r& r r a sk B
# E zz| U #D E1"=
01 01 Lr' c t j s t d t j j fd}j
| S )zsSubmit a coroutine object to a given event loop.
Return a concurrent.futures.Future to access the result.
z A coroutine object is requiredc t j t y # t t f$ r t
$ r'} j
rj | d } ~ ww xY w)Nr$ )r
_chain_futurer
r r r set_running_or_notify_cancelrz )r r] r r% s r& callbackz*run_coroutine_threadsafe..callback s] ! !-4"@& I -. 224$$S) s !% A$"A A$)r rJ rL
concurrentr Futurecall_soon_threadsafe)r] r% r" r s `` @r& r r sM
! !$ '899
&
&
(F h' Mr' c ddd fd
}|S )a= Create a function suitable for use as a task factory on an event-loop.
Example usage:
loop.set_task_factory(
asyncio.create_eager_task_factory(my_task_constructor))
Now, tasks created will be started immediately (rather than being first
scheduled to an event loop). The constructor argument can be any callable
that returns a Task-compatible object and has a signature compatible
with `Task.__init__`; it must have the `eager_start` keyword argument.
Most applications will use `Task` for `custom_task_constructor` and in
this case there's no need to call `create_eager_task_factory()`
directly. Instead the global `eager_task_factory` instance can be
used. E.g. `loop.set_task_factory(asyncio.eager_task_factory)`.
Nr c || ||d S )NTrA r )r% r] r> rB custom_task_constructors r& factoryz*create_eager_task_factory..factory s & t$TK Kr' r )r( r) s ` r& r r s & %)$ K Nr' c . t j | y)z;Register an asyncio Task scheduled to run on an event loop.N)r+ r r= s r& r r s r' c . t j | y)z6Register an asyncio Task about to be eagerly executed.N)r* r r+ s r& r r s Tr' c h t j | }|t d|d|d |t | < y )NzCannot enter into task z while another task z is being executed.r" r# r, r% r= r s r& r r sL !%%d+L 4TH =##/"22E G H H N4r' c h t j | }||urt d|d|d t | = y )Nz
Leaving task z! does not match the current task .r. r/ s r& r r sJ !%%d+L4 ]4( 3//;.>a A B Bt r' c X t j | }| t | = |S |t | < |S rg )r" r# )r% r= r s r& r r s9 ""4(I | 4 $t r' c . t j | y)z'Unregister a completed, scheduled Task.N)r+ discardr+ s r& r r s T"r' c . t j | y)z6Unregister a task which finished its first eager step.N)r* r4 r+ s r& r r
s r' ) r r r r r r r r+ r* r" r rg )Pr __all__concurrent.futuresr# rT r r r- typesr: weakrefr r r r r r r r count__next__rM r r r? _PyFuturer _PyTask_asyncio_CTask ImportErrorr r r r r r r r r r coroutiner r
r
r$ r r r r r r WeakSetr+ r r* r" r r r r r r r _py_current_task_py_register_task_py_register_eager_task_py_unregister_task_py_unregister_eager_task_py_enter_task_py_leave_task_py_swap_current_task_c_current_task_c_register_task_c_register_eager_task_c_unregister_task_c_unregister_eager_task
_c_enter_task
_c_leave_task_c_swap_current_taskr r' r& rS sM 6
%Y__Q'00 $>6
z7 zz "
MM!D6 #D
$ $$44 $$44""00
#
7@
0d)X%$ !% 6 r
" +/ @w~~ : 16 CL?D.4 /t4 #7??$ u
#
" . & 2 * .( ( ( ( #O% 1) 5 M M-i T s$ F5
G 5F>=F>G G |