1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
|
import sys
import asyncio
import inspect
import types
from functools import singledispatch, wraps
from typing import Any, Callable, Generator
PY37 = sys.version_info >= (3, 7)
def _is_awaitable(co: Generator[Any, None, Any]) -> bool:
if PY37:
return inspect.isawaitable(co)
else:
return (isinstance(co, types.GeneratorType) or
isinstance(co, asyncio.Future))
@singledispatch
def sync(co: Any) -> Any:
raise TypeError(f'Called with unsupported argument: {co}')
@sync.register(asyncio.Future)
@sync.register(types.GeneratorType)
def sync_co(co: Generator[Any, None, Any]) -> Any:
if not _is_awaitable(co):
raise TypeError(f'Called with unsupported argument: {co}')
return asyncio.get_event_loop().run_until_complete(co)
@sync.register(types.FunctionType)
@sync.register(types.MethodType) # type: ignore
def sync_fu(f: Callable[..., Any]) -> Callable[..., Any]:
if not asyncio.iscoroutinefunction(f):
raise TypeError(f'Called with unsupported argument: {f}')
@wraps(f)
def run(*args, **kwargs) -> Any:
return asyncio.get_event_loop().run_until_complete(f(*args, **kwargs))
return run
if PY37:
sync.register(types.CoroutineType)(sync_co)
|
