o
    h#                     @   s   d dl Z d dlZd dlZd dlmZ d dlmZmZmZ dZ	dZ
dee ddfd	d
Zdee ddfddZG dd dejZejdkrXG dd dZeG dd de jZdS eG dd dZdS )    N)TracebackType)OptionalTypefinalz5.0.1)timeout
timeout_atTimeoutdelayreturnr   c                 C   s,   t  }| dur| |  }nd}t||S )ar  timeout 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 timeout(0.001):
    ...     async with aiohttp.get('https://github.com') as r:
    ...         await r.text()


    delay - value in seconds or None to disable timeout logic
    N)asyncioget_running_looptimer   )r	   loopdeadline r   j/var/www/html/construction_image-detection-poc/venv/lib/python3.10/site-packages/async_timeout/__init__.pyr      s
   
r   r   c                 C   s   t  }t| |S )a  Schedule the timeout at absolute time.

    deadline argument points on the 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 timeout_at(loop.time() + 10):
    ...     async with aiohttp.get('https://github.com') as r:
    ...         await r.text()


    )r   r   r   )r   r   r   r   r   r   #   s   
r   c                   @   s   e Zd ZdZdZdZdZdS )_StateINITENTERTIMEOUTEXITN)__name__
__module____qualname__r   r   r   r   r   r   r   r   r   6   s
    r   )      c                   @   sZ   e Zd ZdZdeddfddZdefddZdefd	d
ZdefddZ	defddZ
dS )_Expired_valvalr
   Nc                 C   s
   || _ d S Nr   )selfr   r   r   r   __init__B      
z_Expired.__init__c                 C      | j S r    r   r!   r   r   r   __call__E      z_Expired.__call__c                 C   r$   r    r   r%   r   r   r   __bool__H   r'   z_Expired.__bool__c                 C   
   t | jS r    )reprr   r%   r   r   r   __repr__K   r#   z_Expired.__repr__c                 C   r)   r    )strr   r%   r   r   r   __str__N   r#   z_Expired.__str__)r   r   r   	__slots__boolr"   r&   r(   r,   r+   r-   r   r   r   r   r   ?   s    r   c                       s   e Zd Zdee dejddf fddZede	f fddZ
edee fd	d
ZdddZdeddfddZdeddfddZ  ZS )r   r   r   r
   Nc                    s   t  | d S r    )superr"   r!   r   r   	__class__r   r   r"   V   s   Timeout.__init__c                    s   t t  S r    )r   r0   expiredr%   r2   r   r   r5   [   s   Timeout.expiredc                 C   s   |   S r    )whenr%   r   r   r   r   b   s   Timeout.deadlinec                 C   s   |  d dS ) Reject scheduled timeout if any.N
rescheduler%   r   r   r   rejectf   s   Timeout.rejectr	   c                 C   s*   |   }|du rtd| ||  dS zAdvance timeout on delay seconds.

            The delay can be negative.

            Raise RuntimeError if shift is called when deadline is not scheduled
            Nz1cannot shift timeout if deadline is not scheduled)r7   RuntimeErrorr;   r!   r	   r   r   r   r   shiftl   s   Timeout.shiftc                 C   s   |  | dS )l  Set deadline to absolute value.

            deadline argument points on the time in the same clock system
            as loop.time().

            If new deadline is in the past the timeout is raised immediately.

            Please note: it is not POSIX time but a time with
            undefined starting base, e.g. the time of the system power on.
            Nr:   r!   r   r   r   r   updatex   s   Timeout.updater
   N)r   r   r   r   floatr   AbstractEventLoopr"   propertyr   r5   r   r<   rA   rE   __classcell__r   r   r2   r   r   Q   s    
c                	   @   s   e Zd ZdZdee dejddfddZd$dd	Z	d
ee
e  dee dee dee fddZedefddZedee fddZd%ddZd%ddZdeddfddZdeddfddZd%ddZd%ddZd
ee
e  ddfd d!Zd%d"d#ZdS )&r   )	_deadline_loop_state_timeout_handler_taskr   r   r
   Nc                 C   s:   || _ tj| _d | _d | _|d u rd | _d S | | d S r    )rM   r   r   rN   rP   rO   rL   rE   r1   r   r   r   r"      s   
r4   c                    s   |    | S r    )	_do_enterr%   r   r   r   
__aenter__   s   zTimeout.__aenter__exc_typeexc_valexc_tbc                    s   |  | d S r    )_do_exit)r!   rS   rT   rU   r   r   r   	__aexit__   s   
zTimeout.__aexit__c                 C   s   | j tjkS )z$Is timeout expired during execution?)rN   r   r   r%   r   r   r   r5      s   r6   c                 C   r$   r    )rL   r%   r   r   r   r      s   r8   c                 C   s0   | j tjtjfvrtd| j j |   dS )r9   invalid state N)rN   r   r   r   r?   value_rejectr%   r   r   r   r<      s   r=   c                 C   s(   d | _ | jd ur| j  d | _d S d S r    )rP   rO   cancelr%   r   r   r   rZ      s
   


zTimeout._rejectr	   c                 C   s(   | j }|du rtd| ||  dS r>   )rL   r?   rE   r@   r   r   r   rA      s   rB   c                 C   s^   | j tjkr
td| j tjkrtd| jdur| j  || _| j tjkr-| 	  dS dS )rC   z1cannot reschedule after exit from context managerz!cannot reschedule expired timeoutN)
rN   r   r   r?   r   rO   r[   rL   r   _reschedulerD   r   r   r   rE      s   

rF   c                 C   s|   | j tjksJ | j}|d u rd S | j }| jd ur | j  t	 | _
||kr3| j| j| _d S | j|| j| _d S r    )rN   r   r   rL   rM   r   rO   r[   r   current_taskrP   	call_soon_on_timeoutcall_at)r!   r   nowr   r   r   r\      s   



zTimeout._reschedulec                 C   s2   | j tjkrtd| j j tj| _ |   d S )NrX   )rN   r   r   r?   rY   r   r\   r%   r   r   r   rQ      s   zTimeout._do_enterc                 C   sJ   |t ju r| jtjkr| jd usJ d | _d | _t jtj| _| 	  d S r    )
r   CancelledErrorrN   r   r   rP   rO   TimeoutErrorr   rZ   )r!   rS   r   r   r   rV     s   zTimeout._do_exitc                 C   s*   | j d usJ | j   tj| _d | _d S r    )rP   r[   r   r   rN   rO   r%   r   r   r   r_     s   

zTimeout._on_timeout)r
   r   rG   )r   r   r   r.   r   rH   r   rI   r"   rR   r   BaseExceptionr   r/   rW   rJ   r5   r   r<   rZ   rA   rE   r\   rQ   rV   r_   r   r   r   r   r      s>    



	



)r   enumsystypesr   typingr   r   r   __version____all__rH   r   r   Enumr   version_infor   r   r   r   r   r   <module>   s     
5