Python 3.5以上版本中的递归输入

Question

在Python 3.5中，添加了类型注释（请参见此处）。

有没有一种方法来定义递归类型注释，例如用于树状结构？

class Employee(object):
    def __init__(self, name: str, reports: List[Employee]):
       self.name = name
       self.reports = reports

在上面，注释似乎无效List[Employee]。运行代码会导致以下错误：

NameError: name 'Employee' is not defined

Answer 1

您可以使用PEP 484中定义的正向引用

通常发生这种情况的情况是容器类的定义，其中定义的类出现在某些方法的签名中。例如，以下代码（简单的二进制树实现的开始）不起作用：

class Tree:
    def __init__(self, left: Tree, right: Tree):
        self.left = left
        self.right = right

为了解决这个问题，我们写：

class Tree:
    def __init__(self, left: 'Tree', right: 'Tree'):
        self.left = left
        self.right = right

可以将字符串文字用作类型提示的一部分，例如：

class Tree:
    ...
    def leaves(self) -> List['Tree']: