֦i d Z ddlZddlZddlZddlZddlZddlZddlZddl Z ddlm
Z
ddl m Z ddl m
Z
ddl mZ ddl mZ dd l mZ dd
l mZ ej$ d j& Zd.d Zd.d Zd
Z G d dej0 ZeZ ddlZej2 xZZ ddddZ ej j> Z ej j@ Z ej jB Z!de!ddZ"d Z#d Z$d Z%d Z&dddZ'ejP d Z)d.dZ*dd d Z+ G d d ejX Z-d d!d"Z.d# Z/d$ Z0d% Z1 e1e Z2 e jf Z4 e5 Z6i Z7d& Z8d' Z9d( Z:d) Z;d* ZeZ?e8Z@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)tasknamer9 s r' _set_task_namer@ F sM }}H
TN 8 MM 9 )Q
8 8s %AAc e Zd Z dZddddd fd
Z fdZ ee Zd Z d Z
d Z d
Z d Z
d Zd
ZdddZddddZddZd Zd Zd ZddZ fdZd Z xZS )r 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 rC )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? rC rD __class__s r' rI z
Task.__init__o s
d# ! !&&r*%%d+ ).D % | messagesource_traceback) _stater _PENDINGrL rJ rX call_exception_handlerrH __del__)r] rC r_ s r' rf z Task.__del__ sb ;;'** *t/H/H BG %%.2.D.D*+ JJ - -g 6
r( c , t j | S r! )r
_task_reprr] s r' __repr__z
Task.__repr__ s $$T**r( c | j S r! )rT ri s r' get_coroz
Task.get_coro zzr( c | j S r! )rW ri s r' get_contextzTask.get_context s }} r( c | j S r! )rO ri s r' get_namez
Task.get_name rm r( c $ t | | _ y r! )rP rO )r] values r' r9 z
Task.set_name s
Z
r( c t d )Nz*Task does not support set_result operationr- )r] results r'
set_resultzTask.set_result s GHHr( c t d )Nz-Task does not support set_exception operationru )r] exceptions r'
set_exceptionzTask.set_exception s JKKr( )limitc 0 t j | | S r! )r _task_get_stack)r] r{ s r' get_stackzTask.get_stack s ( ))$66r( )r{ filec 2 t j | || S r! )r _task_print_stack)r] r{ r s r' print_stackzTask.print_stack s ++D%>>r( c d| _ | j ry| xj dz
c_ | j | j j | ryd| _ || _ y)NFr msgT)_log_tracebackr1 rQ rS cancelrR _cancel_message)r] r s r' r z Task.cancel sk * $ 99; ##q(# ' &&3&/ " r( c | j S r! rQ ri s r'
cancellingzTask.cancelling s
***r( c d | j dkD r| xj dz c_ | j S )Nr r r ri s r' uncancelz
Task.uncancel s4 & & * ' '1 , '***r( c p t | j | } t | | j j | j
d t
| t | j | }| j r
d | _ d } y t | y # t
| w xY w# | j r
d | _ d } w t | w xY w# t | j | }| j r
d | _ d } w t | w # | j r
d | _ d } w t | w xY wxY wr! )
_swap_current_taskrX _register_eager_taskrW run!_Task__step_run_and_handle_result_unregister_eager_taskr1 rT 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
#'C
D5D %&D5 'D22D5c | 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) r1 r InvalidStateErrorrR
isinstanceCancelledError_make_cancelled_errorrS r rX 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 looprG 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: ) rT sendthrowgetattrr r0 rX r- r[ r\ rW r add_done_callback
_Task__wakeuprS rR r r inspect isgenerator
StopIterationrH 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 wr! )rv r\ r )r] futurer s r' __wakeupz
Task.__wakeup sH MMO
KKM KK s % A AA r! )__name__
__module__ __qualname__rL rI rf classmethodr __class_getitem__rj rl ro rq r9 rw rz r~ r r r r rZ r\ r r
__classcell__r_ s @r' r r S s +. %)d " !> $L1+ IL "& 7. $(d ?(T+
+)&"IVr( r r? rC c t j }||j | }n|j | | }t || |S )NrG )r r" r r@ )r^ r? rC r& r> s r' r r sP " " $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)Nz expect a list of futures, not z Set of Tasks/Futures is empty.zInvalid return_when value: c 3 F K | ] }t j | y wr! )r rK ).0fs r' zwait.. s
1b: ! !! $bs !z6Passing coroutines is forbidden, use tasks explicitly.)r isfuturer rK rM 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 r! )r1 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 { cd d d d { S 7 N# t j
$ r }t |d }~ww xY w7 C7 ;7 -# 1 d { 7 sw Y y xY wwNr ) r
r1 rv _cancel_and_waitr r TimeoutErrorr r )futr r s r' r r s D w!|C 88:::< s### (::< ((y )((
$ (( ( C ' ( ) )(((s ACBC
B C3B74C7B==B9>B= C
B;CB4(B//B44C9B=;C=CC C Cc 2
K |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)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 r1 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 r1 add) r r r r& r r r1 pendingr r r s ` @@@r' r r
s
!FN /6J"gG
( N+ 3 % ! ! #A
" "> 2 E35'D
668 HHQK KKN
= % ! ! #A
" "> 2 s1 AD C( $C&%C( )A=D&C( (,DDc 4 K t j }|j }t j t
| }| j
| | j | d { | j | y 7 # | j | w xY wwr! ) r r" r functoolspartialr r r r )r r& r cbs r' r r 6 s~ F " " $D
!F ?F 3B" %
$ $s0 ABB )B *B .B B BB)r c #
K t j | st j | r!t dt | j ddlm} | t j }t | D ch c] }t || c} d
fd}
fd fd} D ] }|j r||j ||
t t! D ] } |
y c c}w w)Nz#expect an iterable of futures, not r )Queuer% c x D ]$ } | j j d & j y r! )r
put_nowaitclear)r r r1 todos r' _on_timeoutz!as_completed.._on_timeoutd s2 A
" "> 2 OOD !
r( c | sy j | j | sj y y y r! )remover r )r r1 r r s r' r z$as_completed.._on_completionj s; A 2 ! ! # 3tr( c K j d { } | t j | j S 7 &wr! )r$ r r rv )r r1 s r'
_wait_for_onez#as_completed.._wait_for_oner s7 ((* 9 )) )xxz
s A >'A )r r r rK rM r r queuesr r get_event_loopr r
r r ranger ) r r r r& r r r _r r1 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;DC>A.Dc # K d y wr! r r( r' __sleep0r s
s c 2 K | dk rt d { |S t j }|j }|j | t
j || } | d { |j S 7 g7 # |j w xY wwr )r r r" r r r _set_result_unless_cancelledr )delayrv r& r hs r' r
r
s C zj
" " $D
!F << (A|
s: BA>A B$B )B *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)NzRThe future belongs to a different loop than the one specified as the loop argumentTc " K | d { S 7 wr! r ) awaitables r' _wrap_awaitablez&ensure_future.._wrap_awaitable s &&s
Fz:An asyncio.Future, a coroutine or an awaitable is required)r r r0 r r rK r isawaitablerM 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 r% Fr ) r r r rw r
r r0 rL r1 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 )Nc 0 j r!| j s| j y | j rj y | j }|j | y j | j y r! )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 r! )r1 r )r r r s r' _outer_done_callbackz$shield.._outer_done_callback s zz| & &'; < r( )r
r1 r r0 r r )r r& r r r r s @@@r' r r a sp @
# E zz| U #D E1"=
01 01 Lr( c t j s t d t j j fd}j
| S )Nz 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 rK rM
concurrentr Futurecall_soon_threadsafe)r^ r& r" r s `` @r' r r sR ! !$ '899
&
&
(F h' Mr( c d d d fd
}|S )Nr c || ||d S )NTrB r )r& r^ r? rC 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 0 t j | y r! )r, r r> s r' r r s E r( c 0 t j | y r! )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 r! )r# r$ )r& r> r s r' r r s9 ""4(I | 4 $t r( c 0 t j | y r! )r, discardr+ s r' r r s 1 T"r( c 0 t j | y r! )r+ r4 r+ s r' r r
s @ r( ) r r r r r r r r, r+ r# r r! )O__all__concurrent.futuresr# rU r r r. typesr; weakrefr r r r r r r r count__next__rN 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$
F4 G 4F=<F= GG |