o i@sddlZddlZddlZddlZddlZddlmZddlZddl m Z m Z ddl m Z de fddZd ejd efd d Zd ejd efd dZedZedZedZdejeejeeffd ejeejeffddZdS)N)asynccontextmanager)UserCodeException suppress_synchronicity_tb_frames) Interface interfacecs,tr|tjkrfdd}|StS)aLike functools.wraps but maintains `inspect.iscoroutinefunction` and allows custom type annotations overrides Use this when the wrapper function is non-async but returns the coroutine resulting from calling the underlying wrapped `func`. This will make sure that the wrapper is still an async function in that case, and can be inspected as such. Note: Does not forward async generator information other than explicit annotations cstfdd}|S)Nc sjt'z|i|IdHWWdSty*}zd|j_|jd}~ww1s.wYdS)NT)rrexc__suppress_context__)argskwargsuc_exc user_wrapperL/home/ubuntu/.local/lib/python3.10/site-packages/synchronicity/async_wrap.pywrappersz;wraps_by_interface..asyncfunc_deco..wrapper) functoolswraps)rrfuncr rasyncfunc_decosz*wraps_by_interface..asyncfunc_deco)$is_coroutine_function_follow_wrappedr_ASYNC_WITH_BLOCKING_TYPESrr)rrrrrrwraps_by_interfaces  rrreturncC<ddlm}t|drt||dtjkrt|jSt |S)zgDetermine if func returns a coroutine, unwrapping decorators, but not the async synchronicity interace.rTARGET_INTERFACE_ATTR __wrapped__N) synchronizerrhasattrgetattrrBLOCKINGrrinspectiscoroutinefunctionrrrrrr*   rcCr)znDetermine if func returns an async generator, unwrapping decorators, but not the async synchronicity interace.rrrN) rrr r!rr"$is_async_gen_function_follow_wrappedrr#isasyncgenfunctionr%rrrr'3r&r' YIELD_TYPE SEND_TYPEPfcCspt|}|jdd}|dur4|jtjjtjjtjjfvr*t j |j d|jd<|S|jt j kr2|S|Std)z{Wrapper around contextlib.asynccontextmanager that sets correct type annotations The standard library one doesn't rNrzpTo use the fixed @asynccontextmanager, make sure to properly annotate your wrapped function as an AsyncGenerator)_asynccontextmanager__annotations__pop __origin__ collectionsabcAsyncGenerator AsyncIterator AsyncIterabletypingAsyncContextManager__args__ contextlibAbstractAsyncContextManager ValueError)r, acm_factoryold_retrrrrCs  r)collections.abcr1r9rr#r6rr-typing_extensions exceptionsrrrrrCallableboolrr'TypeVarr)r* ParamSpecr+r3r7rrrrs(