python方法参数注解类

Question

以下内容会导致导入错误，因为 Foo 尚未定义:

class Foo:
    def __init__(self, rhs: Foo):
        pass

有没有办法注释 rhs 以指示它应该是 Foo 的另一个实例？

Answer 1

不是真的。 Python 无法像 C++ 那样在定义类之前声明它。

如果您只是想让人们看到 rhs 应该是 Foo，那么您始终可以使用字符串文字:

def __init__(self, rhs: 'Foo'):
# or
def __init__(self, rhs: "<class '__main__.Foo'>"):

这使您的意图非常清晰，并且还允许您让注释说出您想要的任何内容。

但是，如果您想要在 Foo.__init__ 的 __annotations__ 属性内真正引用 Foo 类，那么您将需要手动定义类后更改此属性:

class Foo:
    def __init__(self, rhs):
        pass

Foo.__init__.__annotations__['rhs'] = Foo
print(Foo.__init__.__annotations__)
# {'rhs': <class '__main__.Foo'>}

但我个人只会使用第一个解决方案。函数注释的主要目的是记录您的函数。因此，执行此操作的字符串文字可以满足注释的目的。