如何为使用工厂方法创建的类属性添加类型提示

在Python面向对象开发中，工厂方法常用于封装对象创建逻辑，动态生成类实例并赋值给类属性。但这种场景下，类属性的类型提示往往无法被IDE正确识别，影响开发效率和代码健壮性。下面介绍几种可行的解决方案。

方案一：使用类型注解直接标注

如果工厂方法的返回类型固定，可以直接在类属性定义处添加类型注解，明确标注属性的类型。这种方式适用于工厂方法返回类型单一的场景。

from typing import Optional

class Product:
    def use(self) -> str:
        return "产品使用中"

class Factory:
    @staticmethod
    def create_product() -> Product:
        return Product()

class MyClass:
    # 直接标注类属性类型为Product
    product: Product = Factory.create_product()
    # 如果属性可能为None，可以使用Optional
    optional_product: Optional[Product] = None

# 测试类型提示是否生效
obj = MyClass()
# IDE会识别obj.product为Product类型，可自动补全use方法
print(obj.product.use())

方案二：使用TypeVar标注多态返回场景

当工厂方法会根据不同参数返回不同的子类实例时，可以使用TypeVar来标注类型范围，让类型提示覆盖所有可能的返回类型。

from typing import TypeVar

class ProductA:
    def func_a(self) -> str:
        return "产品A功能"

class ProductB:
    def func_b(self) -> str:
        return "产品B功能"

# 定义类型变量，限定为ProductA或ProductB的子类
ProductType = TypeVar("ProductType", ProductA, ProductB)

class Factory:
    @staticmethod
    def create_product(product_type: str) -> ProductType:
        if product_type == "a":
            return ProductA()
        else:
            return ProductB()

class MyClass:
    # 标注类属性为ProductType类型
    product: ProductType = Factory.create_product("a")

obj = MyClass()
# 类型提示会根据工厂方法的返回逻辑给出对应子类的补全

方案三：使用__init_subclass__动态添加类型提示

如果类属性的赋值逻辑需要在类定义阶段动态完成，可以使用__init_subclass__方法，在子类初始化时动态设置属性并补充类型注解。

from typing import Any, ClassVar

class Product:
    pass

class Factory:
    @staticmethod
    def create_product() -> Product:
        return Product()

class BaseClass:
    def __init_subclass__(cls, **kwargs: Any) -> None:
        # 动态为子类添加类属性并标注类型
        cls.product: ClassVar[Product] = Factory.create_product()
        super().__init_subclass__(**kwargs)

class MyClass(BaseClass):
    pass

obj = MyClass()
# 此时obj.product会被正确识别为Product类型
print(type(obj.product))

方案对比

不同方案的适用场景和特点如下：

注意事项

• 类型提示仅为IDE和类型检查工具提供参考，不会在运行时强制校验类型，实际开发中仍需保证工厂方法的返回类型符合标注。
• 如果使用Python 3.10及以上版本，可以使用|语法替代Optional和Union，例如product: Product | None。
• 对于复杂的工厂方法逻辑，建议配合typing.cast在必要时进行类型转换，避免类型检查报错。

工厂方法类型提示类属性Python类型注解修改时间：2026-07-08 08:48:26