�
!A?h� � � � d Z ddlZddlZddlZddlZddlZddlZddlmZ ddlm Z ddlm
Z
ddlm Z G d� d ej � � Z
G d
� d � � Zdd �d
�Zd� ZdS ))�Runner�run� N� )�
coroutines)�events)�
exceptions)�tasksc � � e Zd ZdZdZdZdS )�_State�created� initialized�closedN)�__name__�
__module__� __qualname__�CREATED� INITIALIZED�CLOSED� � � � � �n�� ��� � )�����
� ��
� !�� � $�� � � r c �. � | � � � | S �N)�
_lazy_init�r# s r � __enter__zRunner.__enter__: s � � ������� r c �. � | � � � d S r& )�close)r# �exc_type�exc_val�exc_tbs r �__exit__zRunner.__exit__> s � � �
�
� � � � � r c � � | j t j urdS | j }t |� � |� |� � � � � |� |� � � � � | j rt j
d� � |� � � d| _ t j | _ dS # | j rt j
d� � |� � � d| _ t j | _ w xY w)z Shutdown and close event loop.N)
r r r r �_cancel_all_tasks�run_until_complete�shutdown_asyncgens�shutdown_default_executorr" r �set_event_loopr+ r )r# �loops r r+ z Runner.closeA s� � � �;�f� 0� 0� 0� �F�
(��:�D� �d� #� #� #� � #� #�D�$;�$;�$=�$=� >� >� >� � #� #�D�$B�$B�$D�$D� E� E� E��#�
,��%�d�+�+�+� �J�J�L�L�L� �D�J� �-�D�K�K�K�� �#�
,��%�d�+�+�+� �J�J�L�L�L� �D�J� �-�D�K� '� '� '� 's
�A$C �A D
c �8 � | � � � | j S )zReturn embedded event loop.)r' r r( s r �get_loopzRunner.get_loopQ s � � �������z�r ��contextc � � t j |� � s"t d� |� � � � �t j � � �t
d� � �| � � � |�| j }| j �
||�� � }t j � � t j
� � u rxt j t j � � t j u rNt% j | j |�� � } t j t j |� � n# t $ r d}Y nw xY wd}d| _ | j � |� � |�Jt j t j � � |u r+t j t j t j � � S S S # t. j $ r< | j dk r/t3 |dd� � }|� |� � dk rt5 � � �� w xY w# |�Jt j t j � � |u r+t j t j t j � � w w w xY w)z/Run a coroutine inside the embedded event loop.z"a coroutine was expected, got {!r}Nz7Runner.run() cannot be called from a running event loopr9 )� main_taskr �uncancel)r � iscoroutine�
ValueError�formatr �_get_running_loop� RuntimeErrorr' r r � create_task� threading�current_thread� main_thread�signal� getsignal�SIGINT�default_int_handler� functools�partial�
_on_sigintr! r2 r �CancelledError�getattr�KeyboardInterrupt)r# �coror: �task�sigint_handlerr= s r r z
Runner.runV s5 � ��%�d�+�+� P� � A� H� H�� N� N�O�O� O� � #� %� %� 1� �I�K� K�
K�
������ �?��m�G��z�%�%�d�G�%�<�<�� � $� &� &�)�*?�*A�*A� A� A�� �� �/�/�6�3M�M�M� &� .�t��$� O� O� O�N�
&��
�f�m�^�<�<�<�<�� �
&�
&�
&� "&����
&���� "�N� !�� � I��:�0�0��6�6� �*��$�V�]�3�3�~�E�E��
�f�m�V�-G�H�H�H�H� +�E�� �(� � � ��$�q�(�(�"�4��T�:�:���'�H�H�J�J�!�O�O�+�-�-�-� � ����� �*��$�V�]�3�3�~�E�E��
�f�m�V�-G�H�H�H�H� +�E���s, �> D � D-�,D-�:F! �!A G,�,G/ �/AH>c �� � | j t j u rt d� � �| j t j u rd S | j �@t
j � � | _ | j s t
j
| j � � d| _ n| � � � | _ | j � | j � | j � � t j � � | _ t j | _ d S )NzRunner is closedT)r r r rB r r r �new_event_loopr r" r5 r � set_debug� contextvars� copy_contextr r( s r r' zRunner._lazy_init� s� � � �;�&�-� '� '� � 1�2�2� 2� �;�&�,� ,� ,� �F� � � %� �.�0�0�D�J��'�
,� �%�d�j�1�1�1�'+��$�� �+�+�-�-�D�J� �;� "� �J� � ��� -� -� -�#�0�2�2��
� �(�� � � r c �� � | xj dz
c_ | j dk rE|� � � s1|� � � | j � d� � � d S t � � �)Nr c � � d S r& r r r r �z#Runner._on_sigint..� s � �D� r )r! �done�cancelr �call_soon_threadsaferP )r# �signum�framer< s r rM zRunner._on_sigint� sn � � � � ��"� � � � �A� %� %�i�n�n�.>�.>� %� � � � � � � �J� +� +�L�L� 9� 9� 9� �F� �!�!�!r ) r r r �__doc__r$ r) r/ r+ r8 r r' rM r r r r r s� � � � � � �� �6 !%�4� %� %� %� %� %�� � �� � �(� (� (� � � �
$(� +I� +I� +I� +I� +I�Z)� )� )�&"� "� "� "� "r r �r c �� � t j � � �t d� � �t |�� � 5 }|� | � � cddd� � S # 1 swxY w Y dS )a� Execute the coroutine and return the result.
This function runs the passed coroutine, taking care of
managing the asyncio event loop and finalizing asynchronous
generators.
This function cannot be called when another asyncio event loop is
running in the same thread.
If debug is True, the event loop will be run in debug mode.
This function always creates a new event loop and closes it at the end.
It should be used as a main entry point for asyncio programs, and should
ideally only be called once.
Example:
async def main():
await asyncio.sleep(1)
print('hello')
asyncio.run(main())
Nz8asyncio.run() cannot be called from a running event looprb )r rA rB r r )�mainr �runners r r r � s� � �0 � �!�!�-�� F�H� H� H�
�e� � � � ���z�z�$� � � � � � � � � � � � � � ���� � � � � � s �A�A� Ac �b � t j | � � }|sd S |D ]}|� � � �| � t j |ddi�� � |D ]V}|� � � r�|� � � �+| � d|� � � |d�� � �Wd S )N�return_exceptionsTz1unhandled exception during asyncio.run() shutdown)�message� exceptionrR )r � all_tasksr] r2 �gather� cancelledri �call_exception_handler)r6 � to_cancelrR s r r1 r1 � s� � ����%�%�I� � ���� � �� � � �
�
�
�
����E�L�)� L�t� L� L�M�M�M�� � �� �>�>� � � � � �>�>� � � '� � '� '�N� !�^�^� -� -� �)� )�
�
�
�� � r )�__all__rW �enumrK rD rG �sys� r r r r �Enumr r r r1 r r r �rt s& ��
�� � � � � � � � � � � � � � � � �
�
�
�
�
�
�
�
� � � � � � � � � � � � � � � � � � � � � � � � �� � � � �T�Y� � � � H"� H"� H"� H"� H"� H"� H"� H"�V � � � � � �B� � � � r