:nSE J d gZ ddlmZ ddlmZ ddlmZ G d d ZdS ) TaskGroup )events)
exceptions)tasksc T e Zd ZdZd Zd Zd Zd ZddddZd e d
e
fd Z d Z d
Z
dS )r a9 Asynchronous context manager for managing groups of tasks.
Example use:
async with asyncio.TaskGroup() as group:
task1 = group.create_task(some_coroutine(...))
task2 = group.create_task(other_coroutine(...))
print("Both tasks have completed now.")
All tasks are awaited when the context manager exits.
Any exceptions other than `asyncio.CancelledError` raised within
a task will cancel all remaining tasks and wait for them to exit.
The exceptions are then combined and raised as an `ExceptionGroup`.
c d| _ d| _ d| _ d | _ d | _ d| _ t
| _ g | _ d | _ d | _
d S )NF) _entered_exiting _aborting_loop _parent_task_parent_cancel_requestedset_tasks_errors _base_error_on_completed_futselfs ?/opt/alt/python-internal/lib64/python3.11/asyncio/taskgroups.py__init__zTaskGroup.__init__ sV
(-%ee !% c t dg}| j r*| dt | j | j r*| dt | j | j r| d n | j r| d d | }d| dS ) N ztasks=zerrors=
cancellingentered z
)r appendlenr r r join)r infoinfo_strs r __repr__zTaskGroup.__repr__( s t ; 5 KK3T[!1!133 4 4 4 < 7 KK5#dl"3"355 6 6 6 > # KK % % % %
] # KK " " "88D>>'H''''r c K | j rt d| d | j t j | _ t j | j | _ | j t d| d d| _ | S )N
TaskGroup z has already been enteredz! cannot determine the parent taskT)r RuntimeErrorr r get_running_loopr current_taskr
r s r
__aenter__zTaskGroup.__aenter__6 s = @ >T>>>@ @
@ : 022DJ ! .tz : : $ FTFFFH H
H
r c K d| _ |#| | r| j || _ |t j u r|nd }| j r | j dk rd }|| j s| | j
r{| j | j
| _ | j d {V n9# t j $ r'}| j s|}| Y d }~nd }~ww xY wd | _ | j
{| j
rJ | j | j |r | j s||(|t j ur| j | | j r% t! d| j }|d # d | _ w xY wd S )NT z unhandled errors in a TaskGroup)r
_is_base_errorr r CancelledErrorr r
uncancelr _abortr r r
create_futurer r BaseExceptionGroup)r etexctb propagate_cancellation_errorexmes r __aexit__zTaskGroup.__aexit__D s
O##C((
( "D 222CC % ( 4 ))++q00 04,
>>
k *%-)-)A)A)C)C&
",,,,,,,,, ,
"
"
"~
" 460KKMMM
" &*D "' k ** ; '" " ( / /. .
>b
(A A A L $ $ $ < $
$'(I4<XXd" # #### $ $s$ 1
B? ?C5 C00C5E2 2 E;N)namecontextc | j st d| d | j r| j st d| d | j rt d| d || j | }n | j || }t j || | | j
| j | |S )zbCreate a new task in this group and return it.
Similar to `asyncio.create_task`.
r& z has not been enteredz is finishedz is shutting downN)r; ) r r' r
r r r create_taskr _set_task_nameadd_done_callback
_on_task_doneadd)r coror: r; tasks r r= zTaskGroup.create_task s
} K ID I I IJJ J = B B @D @ @ @AA A > G ED E E EFF F ?:))$//DD:))$)@@D
T4((( t 1222 r r4 returnc h t |t sJ t |t t f S )N)
isinstance
BaseException
SystemExitKeyboardInterrupt)r r4 s r r- zTaskGroup._is_base_error s. #}-----#
,= >???r c x d| _ | j D ]*}| s| +d S )NT)r r donecancel)r ts r r0 zTaskGroup._abort sB A6688
r c | j | | j :| j s3| j s| j d | rd S | }|d S | j | | | r| j
|| _
| j r,| j
d|d| j d||d d S | j s=| j s8| d| _ | j d S d S d S )NTzTask z% has errored out but its parent task z is already completed)message exceptionrC )r discardr rK
set_result cancelledrP r r r- r r
r call_exception_handlerr r r0 rL )r rC r4 s r r@ zTaskGroup._on_task_done s D!!! ! -dk -)..00
8&11$777 >> Fnn ; F C s # # #(8(@ "D ! ! # #
J - - L4 L L#'#4 L L L / /
F~ 'd&C '&
KKMMM,0D ) $ $ & & & & &+ ' ' ' 'r )__name__
__module__ __qualname____doc__r r$ r* r9 r= rG boolr- r0 r@ r r r r s
&
&
& ( ( ( O$ O$ O$b )-d 0@- @D @ @ @ @ 2' 2' 2' 2' 2'r N)__all__r r r r r rZ r r r\ s
- ^' ^' ^' ^' ^' ^' ^' ^' ^' ^'r