装饰器的Python 3类型提示

Question

考虑以下代码：

from typing import Callable, Any

TFunc = Callable[..., Any]

def get_authenticated_user(): return "John"

def require_auth() -> Callable[TFunc, TFunc]:
    def decorator(func: TFunc) -> TFunc:
        def wrapper(*args, **kwargs) -> Any:
            user = get_authenticated_user()
            if user is None:
                raise Exception("Don't!")
            return func(*args, **kwargs)
        return wrapper
    return decorator

@require_auth()
def foo(a: int) -> bool:
    return bool(a % 2)

foo(2)      # Type check OK
foo("no!")  # Type check failing as intended

这段代码按预期工作。现在想象一下我想扩展它，而不仅仅是执行func(*args, **kwargs)我想在参数中注入用户名。因此，我修改了功能签名。

from typing import Callable, Any

TFunc = Callable[..., Any]

def get_authenticated_user(): return "John"

def inject_user() -> Callable[TFunc, TFunc]:
    def decorator(func: TFunc) -> TFunc:
        def wrapper(*args, **kwargs) -> Any:
            user = get_authenticated_user()
            if user is None:
                raise Exception("Don't!")
            return func(*args, user, **kwargs)  # <- call signature modified

        return wrapper

    return decorator


@inject_user()
def foo(a: int, username: str) -> bool:
    print(username)
    return bool(a % 2)


foo(2)      # Type check OK
foo("no!")  # Type check OK <---- UNEXPECTED

我找不到正确的输入方式。我知道在此示例中，修饰函数和返回函数在技术上应具有相同的签名（但即使未检测到）。

Answer 1

您不能Callable说任何其他参数。它们不是通用的。您唯一的选择是说您的装饰器采用，Callable并且Callable返回不同的值。

您 可以 使用typevar来确定返回类型：

RT = TypeVar('RT')  # return type

def inject_user() -> Callable[[Callable[..., RT]], Callable[..., RT]]:
    def decorator(func: Callable[..., RT]) -> Callable[..., RT]:
        def wrapper(*args, **kwargs) -> RT:
            # ...

即使这样，最终的修饰foo()函数也具有def (*Any, **Any) -> builtins.bool*您使用时的键入签名reveal_type()。

当前正在讨论各种提案以Callable提高灵活性，但这些提案尚未实现。看到

• 允许可变参数的仿制药
• 建议：泛化Callable以便能够指定参数名称和种类
• TypeVar表示Callable的参数
• 出色地支持功能装饰器

对于一些例子。该列表中的最后一个是一张总括票，其中包括您的特定用例，用于更改可调用签名的装饰器：

混乱与返回类型或参数

对于任意函数，您根本无法执行此操作-甚至没有语法。这是我组成的一些语法。