U
>Y @ s& d dl Z dZG dd dZdd ZdS ) Nz1.3.0c @ sn e Zd ZdZddddZdd Zdd Zejd
d Z ejd d
Z
e dd Z dd Z
dd Zdd ZdS )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
Nloopc C sF |d k r|dkrd }|| _ |d kr*t }|| _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 | S N _do_enterr r r r
__enter__ s ztimeout.__enter__c C s | | d S r _do_exitr exc_typeZexc_valZexc_tbr r r
__exit__# s ztimeout.__exit__c C s | S r r r r r r
__aenter__& s ztimeout.__aenter__c C s | | d S r r r r r r
__aexit__* s ztimeout.__aexit__c C s | j S r )r r r r r
expired. s ztimeout.expiredc C s@ | j d k r<t| j| _| jd kr(td| 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 sP |t jkr"| jr"d | _d | _t j| jd k rF| jd k rF| j d | _d | _d S r )r ZCancelledErrorr r
r TimeoutErrorr cancel)r r r r r
r < s
ztimeout._do_exitc C s | 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 jj| d}|d kr(t| dr(| }|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