为正在定义的类型的对象键入Hinting

Question

我收到错误:

NameError: name 'OrgUnit' is not defined

class OrgUnit(object):

    def __init__(self,
                 an_org_name: str,
                 its_parent_org_unit: OrgUnit= None
                 ):
        self.org_unit_name = an_org_name
        self.parent_org_unit = its_parent_org_unit

    def __str__(self):
        if self.parent_org_unit:
            parent_org_unit_name = self.parent_org_unit.__str__()
            return parent_org_unit_name + "->" + self.org_unit_name
        else:
            return self.org_unit_name


if __name__ == '__main__':
    ibm_worldwide = OrgUnit("IBM_Worldwide")
    ibm_usa = OrgUnit("IBM_USA", ibm_worldwide)
    ibm_asia = OrgUnit("IBM_Asia", ibm_worldwide)
    ibm_china = OrgUnit("IBM_China", ibm_asia)
    print(ibm_worldwide)
    print(ibm_usa)
    print(ibm_asia)
    print(ibm_china)

我确信这是一个已知的范例,因为它似乎是一个非常常见的分层类使用问题(自引用类).我知道我可以改变的类型its_parent_org_unit是object和它的作品,但是这似乎是错误的做法,主要是因为它泄水我检查类型的我的电话的能力.随着its_parent_org_unit改变是一种类型的object我得到正确的结果:

IBM_Worldwide
IBM_Worldwide->IBM_USA
IBM_Worldwide->IBM_Asia
IBM_Worldwide->IBM_Asia->IBM_China

我对这些想法和建议持开放态度.做这种事情最"pythonic"的方法是什么？

PS:这种"自引用类"范例/问题的名称是什么,我可以用来查找其他建议？

Answer 1

您的问题是您希望使用类型提示,但您希望此类本身能够获取其自己类型的参数.

类型提示PEP(0484)解释了您可以使用类型名称的字符串版本作为前向引用.这个例子的Tree数据结构听起来非常类似于这个OrgUnit.

例如,这有效:

class OrgUnit(object):

    def __init__(self,
                 an_org_name: str,
                 its_parent_org_unit: 'OrgUnit' = None
                 ):

在Python 3.7,你将能够激活注释推迟评价用from __future__ import annotations.这将自动将注释存储为字符串而不是评估它们,因此您可以这样做

from __future__ import annotations

class OrgUnit(object):
    def __init__(self,
                 an_org_name: str,
                 its_parent_org_unit: OrgUnit= None
                 ):
        ...

这计划成为Python 4.0中的默认设置.