如何为泛型Wrapper类的__init__添加类型注解以实现静态类型检查？

要实现Wrapper初始化时参数的静态类型检查与提示，核心是利用Python 3.10+支持的PEP 612 ParamSpec特性，它能精准捕获类__init__方法的参数签名。

实现步骤 #

• 导入TypeVar、Generic、ParamSpec等类型工具
• 定义参数规格变量P，用于捕获目标类的__init__参数
• 让Wrapper类同时绑定类型T和参数规格P
• 约束__init__方法的参数与P完全匹配，确保静态检查生效

完整代码示例 #

from typing import TypeVar, Generic, ParamSpec

T = TypeVar('T')
P = ParamSpec('P')

class Wrapper(Generic[T, P]):
    def __init__(self, *args: P.args, **kwargs: P.kwargs):
        self.args = args
        self.kwargs = kwargs

    def __call__(self) -> T:
        target_cls = self.__orig_class__.__args__[0]
        return target_cls(*self.args, **self.kwargs)

# 使用示例
class SomeClass:
    def __init__(self, name: str, age: int = 18):
        self.name = name
        self.age = age

# 静态检查器会自动校验参数类型、必填项
wrapped = Wrapper[SomeClass]("Alice", age=20)
instance = wrapped()

关键说明 #

• ParamSpec的作用：P会完整捕获目标类__init__的参数签名，包括位置参数、关键字参数及默认值，让mypy、pyright等静态检查工具能准确校验Wrapper初始化时的参数是否合规。
• 自动类型推断：主流编辑器（如VS Code + Pylance）支持自动推断P的类型，无需手动指定参数列表，直接写Wrapper[SomeClass]就能获得实时参数提示。
• 兼容第三方类：无论T是自定义类还是第三方库类，只要其__init__有类型注解，静态检查就能生效；即使无注解，部分工具也能通过运行时信息做基础推断。

旧Python版本兼容方案 #

若使用Python 3.9及以下版本，可通过typing_extensions库导入ParamSpec实现相同效果；若完全无法使用ParamSpec，可通过Callable约束构造函数类型：

from typing import TypeVar, Generic, Callable
from typing_extensions import ParamSpec

T = TypeVar('T')
P = ParamSpec('P')

class Wrapper(Generic[T]):
    def __init__(self, *, constructor: Callable[P, T], **kwargs: P.kwargs):
        self.constructor = constructor
        self.kwargs = kwargs

    def __call__(self) -> T:
        return self.constructor(**self.kwargs)

# 使用示例
wrapped = Wrapper[SomeClass](constructor=SomeClass, name="Charlie")

这种方式需显式传入构造函数，但同样能实现静态参数检查。

内容的提问来源于stack exchange，提问作者Jiří Kubica