�
!A?h! � �J � d gZ ddlmZ ddlmZ ddlmZ G d� d � � ZdS )� TaskGroup� )�events)�
exceptions)�tasksc �T � e Zd ZdZd� Zd� Zd� Zd� Zddd�d�Zd 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_requested�set�_tasks�_errors� _base_error�_on_completed_fut��selfs �?/opt/alt/python-internal/lib64/python3.11/asyncio/taskgroups.py�__init__zTaskGroup.__init__ sV � � ��
� ��
� �����
� ���(-��%��e�e�� ��� � ���!%�� � � � c �t � dg}| j r*|� dt | j � � � �� � | j r*|� dt | j � � � �� � | j r|� d� � n | j r|� d� � d� |� � }d|� d�S ) N� ztasks=zerrors=�
cancelling�entered� z
)r �append�lenr r r �join)r �info�info_strs r �__repr__zTaskGroup.__repr__( s� � ��t�� �;� 5� �K�K�3��T�[�!1�!1�3�3� 4� 4� 4� �<� 7� �K�K�5�#�d�l�"3�"3�5�5� 6� 6� 6� �>� #� �K�K� � %� %� %� %�
�]� #� �K�K� � "� "� "��8�8�D�>�>��'�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�>�>�>�@� @�
@� �:� � �0�2�2�D�J� !� .�t�z� :� :��� � � $� �F�T�F�F�F�H� 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 �| 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 �et�exc�tb� propagate_cancellation_error�ex�mes r � __aexit__zTaskGroup.__aexit__D s� � � � � ��
� �O��#�#�C�(�(�
�� �(� "�D� � ��2�2�2�C�C�� %� � (� 4�� �)�)�+�+�q�0�0� 04�,�
�>��>�
� � � �
�
�
� �k� *��%�-�)-��)A�)A�)C�)C��&�
"��,�,�,�,�,�,�,�,�,�� �,�
"�
"�
"��~�
"� 46�0��K�K�M�M�M����������
"���� &*�D� "�' �k� *�. � � '��"� "� (� /�� � /�.� .�
�>�b�
�(A� A� A� �L� � �� $� $� $� �<� $�
$�'�(I�4�<�X�X���d�"�� #�� �#�#�#�#� $� $s$ �1
B? �?C5� C0�0C5�E) �) E2N)�name�contextc � � | 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_name�add_done_callback�
_on_task_done�add)r �coror: r; �tasks r r= zTaskGroup.create_task� s� � �
�}� K� � I�D� I� I� I�J�J� J� �=� B��� B� � @�D� @� @� @�A�A� A� �>� G� � E�D� E� E� E�F�F� F� �?��:�)�)�$�/�/�D�D��:�)�)�$��)�@�@�D�
� �T�4�(�(�(� � � �t� 1�2�2�2� � ���� � � �� r r4 �returnc �: � t |t t f� � S )N)�
isinstance�
SystemExit�KeyboardInterrupt)r r4 s r r- zTaskGroup._is_base_error� s � ��#�
�,=� >�?�?�?r c �x � d| _ | j D ]*}|� � � s|� � � �+d S )NT)r r �done�cancel)r �ts r r0 zTaskGroup._abort� sB � � ����� � �A��6�6�8�8�
����
�
�
�� � 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 rJ �
set_result� cancelledrO r r r- r r
r �call_exception_handlerr r r0 rK )r rC r4 s r r@ zTaskGroup._on_task_done� s� � � � ���D�!�!�!� � !� -�d�k� -��)�.�.�0�0�
8��&�1�1�$�7�7�7� �>�>� � � � �F��n�n� � �� �;� �F� � ���C� � � � � � �s� #� #� #��(8�(@� "�D� � � � !� !� #� #� �
�J� -� -� L�4� L� L�#'�#4� L� L� L� � � /� /�
�
�
�
�F��~� '�d�&C� '�&
�K�K�M�M�M�,0�D� )� � � $� $� &� &� &� &� &�+ '� '� '� 'r )�__name__�
__module__� __qualname__�__doc__r r$ r* r9 r= �
BaseException�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