如何为Python的factory_boy Factory类添加静态类型提示？

当然可以给factory_boy的工厂类添加静态类型提示，下面是几种实用的实现方式：

1. 使用泛型基类定义类型约束 #

通过typing.Generic和TypeVar创建泛型Factory基类，让子类明确指定生成的模型类型，所有工厂方法会自动获得正确的类型提示：

from typing import TypeVar, Generic
from factory import Factory

T = TypeVar('T')

class TypedFactory(Factory, Generic[T]):
    class Meta:
        abstract = True  # 标记为抽象基类，不会被实例化

    @classmethod
    def create(cls, **kwargs) -> T:
        return super().create(**kwargs)
    
    @classmethod
    def build(cls, **kwargs) -> T:
        return super().build(**kwargs)

修改你的TableFactory继承这个泛型基类：

class TableFactory(TypedFactory[Table]):
    class Meta:
        model = Table

    id = LazyAttribute(lambda x: uuid4())
    color = SubFactory(ColorFactory)
    reference = 'WT9034'
    weight = 120
    height = 50
    length = 3
    width = 1 

调用TableFactory.create()时，类型检查工具（如mypy、PyCharm）会自动识别返回值为Table类型。

2. 直接给工厂方法添加返回类型注解 #

如果不想创建泛型基类，可直接给工厂的create和build方法添加类型注解：

class TableFactory(Factory):
    class Meta:
        model = Table

    id = LazyAttribute(lambda x: uuid4())
    color = SubFactory(ColorFactory)
    reference = 'WT9034'
    weight = 120
    height = 50
    length = 3
    width = 1 

    @classmethod
    def create(cls, **kwargs) -> Table:
        return super().create(**kwargs)
    
    @classmethod
    def build(cls, **kwargs) -> Table:
        return super().build(**kwargs)

这种方式更直接，适合仅给单个工厂类添加类型提示的场景。

3. 利用factory_boy内置泛型支持（Python 3.10+） #

从factory_boy 3.2.0版本开始，官方已支持泛型类型提示，可直接在继承Factory时指定模型类型：

from factory import Factory

class TableFactory(Factory[Table]):
    class Meta:
        model = Table

    # 字段定义保持不变

这种方式最简洁，无需自行实现泛型基类，直接利用库的内置支持。

注意事项 #

• 确保类型检查工具（如mypy）已正确配置，开启对泛型和类型注解的支持。
• 若使用SQLAlchemy模型，需确保模型类本身有正确的字段类型注解，这样工厂生成实例的类型提示会更准确。

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