JezK
Edit File: taskgroups.cpython-311.pyc
� !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 �+/usr/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 <TaskGroup�>)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 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 �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�E2 �2 E;N)�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 �h � t |t � � sJ �t |t t f� � S )N)� isinstance� BaseException� 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 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�!�!�!� � !� -�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= 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 �<module>r\ s� �� �-�� � � � � � � � � � � � � � � � � � �^'� ^'� ^'� ^'� ^'� ^'� ^'� ^'� ^'� ^'r