如何在类中使用初始化传入的模型作为方法的类型注解？

直接用self.model做类型注解行不通，因为类型注解是在类定义阶段解析的，这时候self还没实例化出来。下面是几种可行的实现方式：

1. 泛型类方案（推荐，支持静态类型检查） #

通过泛型绑定传入的Model类型，让类型检查工具能直接识别参数类型：

from typing import Generic, TypeVar
from pydantic import BaseModel

# 定义一个绑定BaseModel的类型变量
T = TypeVar('T', bound=BaseModel)

class A(Generic[T]):
    def __init__(self, model: type[T]):
        self.model = model
    
    def func(self, x: T):
        # 这里x会被静态类型检查器识别为传入的model类型
        pass

实例化时传入具体的Pydantic模型，比如：

class MyModel(BaseModel):
    name: str

a = A(MyModel)
a.func(MyModel(name="test"))  # 类型检查通过
a.func({"name": "test"})      # 类型检查工具会提示错误

2. 运行时动态校验方案（适合不需要静态检查的场景） #

如果不需要静态类型检查，只需要运行时确保参数符合model类型，可以用Pydantic的验证方法，配合动态修改注解：

from pydantic import BaseModel, validate_instance

class A:
    def __init__(self, model: type[BaseModel]):
        self.model = model
        # 动态修改func的参数注解
        self.func.__annotations__['x'] = model
    
    def func(self, x):
        # 运行时验证x是否符合model类型
        validate_instance(x, self.model)
        pass

3. Python 3.11+ 用Self延迟注解 #

Python 3.11引入了Self类型，可以用字符串形式的注解延迟解析：

from typing import Self
from pydantic import BaseModel

class A:
    def __init__(self, model: type[BaseModel]):
        self.model = model
    
    def func(self, x: 'Self.model'):
        pass

这种方式需要你的类型检查工具（比如Pyright）支持解析Self的属性，mypy目前对这种写法的支持有限。

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