관리-도구
편집 파일: taskgroups.cpython-312.pyc
� �Q�f+" � �@ � d Z ddlmZ ddlmZ ddlmZ G d� d� Zy))� TaskGroup� )�events)� exceptions)�tasksc �R � e Zd ZdZd� Zd� Zd� Zd� Zddd�d�Zd e d e fd �Z d � Z d � Z y)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 | _ y )NF) �_entered�_exiting� _aborting�_loop� _parent_task�_parent_cancel_requested�set�_tasks�_errors� _base_error�_on_completed_fut��selfs �9/opt/alt/python312/lib64/python3.12/asyncio/taskgroups.py�__init__zTaskGroup.__init__ sN � � �� � �� � ����� � ���(-��%��e�� ��� � ���!%�� � c �x � dg}| j r'|j dt | j � � �� | j r'|j dt | j � � �� | j r|j d� n | j r|j d� dj |� }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�� �;�;� �K�K�&��T�[�[�!1� 2�3� 4� �<�<� �K�K�'�#�d�l�l�"3�!4�5� 6� �>�>� �K�K� � %� �]�]� �K�K� � "��8�8�D�>���H�:�Q�'�'r c � � K � | j rt d| �d�� �| j �t j � | _ t j | j � | _ | j �t d| �d�� �d| _ | S �w)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�H�$=�>�@� @� �:�:� � �0�0�2�D�J� !� .� .�t�z�z� :��� � � � $� � �T�H�$E�F�H� H� �� �� �s �B B c � � K � d| _ |�$| j |� r| j �|| _ |t j u r|nd }| j r | j j � dk( rd }|� | j s| j � | j rT| j � | j j � | _ | j � d { ��� d | _ | j r�T| j rJ �| j � | j �|r| j s|�|�-|t j ur| j j |� | j r t! d| j � }|d �y 7 ��# t j $ r(}| j s|}| j � Y d }~��d }~ww xY w# d | _ w xY w�w)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�� %� � (� (�� � �)�)�+�q�0� 04�,� �>��>�>� � � � � �k�k��%�%�-�)-���)A�)A�)C��&� "��,�,�,�,� &*�D� "�' �k�k�* �;�;� �� � � � '��"�"� "� (�� � �.� .� �>�b� �(A�(A� A� �L�L� � �� $� �<�<� $�'�(I�4�<�<�X���d�"� �; -�� �,�,� "��~�~� 46�0��K�K�M��� "��H $�� �sa �CG�E= �E;�E= �G�/A1G�!F; �:G�;E= �=F8� F3�.G�3F8�8G�; G�GN)�name�contextc � � | j st d| �d�� �| j r| j st d| �d�� �| j rt d| �d�� �|� | j j |� }n | j j ||�� }t j ||� |j � r| j |� |S | j 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 down)r; ) r r' r r r r � create_taskr �_set_task_name�done� _on_task_done�add�add_done_callback)r �coror: r; �tasks r r= zTaskGroup.create_task� s� � � �}�}� ��D�8�3H� I�J� J� �=�=���� ��D�8�<� @�A� A� �>�>� ��D�8�3D� E�F� F� �?��:�:�)�)�$�/�D��:�:�)�)�$��)�@�D� � � �T�4�(� �9�9�;� � � �t� $� � � �K�K�O�O�D� !� � "� "�4�#5�#5� 6�� r r4 �returnc �R � 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 �v � d| _ | j D ]# }|j � r�|j � �% y )NT)r r r? �cancel)r �ts r r0 zTaskGroup._abort� s) � � ������A��6�6�8���� � r c �� � | j j |� | j �A| j s5| j j � s| j j d� |j � ry |j � }|�y | j j |� | j |� r| j �|| _ | j j � r1| j j d|�d| j � d�||d�� y | j s?| j s2| j! � d| _ | j j# � y y y )NTzTask z% has errored out but its parent task z is already completed)�message� exceptionrD )r �discardr r? � set_result� cancelledrP r r r- r r r �call_exception_handlerr r r0 rL )r rD r4 s r r@ zTaskGroup._on_task_done� s3 � � � � ���D�!� � !� !� -�d�k�k��)�)�.�.�0��&�&�1�1�$�7� �>�>� � ��n�n� �� �;� � � � ���C� � � � �s� #��(8�(8�(@� "�D� � � � � !� !� #� �J�J� -� -� "�4�(� +#�#'�#4�#4�"5�5J� L� � � /� � ��~�~�d�&C�&C�& �K�K�M�,0�D� )� � � � $� $� &�+ 'D�~r )�__name__� __module__� __qualname__�__doc__r r$ r* r9 r= rH �boolr- r0 r@ � r r r r sI � �� &� (� � O$�b )-�d� �<@�-� @�D� @��2'r r N)�__all__r r r r r rZ r r �<module>r\ s! �� �� � � �d'� d'r