我想用 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=[])
])
您可以将类名包装在字符串中,以便前向声明注释:
from dataclasses import dataclass
from typing import List
@dataclass
class Category:
title: str
children: List['Category']
您可以包含
__future__
导入,以便默认情况下所有注释都是 前向声明,如下所示。在这种情况下,您还可以消除 typing
导入并在 Python 3.7 及更高版本中使用新样式注释。
from __future__ import annotations
from dataclasses import dataclass
@dataclass
class Category:
title: str
children: list[Category]