Параметризация параметризованного типа в Python 3.8 ⇐ Python

[Anonymous]

Я пытаюсь создать фабрику декораторов, базовая структура которой не вызывает особого удивления:

Код: Выделить всё

def make_wrapped_awaitable_or_aiter(
awaitable_or_aiter, span_wrapper, ContextManager]
) -> AwaitableOrAIterT:
if isinstance(awaitable_or_aiter, AsyncIterator):
async def inner_iterable():
with span_wrapper():
async for v in awaitable_or_aiter:
yield v
return inner_iterable()
else:
async def inner_awaitable():
with span_wrapper():
return await awaitable_or_aiter

return inner_awaitable()

def decorator_factory(decorator_factory_arg1):
def decorator(original_func):
def wrapped(*args, **kwargs):
return awaitable_or_aiter(original_func(*args, **kwargs), some_contextmanager)
return wrapped
return decorator

@decorator_factory('foo')
def foobar('anything'):
...

Основной момент здесь в том, что original_func всегда будет либо асинхронной функцией, либо асинхронной итерацией.
Я хотел бы ввести его на Python 3.8, чтобы mypy понимал, что тип возвращаемого значения обернутой функции будет таким же, как тип возвращаемого значения original_func, и что он также будет иметь тот же тип возвращаемого значения, который можно вызвать асинхронно. На данный момент у меня это есть:

Код: Выделить всё

AwaitableOrAIterT = TypeVar("AwaitableOrAIterT", Awaitable, AsyncIterator)

def make_wrapped_awaitable_or_aiter(
awaitable_or_aiter: AwaitableOrAIterT, span_wrapper: Callable[[], ContextManager]
) -> AwaitableOrAIterT:
if isinstance(awaitable_or_aiter, AsyncIterator):
async def inner_iterable():
with span_wrapper():
async for v in awaitable_or_aiter:
yield v
return inner_iterable()
else:
async def inner_awaitable():
with span_wrapper():
return await awaitable_or_aiter
return inner_awaitable()

def decorator_factory(decorator_factory_arg1: str) -> Callable[[Callable[..., AwaitableOrAIterT]], Callable[..., AwaitableOrAIterT]]:
def decorator(original_func: Callable[..., AwaitableOrAIterT]):
def wrapped(*args, **kwargs) -> AwaitableOrAIterT:
make_wrapped_awaitable_or_aiter(original_func(*args, **kwargs), some_contextmanager)
return wrapped
return decorator

@decorator_factory('foo')
async def foobar(some_arg: str) -> bool:
return await some_query_typed_to_bool()

Это позволяет хинтеру узнать, что асинхронная конструкция «original_func» — это то, что он ожидает (т. е. он выводит Awaitable, и это что-то, что вызывает асинхронность с помощью foobar()< /code> неверен, и это также означает, что попытка декорировать неасинхронную функцию неверна), но, похоже, это специально выводит Awaitable[Unknown], а не Awaitable, и я не совсем уверен, как это сделать иначе (вероятно, потому, что в make_wrapped_awaitable_or_aiter нет никакой информации о типе возвращаемого значения из Internal_*).
По сути, я считаю, что мне нужен способ параметризовать AwaitableOrAIterT как универсальный (используя возвращаемый тип типа Awaitable или AsyncIterator), чтобы я мог комментировать

Код: Выделить всё

def make_wrapped_awaitable_or_aiter(
awaitable_or_aiter: AwaitableOrAIterT[T], span_wrapper: Callable[[], ContextManager]
) -> AwaitableOrAIterT[T]:
if isinstance(awaitable_or_aiter, AsyncIterator):
async def inner_iterable() -> T:
with span_wrapper():
async for v in awaitable_or_aiter:
yield v
return inner_iterable()
..etc..

но я не знаю, как это сделано в 3.8.


Подробнее здесь: https://stackoverflow.com/questions/735 ... python-3-8