Format: application/x-python-code
1 o
2
3 -h+ � @ sJ d dl Z d dlZd dlmZ d dlmZ d dlmZmZ G dd� d�ZdS )� N)�
4 TracebackType)�Any)�Optional�Typec @ s� e Zd ZdZdd�d ee deej ddfdd�Zd dd �Z d
5 e
6 e d e d e dee
7 fd
8 d�Zd dd�Zd
9 e
10 e d e d e ddfdd�Zede
11 fdd��Zedee fdd��Zd dd�Zd
12 e
13 e ddfdd�Zd dd �ZdS ) �timeouta� timeout context manager.
14
15 Useful in cases when you want to apply timeout logic around block
16 of code or in cases when asyncio.wait_for is not suitable. For example:
17
18 >>> with timeout(0.001):
19 ... async with aiohttp.get('https://github.com') as r:
20 ... await r.text()
21
22
23 timeout - value in seconds or None to disable timeout logic
24 loop - asyncio compatible event loop
25 N)�loopr �returnc C sF || _ |d u r t�� }nt�dt� || _d | _d| _d | _ d | _
26 d S )Nz-The loop argument to timeout() is deprecated.F) �_timeout�asyncio�get_running_loop�warnings�warn�DeprecationWarning�_loop�_task�
27 _cancelled�_cancel_handler�
28 _cancel_at)�selfr r � r �EC:\Users\shubh\Videos\agent\nova\Lib\site-packages\asgiref/timeout.py�__init__ s
29 �
30 ztimeout.__init__c C s | � � S �N�� _do_enter�r r r r � __enter__2 s ztimeout.__enter__�exc_type�exc_val�exc_tbc C s | � |� d S r ��_do_exit�r r r r r r r �__exit__5 s
31 ztimeout.__exit__c � s
32 �| � � S r r r r r r �
33 __aenter__> s �ztimeout.__aenter__c � s �| � |� d S r r r" r r r � __aexit__A s �ztimeout.__aexit__c C s | j S r )r r r r r �expiredI s ztimeout.expiredc C s$ | j d urt| j | j�� d�S d S )Ng )r �maxr �timer r r r � remainingM s
34 ztimeout.remainingc C st | j d u r| S t�| j�| _| jd u rtd��| j dkr%| j�| j� | S | j�� | j | _ | j�
35 | j | j�| _ | S )Nz4Timeout context manager should be used inside a taskr ) r r
36 � current_taskr r � RuntimeError� call_soon� _cancel_taskr( r �call_atr r r r r r T s
37
38 �
39 ztimeout._do_enterc C sP |t ju r| jrd | _d | _t j�| jd ur#| jd ur#| j�� d | _d | _d S r )r
40 �CancelledErrorr r r � TimeoutErrorr �cancel)r r r r r r! h s
41 ztimeout._do_exitc C s" | j d ur| j �� d| _d S d S )NT)r r1 r r r r r r- s s
42
43
44 �ztimeout._cancel_task)r r )r N)�__name__�
45 __module__� __qualname__�__doc__r �floatr
46 �AbstractEventLoopr r r �
47 BaseExceptionr �boolr# r$ r% �propertyr& r) r r! r- r r r r r sJ ���
48 �
49 ���
50 �
51 ���
52 �
53 r ) r
54 r �typesr �typingr r r r r r r r �<module> s