Nicer asyncio.to_thread

to_thread2

from typing import ParamSpec, TypeVar
from collections.abc import Callable, Coroutine
import functools
P = ParamSpec('P')
R = TypeVar('R')

def to_thread2(fn: Callable[P, R]) -> Callable[P, Coroutine[Any, Any, R]]:
    functools.wraps(fn)
    async def wrapper(*args: P.args, **kwargs: P.kwargs) -> R:
        return await asyncio.to_thread(fn, *args, **kwargs)
    return wrapper

usage.py

def a(a: str) -> str:
    return a+" in thread"

async def x():
    x = to_thread2(a)
    await x('b')
    
    await to_thread2(a)('b')