o ߥi(@sddlZddlZddlmZddlmZddlmZm Z m Z m Z ddgZ e dZ GdddeZd e d e fd ed ed e fddZdS)N)ThreadPoolExecutor) copy_context)AnyCallableTypeVarcastrrun_in_threadpoolTcs eZdZdZfddZZS)rz\ Thread pool with ContextVars - https://github.com/python/cpython/issues/78195 cs&tjttj|g|Ri|S)N)supersubmit functoolspartialrrun)self_ThreadPoolExecutor__fnargskwargs __class__E/home/ubuntu/.local/lib/python3.10/site-packages/baize/concurrency.pyr szThreadPoolExecutor.submit)__name__ __module__ __qualname____doc__r __classcell__rrrrrs__fn.rrreturncsLt}t}tj|j|g|Ri|}|dttgt f|IdHS)a Asynchronously run function fn in a separate thread. Any *args and **kwargs supplied for this function are directly passed to fn. Also, the current :class:`contextvars.Context` is propogated, allowing context variables from the main thread to be accessed in the separate thread. https://github.com/python/cpython/blob/0f56263e62ba91d0baae40fb98947a3a98034a73/Lib/asyncio/threads.py N) asyncioget_running_looprr r rrun_in_executorrrr )rrrloopctx func_callrrrrs   )rr concurrent.futuresr_ThreadPoolExecutor contextvarsrtypingrrrr__all__r rrrrrs  &