类函数装饰器的正确类型

Question

键入print_before 函数装饰器的正确方法是什么，以便 Wrapped 函数具有正确的类型，但我不能在不起作用的类上使用装饰器？

谢谢

def print_before(func):
    def func_wrapper(self, *args):
        self.print_hi()
        return func(self, *args)
    return func_wrapper


class PrintThings:
    def print_hi(self):
        print("hi")

    @print_before
    def add_nums(self, a: int, b: int) -> int:
        return a + b


pt = PrintThings()
pt.add_nums(5, 4)


class ShouldNotWork:
    @print_before
    def add_nums(self, a: int, b: int) -> int:
        return a + b


snw = ShouldNotWork()
snw.add_nums(4, 5)

Answer 1

装饰器有一个非常通用的类型。您对包装器的唯一了解是它至少接受一个未知类型的参数，并返回一些未知类型。至于装饰器本身，你只知道它返回的东西与它的参数类型相同。

实际上，我们还知道一件事。无论self 的类型是什么，它都必须有一个print_hi 方法。我们可以使用Protocol 来表示。

我们还引入了一个类型变量T，每次在print_before的签名中使用它代表同一个类（或其子类）。

from typing import Any, Tuple, Protocol, TypeVar

class PrintsHi(Protocol):
    def print_hi(self):
        pass


T = TypeVar('T', bound=PrintsHi)


def print_before(func: Callable[[T, Tuple[Any,...]], Any]) -> Callable[[T, Tuple[Any,...]], Any]:
    def func_wrapper(self: T, *args: Tuple[Any,...]):
        self.print_hi()
        return func(self, *args)
    return func_wrapper