
    4bh                         d dl Z d dlmZ d dlmZmZmZ ddlmZ ddlm	Z	 ddlm
Z
 dZ G d	 d
e j                        Ze G d d             Zdee   defdZdee   defdZy)    N)TracebackType)finalOptionalType   )events)
exceptions)tasks)Timeouttimeout
timeout_atc                        e Zd ZdZdZdZdZdZy)_StatecreatedactiveexpiringexpiredfinishedN)__name__
__module____qualname__CREATEDENTEREDEXPIRINGEXPIREDEXITED     :/opt/python-3.12.12/usr/lib/python3.12/asyncio/timeouts.pyr   r      s    GGHGFr   r   c            	           e Zd Z	 dee   ddfdZdee   fdZdee   ddfdZdefdZ	de
fdZdd	Zd
eee      dee   dee   dee   fdZddZy)r   whenreturnNc                 Z    	 t         j                  | _        d | _        d | _        || _        y N)r   r   _state_timeout_handler_task_when)selfr!   s     r   __init__zTimeout.__init__!   s*    	 nn>B+/

r   c                     	 | j                   S r$   )r(   r)   s    r   r!   zTimeout.when.   s    *zzr   c                    	 | j                   t        j                  urJ| j                   t        j                  u rt	        d      t	        d| j                   j
                   d      || _        | j                  | j                  j                          |d | _        y t        j                         }||j                         k  r!|j                  | j                        | _        y |j                  || j                        | _        y )NzTimeout has not been enteredzCannot 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   
reschedulezTimeout.reschedule2   s    %;;fnn,{{fnn,"#ABB)$++*;*;)<HE  
  ,!!((*<$(D!**,Dtyy{"(,t7G7G(H%(,T4;K;K(L%r   c                 \    	 | j                   t        j                  t        j                  fv S r$   )r%   r   r   r   r,   s    r   r   zTimeout.expiredI   s!    2{{v???r   c                    dg}| j                   t        j                  u r8| j                  t	        | j                  d      nd }|j                  d|        dj                  |      }d| j                   j                   d| dS )N    zwhen= z
<Timeout []>)r%   r   r   r(   roundappendjoinr/   )r)   infor!   info_strs       r   __repr__zTimeout.__repr__M   ss    t;;&..(+/::+A5Q'tDKK%v'88D>DKK--.az;;r   c                 J  K   | j                   t        j                  urt        d      t	        j
                         }|t        d      t        j                  | _         || _        | j                  j                         | _	        | j                  | j                         | S w)Nz Timeout has already been enteredz$Timeout should be used inside a task)r%   r   r   r.   r
   current_taskr   r'   
cancelling_cancellingr7   r(   )r)   tasks     r   
__aenter__zTimeout.__aenter__U   s}     ;;fnn,ABB!!#<EFFnn
::002

#s   B!B#exc_typeexc_valexc_tbc                   K   | j                   !| j                   j                          d | _         | j                  t        j                  u rVt        j
                  | _        | j                  j                         | j                  k  r|t        j                  u rt        |y | j                  t        j                  u rt        j                  | _        y wr$   )r&   r0   r%   r   r   r   r'   uncancelrH   r	   CancelledErrorTimeoutErrorr   r   )r)   rK   rL   rM   s       r   	__aexit__zTimeout.__aexit__a   s        ,!!((*$(D!;;&//) ..DKzz""$(8(88XIbIb=b #/  [[FNN* --DKs   CCc                 p    | j                   j                          t        j                  | _        d | _        y r$   )r'   r0   r   r   r%   r&   r,   s    r   r4   zTimeout._on_timeouty   s%    

oo $r   )r"   r   )r"   N)r   r   r   r   floatr*   r!   r7   boolr   strrD   rJ   r   BaseExceptionr   rR   r4   r   r   r   r   r      s    
Xe_  huo Mx M4 M.@ @<# <
4./ -( '	
 
$0%r   r   delayr"   c                 t    	 t        j                         }t        | |j                         | z         S d       S r$   )r   r1   r   r2   )rX   r6   s     r   r   r      s:     ""$D%*;499;&FFFFr   r!   c                     	 t        |       S r$   )r   )r!   s    r   r   r      s    $ 4=r   )enumtypesr   typingr   r   r   r:   r   r	   r
   __all__Enumr   r   rT   r   r   r   r   r   <module>r`      s      ( (   TYY  c% c% c%LG8E? Gw G(Xe_  r   