cj P d Z dZddlZddlmZ ddlmZ ddlmZ ddlmZ dd d
Z y) zFSupport for running coroutines in parallel with staggered start times.)staggered_race N )events)
exceptions)locks)tasks)loopc
K xs t j t | ddg g t d
fd
d
f d d} t j
}j
|d } j | |j
|j d} r j
d{ d
r r t d || f ~S 7 ## t j $ r,}|} D ] } |j |j Y d}~[d}~ww xY w# ~w xY ww)a Run coroutines with staggered start times and take the first to finish.
This method takes an iterable of coroutine functions. The first one is
started immediately. From then on, whenever the immediately preceding one
fails (raises an exception), or when *delay* seconds has passed, the next
coroutine is started. This continues until one of the coroutines complete
successfully, in which case all others are cancelled, or until all
coroutines fail.
The coroutines provided should be well-behaved in the following way:
* They should only ``return`` if completed successfully.
* They should always raise an exception if they did not complete
successfully. In particular, if they handle cancellation, they should
probably reraise, like this::
try:
# do work
except asyncio.CancelledError:
# undo partially completed work
raise
Args:
coro_fns: an iterable of coroutine functions, i.e. callables that
return a coroutine object when called. Use ``functools.partial`` or
lambdas to pass arguments.
delay: amount of time, in seconds, between starting coroutines. If
``None``, the coroutines will run sequentially.
loop: the event loop to use.
Returns:
tuple *(winner_result, winner_index, exceptions)* where
- *winner_result*: the result of the winning coroutine, or ``None``
if no coroutines won.
- *winner_index*: the index of the winning coroutine in
``coro_fns``, or ``None`` if no coroutines won. If the winning
coroutine may return None on success, *winner_index* can be used
to definitively determine whether any coroutine won.
- *exceptions*: list of exceptions returned by the coroutines.
``len(exceptions)`` is equal to the number of coroutines actually
started, and the order is the same as in ``coro_fns``. The winning
coroutine's entry is ``None``.
Nc j | #j ssj d | j ry | j }|y j | y )N)discarddone
set_result cancelled exceptionappend)taskexcon_completed_fut
running_tasksunhandled_exceptionss */usr/lib64/python3.12/asyncio/staggered.py task_donez!staggered_race..task_doneJ sc d# ($))+! ' ' - >> nn ; ##C( c K | j d { |Xt j t j 5 t j |j d { d d d t \ }}t j }t j }j || }j | |j |j
j d t!
|dz k( sJ | d { }J ||t j" }D ] } | |us| j% y 7 N7 # 1 sw Y xY w# t $ r Y y w xY w7 ^# t& t( f$ r t* $ r }
|
|< |j Y d }
~
y d }
~
ww xY ww)Nr )wait
contextlibsuppressexceptions_mod TimeoutErrorr wait_fornext
StopIterationr Event create_taskaddadd_done_callbacksetr len current_taskcancel
SystemExitKeyboardInterrupt
BaseException) ok_to_startprevious_failed
this_indexcoro_fn this_failednext_ok_to_start next_taskresultr) tedelay
enum_coro_fnsr r run_one_coror r winner_index
winner_results r r: z$staggered_race..run_one_coro[ s &$$^%@%@A
nn_%9%9%;UCCC B "&}"5 J kkm ;;=$$\2BK%PQ )$##I. $ :*q.000 "9_F ' ''%L "M !--d3L"L(HHJ #a ! D BA % -. %&Jz " OO s GE%)G(E*)E(*E*.G7E6 BG
F &F'F +&GG(E**E3/G6 F?GFGF F> F94G9F>>Gzstaggered race failed)returnN)r get_running_loop enumerater' r r# r$ r% r&
create_futurer CancelledErrorr* argsExceptionGroup)coro_fnsr8 r propagate_cancellation_errorr.
first_taskexr r9 r r r: r r r r; r< s `` @@@@@@@@@r r r
s` h ,6**,D h'MML JEM)"6 6 p $( Kkkm %%l;&EF
*%$$Y/'+$ # 1 1 3
*&&&
$ . !!8:NO O ' 3. . lJ6 46J '!00
*/1, )D DKK) *
*
46Jsa AE A2D= C; C9C; D= D= 5E9C; ;D:"D50D= 5D::D= =EE)
__doc____all__r r r r r r r r r rL s( L
* 37 aKr