Python 3.10类方法装饰器类型提示：mypy报错及兼容方案咨询

问题核心 #

你需要实现一个仅允许用于Foo类实例方法的装饰器，要求通过类型系统严格约束，避免被误用于自由函数；同时解决Concatenate结合ParamSpec时mypy报错的问题。

正确实现代码 #

from typing import Callable, ParamSpec, TypeVar, Concatenate, cast

# 定义参数规格和返回值类型变量
P = ParamSpec("P")
R = TypeVar("R")

class Foo:
    def __init__(self, flags: set[str]) -> None:
        self.flags = flags

def require_flag(flag: str) -> Callable[[Callable[Concatenate[Foo, P], R]], Callable[P, R]]:
    def decorator(func: Callable[Concatenate[Foo, P], R]) -> Callable[P, R]:
        def wrapper(*args: P.args, **kwargs: P.kwargs) -> R:
            # 提取类方法的self实例，类型已被约束为Foo
            self_instance = cast(Foo, args[0])
            if flag in self_instance.flags:
                # 调用原始方法，传入self和剩余参数
                return func(self_instance, *args[1:], **kwargs)
            raise ValueError(f"Missing required flag: {flag}")
        return wrapper
    return decorator

# 合法用法：Foo类的实例方法
class Foo:
    def __init__(self, flags: set[str]) -> None:
        self.flags = flags

    @require_flag("admin")
    def admin_action(self, param: str) -> None:
        print(f"Admin action executed with param: {param}")

# 非法用法：自由函数（会触发mypy类型错误）
@require_flag("admin")
def free_func(param: str) -> None:
    print(f"Free func called: {param}")

关键解释 #

1. 类型约束逻辑

   • 装饰器的输入类型Callable[Concatenate[Foo, P], R]明确要求被装饰函数的第一个参数必须是Foo实例，对应类方法的self参数。
   • 装饰器的返回类型Callable[P, R]匹配类方法的调用签名（调用时无需显式传入self），解决了原写法中mypy报错的参数不兼容问题。

2. 为什么原写法报错
   原代码中仅定义了Wrappable类型，但装饰器的返回值没有匹配类方法的调用签名。类方法的实际调用签名是Callable[P, R]（隐式传入self），而原始函数签名是Callable[Concatenate[Foo, P], R]，装饰器需要完成这两种签名的转换。

3. 严格约束的生效效果
   当尝试将装饰器用于自由函数时，mypy会抛出类型错误，提示被装饰函数的参数列表不符合Callable[Concatenate[Foo, P], R]的要求，完全实现了你需要的严格约束。

对比宽松写法的优势 #

将Wrappable改为Callable[P, None]确实能通过类型检查，但会丢失对self参数的类型约束，导致自由函数也能被装饰。而上述写法既保证了类型安全，又严格限制了装饰器的适用范围。

内容的提问来源于stack exchange，提问作者Coxy