如何使用类型提示在生成器上注释装饰器？

Question

我正在使用作为协同程序的生成器，正如 David Beazley（http://www.dabeaz.com/coroutines/）的精彩演讲中所描述的那样，我不知道如何键入装饰器 consumer。到目前为止，这是我所拥有的：

from typing import Any, Callable, Generator, Iterable

ArbGenerator = Generator[Any, Any, Any]

def consumer(fn: ❓) -> ❓:
    @wraps(fn)
    def start(*args: Any) -> ArbGenerator:
        c = fn(*args)
        c.send(None)
        return c
return start

使用示例，有点删节：

@consumer
def identity(target: ArbGenerator) -> ArbGenerator:
    while True:
        item = yield
        target.send(item)

@consumer
def logeach(label: Any, target: ArbGenerator) -> ArbGenerator:
    while True:
        item = yield
        print(label, item)
        target.send(item)

pipeline = identity(logeach("EXAMPLE", some_coroutine_sink()))

粗体 ❓ 标记我不确定的地方 - 我也不确定我定义的类型 ArbGenerator。 （问题是，如果没有（装饰器）函数consumer 本身键入，我不确定mypy 是否正在使用该装饰器分析任何生成器函数，这就是为什么我不确定ArbGenerator。）

我对最紧凑的类型感兴趣，比Any 更好，这样当我编写这些协程的链时，mypy 会在链设置不正确时给我很好的警告。

（如果重要，Python 3.5。）

Answer 1

作为一种更具体的方式，您可以做以下几件事：

1. 使用Callable 类型而不是问号。

2. 将typing.Coroutine 用于targets 并删除ArbGenerator。

3. 协程返回一个生成器，返回类型可以是Generator 或其超类型之一

你应该使用 callable 而不是问号的原因是 fn 一开始应该是一个可调用对象，这就是你用装饰器包装它的原因。 Coroutine 将在调用对象后创建，返回类型是/应该显然也是一个可调用对象。

from typing import Any, Callable,Generator, Coroutine
from functools import wraps


def consumer(fn: Callable) -> Callable:
    @wraps(fn)
    def start(*args: Any) -> Coroutine:
        c = fn(*args)  # type: Coroutine
        c.send(None)
        return c
    return start


@consumer
def identity(target: Coroutine) -> Generator:
    while True:
        item = yield
        target.send(item)

@consumer
def logeach(label: Any, target: Coroutine) -> Generator:
    while True:
        item = yield
        print(label, item)
        target.send(item)

注意：正如文档中也提到的，如果您想使用更精确的语法来注释生成器类型，可以使用以下语法：

Generator[YieldType, SendType, ReturnType]

阅读更多：https://docs.python.org/3/library/typing.html#typing.Generator