o aZ@sHddlZddlmZddlmZGdddZGdddZd d ZdS) N)wraps)isasyncgenfunctionc@s$eZdZddZddZddZdS)aclosingcCs ||_dSN_aiter)selfaiterr 9usr/lib/python3.10/site-packages/async_generator/_util.py__init__s zaclosing.__init__cs|jSrrr r r r __aenter__ szaclosing.__aenter__cs|jIdHdSr)raclose)r argsr r r __aexit__ szaclosing.__aexit__N)__name__ __module__ __qualname__r rrr r r r rs rc@s4eZdZddZddZddZddZd d Zd S) _AsyncGeneratorContextManagercCs |j|_||i||_dSr)r _func_name __aiter___agen)r funcrkwdsr r r r sz&_AsyncGeneratorContextManager.__init__csHtjdkr |jIdH|_z |jdIdHWSty#tddw)N)zasync generator didn't yield)sys version_inforasendStopAsyncIteration RuntimeErrorrr r r rs   z(_AsyncGeneratorContextManager.__aenter__c st|j4IdH|dur1z|jdIdHWtdty0YWdIdHdSw|dus7Jz|j|||IdHtdtyf}z||uWYd}~WdIdHSd}~wty}z1||urWYd}~WdIdHdSt|ttfr|j|urWYd}~WdIdHdSd}~wt d|urYWdIdHdS1IdHswYdS)NFzasync generator didn't stopz*async generator didn't stop after athrow()r) rrr!r"r#athrow isinstance StopIteration __cause__rexc_info)r typevalue tracebackexcr r r r&sH     "-z'_AsyncGeneratorContextManager.__aexit__cCstdj|jd)Nz>use 'async with {func_name}(...)', not 'with {func_name}(...)') func_name)r#formatrrr r r __enter__Vs z'_AsyncGeneratorContextManager.__enter__cCsJd)NFz#Never called, but should be definedr rr r r __exit__\sz&_AsyncGeneratorContextManager.__exit__N)rrrr rrr/r0r r r r rs 0 rcs.tstdtfdd}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.cs t||Sr)r)rrrr r helperhs z#asynccontextmanager..helperT)r TypeErrorrZ__returns_acontextmanager__)rr2r r1r asynccontextmanager`sr4)r functoolsr_implrrrr4r r r r s    G