如何让静态分析器（如VS Code IntelliSense）识别工厂创建的类？

问题根源在于Python静态分析器（如Pylance）不允许将变量直接用作类型表达式，且动态生成的类无法被自动识别为独立的类型实体。以下两种方案可以在避免手动编写大量重复类的同时，让静态分析器提供完整的类型提示和代码补全：

方案一：利用TYPE_CHECKING区分静态分析与运行时逻辑 #

核心思路是给静态分析器提供“假”的手动定义类，运行时则使用工厂生成的类，兼顾代码复用和类型提示：

from abc import ABC, abstractmethod
from typing import Type, TYPE_CHECKING

class ExampleClass(ABC):
    @abstractmethod
    def get(self) -> str:
        pass

# 仅静态分析器会执行此分支，运行时跳过
if TYPE_CHECKING:
    # 手动定义与工厂类行为一致的静态分析专用类
    class ASubclass(ExampleClass):
        def get(self) -> str:
            return "A"
    
    class BSubclass(ExampleClass):
        def get(self) -> str:
            return "B"
else:
    # 运行时使用工厂函数生成类
    def class_factory(s: str) -> Type[ExampleClass]:
        class SpecificClass(ExampleClass):
            def get(self) -> str:
                return s
        return SpecificClass

    ASubclass = class_factory("A")
    BSubclass = class_factory("B")

def f(x: ASubclass):
    # 输入x.会自动提示get()，悬停x会显示类型为ASubclass
    x.get()

说明 #

TYPE_CHECKING是typing模块的常量，仅在静态分析阶段为True，运行时为False。这样静态分析器会读取手动定义的子类，提供完整的类型提示；实际运行时则使用工厂生成的类，避免重复编码。

方案二：泛型+字面量类型实现严格类型约束 #

如果需要让静态分析器识别子类方法的返回值（比如ASubclass.get()返回的是"A"而非任意字符串），可以用泛型结合字面量类型：

from abc import ABC, abstractmethod
from typing import TypeVar, Type, Generic, Literal, TypeAlias

# 定义绑定到字符串的类型变量
T = TypeVar('T', bound=str)

class ExampleClass(ABC, Generic[T]):
    @abstractmethod
    def get(self) -> T:
        pass

def class_factory(s: T) -> Type[ExampleClass[T]]:
    class SpecificClass(ExampleClass[T]):
        def get(self) -> T:
            return s
    return SpecificClass

# 生成工厂类并定义类型别名
ASubclass = class_factory("A")
ASubclassType: TypeAlias = ExampleClass[Literal["A"]]

BSubclass = class_factory("B")
BSubclassType: TypeAlias = ExampleClass[Literal["B"]]

def f(x: ASubclassType):
    # 悬停x.get()会显示返回类型为Literal["A"]，同时有代码提示
    x.get()

说明 #

通过泛型ExampleClass[T]，每个工厂生成的类会关联一个独特的字面量类型。使用TypeAlias可以将复杂的泛型类型简化为简洁的别名，保持类型注解的可读性。

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