a wh=@sddlZddlZddlmZddlmZmZddlmZm Z m Z ej dkrXddlm Z n ddl m Z e dZe dZe d ZGd d d ZGd d d eZdS)N)deque)ExecutorFuture)AnyCallableTypeVar) ) ParamSpec_T_P_Rc@s>eZdZdZdeeefejejdddZ dddd Z dS) _WorkItemz Represents an item needing to be run in the executor. Copied from ThreadPoolExecutor (but it's private, so we're not going to rely on importing it) Future[_R]futurefnargskwargscOs||_||_||_||_dS)Nr)selfrrrrrY/var/www/battlenet.uz/venv/lib/python3.9/site-packages/asgiref/current_thread_executor.py__init__sz_WorkItem.__init__N)returnc Cspd}|jsdSz|j|ji|j}Wn4ty^}z|j|d}WYd}~nd}~00|j|dSNT)rZset_running_or_notify_cancelrrr BaseExceptionZ set_exceptionZ set_result)rZ__traceback_hide__resultexcrrrrun#s  z _WorkItem.run) __name__ __module__ __qualname____doc__rr r rrrrrrrrrs  rc@sPeZdZdZdddddZdddd d Zeeefej ej d d d dZ dS)CurrentThreadExecutorz An Executor that actually runs code in the thread it is instantiated in. Passed to other threads running async code, so they can run sync code in the thread they came from. zCurrentThreadExecutor | NoneN) old_executorrcCs6t|_tt|_tt|_d|_ ||_ dS)NF) threadingcurrent_thread _work_thread ConditionLock _work_readyrr _work_items_broken _old_executor)rr$rrrr8s   zCurrentThreadExecutor.__init__ Future[Any]rrcstjkrtddddfdd }||jFjsTjsTjq<jshWdqj }Wdn1s0Y| ~q4dS)z Runs the code in the work queue until a result is available from the future. Should be run from the thread the executor is initialised in. z.done) r%r&r' RuntimeErrorZadd_done_callbackr*r+r,waitpopleftr)rrr2 work_itemrr1rrun_until_future?s   (z&CurrentThreadExecutor.run_until_futurer)rrrrcOst|jkrtdt}t||g|Ri|}|}|j:|jsj|j ||j WdqWdn1s~0Y|j durtd|j }q8|S)Nz@You cannot submit onto CurrentThreadExecutor from its own threadz/CurrentThreadExecutor already quit or is broken) r%r&r'r3rrr*r,r+appendr0r-)rrrrfr6executorrrrsubmit]s   , zCurrentThreadExecutor.submit) rr r!r"rr7rr r rrr;rrrrr#1s r#)sysr% collectionsrconcurrent.futuresrrtypingrrr version_infor Ztyping_extensionsr r r rr#rrrrs