U
|
K ±d ã @ sH d dl Z d dlmZ ddlmZ G dd dZG dd dZd d
|
ZdS ) é N)Úwrapsé )Úisasyncgenfunctionc @ s$ e Zd Zdd Zdd Zdd ZdS )Úaclosingc C s
|
|| _ d S ©N©Ú_aiter)ÚselfZaiter© r
|
úLd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\async_generator/_util.pyÚ__init__ s zaclosing.__init__c à s | j S r r ©r r
|
r
|
r Ú
|
__aenter__
|
s zaclosing.__aenter__c Ç s | j ¡ I d H d S r )r Úaclose)r Úargsr
|
r
|
r Ú __aexit__ s zaclosing.__aexit__N)Ú__name__Ú
|
__module__Ú __qualname__r r r r
|
r
|
r
|
r r s r c @ s4 e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd S ) Ú _AsyncGeneratorContextManagerc C s |j | _||| ¡ | _d S r )r Ú
|
_func_nameÚ __aiter__Ú_agen)r Úfuncr Úkwdsr
|
r
|
r r s z&_AsyncGeneratorContextManager.__init__c à sP t jdk r| jI d H | _z| j d ¡I d H W S tk
|
rJ tdd Y nX d S )N)é é é z async generator didn't yield)ÚsysÚ version_infor ÚasendÚStopAsyncIterationÚ RuntimeErrorr r
|
r
|
r r s
|
z(_AsyncGeneratorContextManager.__aenter__c à s t | j4 I d H z |d krhz| j d ¡I d H W n& tk
|
rZ Y W 5 Q I d H R £ dS X tdn"|d k sttz"| j |||¡I d H tdW nò tk
|
rÖ } z"||k W Y ¢W 5 Q I d H R £ S d }~X Y n´ tk
|
rT } z`||kr W Y ¢PW 5 Q I d H R £ dS t|ttfrB|j |krBW Y ¢W 5 Q I d H R £ dS W 5 d }~X Y n6 t
|
¡ d |krY W 5 Q I d H R £ dS Y nX W 5 Q I d H R X d S )NFzasync generator didn't stopz*async generator didn't stop after athrow()r ) r r r r! r" ÚAssertionErrorÚathrowÚ
|
isinstanceÚ StopIterationÚ __cause__r Úexc_info)r ÚtypeÚvalueÚ tracebackÚexcr
|
r
|
r r & s6 ÿ.
|
ÿz'_AsyncGeneratorContextManager.__aexit__c C s t dj| jdd S )Nz>use 'async with {func_name}(...)', not 'with {func_name}(...)')Ú func_name)r" Úformatr r r
|
r
|
r Ú __enter__V s
|
ÿÿz'_AsyncGeneratorContextManager.__enter__c C s ds t dd S )NFz#Never called, but should be defined)r# r r
|
r
|
r Ú__exit__\ s z&_AsyncGeneratorContextManager.__exit__N)r r r r r r r/ r0 r
|
r
|
r
|
r r s
|
0r c s. t stdt fdd}d|_|S )z Like @contextmanager, but async.z{must be an async generator (native or from async_generator; if using @async_generator then @acontextmanager must be on top.c s t | |S r )r )r r ©r r
|
r Úhelperh s z#asynccontextmanager.<locals>.helperT)r Ú TypeErrorr Z__returns_acontextmanager__)r r2 r
|
r1 r Úasynccontextmanager` s ÿr4 )r Ú functoolsr Ú_implr r r r4 r
|
r
|
r
|
r Ú<module> s
|
G
|
