创建具有自引用类型提示的递归数据类

Question

我想用 Python 编写一个数据类定义，但不能在声明中引用同一个类。

我主要要实现的是这个嵌套结构的类型化，如下图：

 @dataclass
 class Category:
     title: str
     children: [Category] # I can't refer to a "Category"
  
 tree = Category(title='title 1', children=[
     Category('title 11', children=[]),
     Category('title 12', children=[])
 ])

Answer 1

选项 #1

您可以将类名包装在一个字符串中，以便转发声明注解：

from dataclasses import dataclass
from typing import List


@dataclass
class Category:
    title: str
    children: List['Category']

选项 #2

您可以包含__future__ 导入，以便默认情况下所有注释向前声明，如下所示。在这种情况下，您还可以在 Python 3.7 及更高版本中消除 typing 导入并使用新样式的注释。

from __future__ import annotations

from dataclasses import dataclass


@dataclass
class Category:
    title: str
    children: list[Category]