U 7T_@4@sddlZddlZddlZddlZddlZddlZddlmZmZddl m Z ddl m Z z ddl Z Wnek r|dZ YnXddZGdd d ZGd d d ZeZeZdS) N)FutureThreadPoolExecutor)CurrentThreadExecutor)Localc CsZ|D]P}z&|||kr,|||Wqtk rR|||YqXqdSN)getset LookupError)contextZcvarr 0/tmp/pip-unpacked-wheel-_lltyzrh/asgiref/sync.py_restore_contexts rc@sDeZdZdZiZeZdddZddZddZ d d Z d d Z d S) AsyncToSynca Utility class which turns an awaitable that only works on the thread with the event loop into a synchronous callable that works in a subthread. If the call stack contains an async loop, the code runs there. Otherwise, the code runs in a new loop in a new thread. Either way, this thread then pauses and waits to run any thread_sensitive code called from further down the call stack using SyncToAsync, before finally exiting once the async task returns. FcCsn||_z|jj|_Wntk r(YnX|r6d|_n4zt|_Wn$tk rhttj dd|_YnXdS)Nmain_event_loop) awaitable__self__AttributeErrorrasyncioget_event_loop RuntimeErrorgetattr SyncToAsync threadlocal)selfrZforce_new_loopr r r __init__1szAsyncToSync.__init__c OsLz t}Wntk r YnX|r2tdtdk rFtg}nd}t}t}t |j drn|j j }nd}t }||j _ z|||||t|} |jr|jst} tdd} | |j| | } |r|| | n"|j|jj| |r||W5t |j dr|j ` |r,||j _ tdk rBt |dX|S)NznYou cannot use AsyncToSync in the same thread as an async event loop - just await the async function directly.currentrr max_workers)rrrZ is_running contextvars copy_contextr threadingcurrent_threadhasattr executorsrrr main_wrapsysexc_inforZnew_event_looprZsubmit_run_event_loopZrun_until_futureresultZcall_soon_threadsafeZ create_task) rargskwargsZ event_loopr call_result source_threadZold_current_executorZcurrent_executorrloopZ loop_executorZ loop_futurer r r __call__Dsf        zAsyncToSync.__call__cCst|z| |W5ztjdkr2t|}n tj|}|D] }|qB| tj |ddi|D]0}| rxqj| dk rj| d| |dqjt|dr| |W5|t|jXXdS)zP Runs the given event loop (designed to be called in a thread). )rZreturn_exceptionsTNz(unhandled exception during loop shutdown)message exceptiontaskshutdown_asyncgens)rZset_event_loopcloserr& version_infoZ all_tasksTaskcancelZrun_until_completeZgatherZ cancelledr3Zcall_exception_handlerr#r5)rr.coroZtasksr4r r r r(s0       zAsyncToSync._run_event_loopcCst|j|}t||jSz* Include self for methods ) functoolspartialr/update_wrapperr)rparentobjtypefuncr r r __get__szAsyncToSync.__get__c s|dk rt|dt}||j|<zzL|dr`z |dWqr|j||IdH}YqrXn|j||IdH}Wn,tk r} z|| W5d} ~ XYn X| |W5|j|=|dk rt|d<XdS)zs Wraps the awaitable with something that puts the result into the result/exception future. Nrr) rrget_current_task launch_maprr r ExceptionZ set_exceptionZ set_result) rr*r+r,r-r'r current_taskr)er r r r%s"   zAsyncToSync.main_wrapN)F) __name__ __module__ __qualname____doc__rDrr$rr/r(rBr%r r r r rs  I#rc@seZdZdZdejkr8eZe e e ejddiZ e Ze ddZdddZdd Zd d Zd d ZeddZdS)raK Utility class which turns a synchronous callable into an awaitable that runs in a threadpool. It also sets a threadlocal inside the thread so calls to AsyncToSync can escape it. If thread_sensitive is passed, the code will run in the same thread as any outer code. This is needed for underlying Python code that is not threadsafe (for example, code which handles SQLite database connections). If the outermost program is async (i.e. SyncToAsync is outermost), then this will be a dedicated single sub-thread that all sync code runs in, one after the other. If the outermost program is sync (i.e. AsyncToSync is outermost), this will just be the main thread. This is achieved by idling with a CurrentThreadExecutor while AsyncToSync is blocking its sync parent, rather than just blocking. Z ASGI_THREADSrrFcCsH||_t||||_tjj|_z |j|_Wntk rBYnXdSr) rAr<r>_thread_sensitiverZ coroutinesZ _is_coroutinerr)rrAZthread_sensitiver r r rs   zSyncToAsync.__init__c st}|jr,ttjdr$tjj}q0|j}nd}tdk rft }t j |j f||}|j }|f}i}n|j }||t j |j||t|f||}tj|ddIdH} tdk rt|| S)Nr)timeout)rrrLr#rr$rsingle_thread_executorrr r<r=rArunZrun_in_executorthread_handlerrCr&r'wait_forr) rr*r+r.executorr childrAfutureretr r r r/s>   zSyncToAsync.__call__cCst|j|Sr;)r<r=r/)rr?r@r r r rB/szSyncToAsync.__get__c Os||j_t}tj||kr&d}n||j|<d}zD|drhz |dWqv|||YWSXn|||WSW5|r|j|=XdS)zE Wraps the sync application with exception handling. FTrN)rrr!r"rrDr) rr.Z source_taskr'rAr*r+r"Z parent_setr r r rP5s  zSyncToAsync.thread_handlercCs@z$ttdrtWStjWSWntk r:YdSXdS)zs Cross-version implementation of asyncio.current_task() Returns None if there is no task. rFN)r#rrFr8rr r r r rCUs   zSyncToAsync.get_current_taskN)F)rHrIrJrKosenvironrrr.Zset_default_executorrintrDr!localrrNrr/rBrP staticmethodrCr r r r rs   + r)rZasyncio.coroutinesr<rVr&r!concurrent.futuresrrZcurrent_thread_executorrrYrr ImportErrorrrrZ sync_to_asyncZ async_to_syncr r r r s&     <