>Y @ s5 d d l Z d Z Gd d d Z d d Z d S) Nz1.3.0c @ s e Z d Z d Z d d d d Z d d Z d d Z e j d
d Z e j d d
Z
e d d Z d d Z
d d Z d d Z d S)timeouta 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:
>>> with timeout(0.001):
... async with aiohttp.get('https://github.com') as r:
... await r.text()
timeout - value in seconds or None to disable timeout logic
loop - asyncio compatible event loop
loopNc C sg | d k r | d k r d } | | _ | d k r? t j } | | _ d | _ d | _ d | _ d S)Nr F)_timeoutasyncioZget_event_loop_loop_task
_cancelled_cancel_handler)selfr r r /__init__.py__init__ s ztimeout.__init__c C s
| j S)N) _do_enter)r
r r r __enter__ s ztimeout.__enter__c C s | j | d S)N)_do_exit)r
exc_typeexc_valexc_tbr r r __exit__# s ztimeout.__exit__c C s
| j S)N)r )r
r r r
__aenter__& s ztimeout.__aenter__c C s | j | d S)N)r )r
r r r r r r __aexit__* s ztimeout.__aexit__c C s | j S)N)r )r
r r r expired. s ztimeout.expiredc C s^ | j d k rZ t | j | _ | j d k r< t d | j j | j | j | _ | S)Nz4Timeout context manager should be used inside a task)r current_taskr r RuntimeErrorZ
call_later _cancel_taskr )r
r r r r 2 s ztimeout._do_enterc C st | t j k r3 | j r3 d | _ d | _ t j | j d k rg | j d k rg | j j d | _ d | _ d S)N)r CancelledErrorr r r TimeoutErrorr cancel)r
r r r r r < s
ztimeout._do_exitc C s | j j d | _ d S)NT)r r r )r
r r r r F s
ztimeout._cancel_task)__name__
__module__ __qualname____doc__r
r r r coroutiner r propertyr r r r r r r r r s
r c C s@ t j j d | } | d k r< t | d r< | j } | S)Nr r )r ZTaskr hasattr)r Ztaskr r r r K s
r )r __version__r r r r r r <module> s D