Python泛型类DAGNode__init__方法依赖参数类型提示问题解决

一、与__init__()内设置类实例等价的实现方式 #

Python里有几种和__init__初始化实例等价的写法，核心都是完成实例属性的设置：

• 使用__new__方法：__new__是负责创建实例的方法，若需在实例创建阶段就完成属性赋值，可重写该方法：

class MyClass:
    def __new__(cls, id, dependencies):
        instance = super().__new__(cls)
        instance.id = id
        instance.dependencies = dependencies
        return instance

• 工厂类方法：封装实例创建逻辑到类方法中，对外提供更直观的实例入口：

class MyClass:
    def __init__(self, id, dependencies):
        self.id = id
        self.dependencies = dependencies
    
    @classmethod
    def create(cls, id, dependencies):
        return cls(id, dependencies)

• dataclasses装饰器：Python 3.7+自带的dataclasses可自动生成__init__方法，无需手动编写属性赋值代码：

from dataclasses import dataclass

@dataclass
class MyClass:
    id: int
    dependencies: set

二、解决DAGNode[T]类型未定义的报错问题 #

你遇到的是前向引用问题——类定义过程中，自身尚未完全初始化，直接引用DAGNode[T]会触发报错，以下三种方法可解决：

1. 字符串字面量类型提示（Python 3.7+） #

将类型用引号包裹，Python类型检查器（如mypy）可正常识别：

from typing import TypeVar, Generic

T = TypeVar('T')
class DAGNode(Generic[T]):
    def __init__(self, id: T, dependencies: set['DAGNode[T]']):  
        self.id = id
        self.dependencies = dependencies

2. 导入__future__.annotations（Python 3.7+） #

在文件顶部导入该特性后，所有类型提示会被自动当作字符串处理，无需手动加引号：

from __future__ import annotations
from typing import TypeVar, Generic

T = TypeVar('T')
class DAGNode(Generic[T]):
    def __init__(self, id: T, dependencies: set[DAGNode[T]]):  
        self.id = id
        self.dependencies = dependencies

3. 使用typing.Self（Python 3.11+） #

若你的Python版本为3.11及以上，可直接用Self指代当前类的实例，写法更简洁：

from typing import TypeVar, Generic, Self

T = TypeVar('T')
class DAGNode(Generic[T]):
    def __init__(self, id: T, dependencies: set[Self]):  
        self.id = id
        self.dependencies = dependencies

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