�
*�Kg� � �� � d dl Z d dlmZ d dlmZmZmZ ddlmZ ddlm Z ddlm
Z
dZ G d � d
e j � � Z
e G d � d � � � � Zd
ee defd�Zdee defd�ZdS )� N)�
TracebackType)�final�Optional�Type� )�events)�
exceptions)�tasks)�Timeout�timeout�
timeout_atc �" � e Zd ZdZdZdZdZdZdS )�_State�created�active�expiring�expired�finishedN)�__name__�
__module__� __qualname__�CREATED�ENTERED�EXPIRING�EXPIRED�EXITED� � �'/usr/lib/python3.11/asyncio/timeouts.pyr r s' � � � � � ��G��G��H��G�
�F�F�Fr r c �� � e Zd Zdee ddfd�Zdee fd�Zdee ddfd�Zdefd�Z de
fd�Z dd �Z d
ee
e d ee d ee dee fd
�Zdd�ZdS )r �when�returnNc �R � t j | _ d | _ d | _ || _ d S �N)r r �_state�_timeout_handler�_task�_when)�selfr! s r �__init__zTimeout.__init__ s% � � �n�� �>B�� �+/��
���
�
�
r c � � | j S r$ )r( �r) s r r! z Timeout.when$ s
� ��z�r c �� � | j t j usJ �| j t j ur t d| j j � d�� � �|| _ | j �| j � � � |� d | _ d S t j
� � }||� � � k r!|� | j
� � | _ d S |� || j
� � | _ d S )NzCannot change state of z Timeout)r% r r r � RuntimeError�valuer( r& �cancelr �get_running_loop�time� call_soon� _on_timeout�call_at)r) r! �loops r �
reschedulezTimeout.reschedule' s� � ��{�&�.�0�0�0�0� �;�f�n� ,� ,� �E�$�+�*;�E�E�E�� �
� ��
� � � ,� � !� (� (� *� *� *� �<�$(�D� !� !� !��*�,�,�D��t�y�y�{�{�"�"�(,���t�7G�(H�(H��%�%�%�(,� � �T�4�;K�(L�(L��%�%�%r c �@ � | j t j t j fv S )z$Is timeout expired during execution?)r% r r r r, s r r zTimeout.expired< s � ��{�v���� ?�?�?r c �� � dg}| j t j u r6| j �t | j d� � nd }|� d|� �� � d� |� � }d| j j � d|� d�S )N� � zwhen=� z
<Timeout [�]�>)r% r r r( �round�append�joinr/ )r) �infor! �info_strs r �__repr__zTimeout.__repr__@ sz � ��t�� �;�&�.� (� (�+/�:�+A�5���Q�'�'�'�t�D� �K�K����� '� '� '��8�8�D�>�>��;�D�K� -�;�;��;�;�;�;r c � � K � t j | _ t j � � | _ | j �t
d� � �| � | j � � | S )Nz$Timeout should be used inside a task) r r r% r
� current_taskr' r. r7 r( r, s r �
__aenter__zTimeout.__aenter__H sO � � � � �n�� ��'�)�)��
� �:� � � E�F�F� F� ����
�#�#�#�� r �exc_type�exc_val�exc_tbc � � K � | j t j t j fv sJ �| j � | j � � � d | _ | j t j u rDt j | _ | j � � � dk r|t j
u rt �n$| j t j u rt j | _ d S )Nr )
r% r r r r& r0 r r' �uncancelr �CancelledError� TimeoutErrorr )r) rH rI rJ s r � __aexit__zTimeout.__aexit__P s� � � � � �{�v�~�v�� ?�?�?�?�?� � � ,� � !� (� (� *� *� *�$(�D� !� �;�&�/� )� )� �.�D�K��z�"�"�$�$��)�)�h�*�:S�.S�.S� #�"��
�[�F�N�
*�
*� �-�D�K��tr c � � | j t j u sJ �| j � � � t j | _ d | _ d S r$ )r% r r r' r0 r r& r, s r r4 zTimeout._on_timeouth sB � ��{�f�n�,�,�,�,� �
������ �o�� � $�� � � r )r" r )r" N)r r r r �floatr* r! r7 �boolr �strrD rG r �
BaseExceptionr rO r4 r r r r r s: � � � � � ��X�e�_� �� � � � ��h�u�o� � � � �M�x��� M�4� M� M� M� M�*@�� @� @� @� @�<�#� <� <� <� <�� � � ���4�
�.�/�� �-�(�� � �'� �
�$� � � � � �0%� %� %� %� %� %r r �delayr" c �x � t j � � }t | �|� � � | z nd� � S )a Timeout async context manager.
Useful in cases when you want to apply timeout logic around block
of code or in cases when asyncio.wait_for is not suitable. For example:
>>> async with asyncio.timeout(10): # 10 seconds timeout
... await long_running_task()
delay - value in seconds or None to disable timeout logic
long_running_task() is interrupted by raising asyncio.CancelledError,
the top-most affected timeout() context manager converts CancelledError
into TimeoutError.
N)r r1 r r2 )rU r6 s r r r p s7 � � � "� $� $�D� �%�*;�4�9�9�;�;��&�&�� F� F�Fr r! c � � t | � � S )ab Schedule the timeout at absolute time.
Like timeout() but argument gives absolute time in the same clock system
as loop.time().
Please note: it is not POSIX time but a time with
undefined starting base, e.g. the time of the system power on.
>>> async with asyncio.timeout_at(loop.time() + 10):
... await long_running_task()
when - a deadline when timeout occurs or None to disable timeout logic
long_running_task() is interrupted by raising asyncio.CancelledError,
the top-most affected timeout() context manager converts CancelledError
into TimeoutError.
)r )r! s r r
r
� s � �& �4�=�=�r )�enum�typesr �typingr r r r: r r r
�__all__�Enumr r rQ r r
r r r �<module>r] s@ �� � � � � � � � � � � (� (� (� (� (� (� (� (� (� (� � � � � � � � � � � � � � � � � � � ��� � � � �T�Y� � � � �R%� R%� R%� R%� R%� R%� R%� ��R%�jG�8�E�?� G�w� G� G� G� G�(�X�e�_� �� � � � � � r |