键入python：在类定义中使用类自己的类型

Question

以下代码不能按预期方式工作。显然，我不能使用类自己的类型内的类定义：

class Foo: 
    def __init__(self, key :str) -> None: 
     self.key = key 

    def __eq__(self, other :Foo) -> bool: 
     return self.key == other.key 

print('should be true: ', Foo('abc') == Foo('abc')) 
print('should be false: ', Foo('abc') == Foo('def')) 

运行它的结果是：

Traceback (most recent call last): 
    File "class_own_type.py", line 1, in <module> 
    class Foo: 
    File "class_own_type.py", line 5, in Foo 
    def __eq__(self, other :Foo) -> bool: 
NameError: name 'Foo' is not defined 

而且，与mypy返回检查代码：

class_own_type.py:5: error: Argument 1 of "__eq__" incompatible with supertype "object" 

如何纠正这段代码对于Python和mypy都有效？

Answer 1

名称Foo尚未绑定，因为在尝试使用该名称时该类本身尚未定义（请记住，函数参数在函数定义时计算，而不是在函数调用时计算） 。

您可以使用字符串常量，延迟类型的评价：

class Foo: 
    def __init__(self, key :str) -> None: 
     self.key = key 

    def __eq__(self, other: 'Foo') -> bool: 
     return self.key == other.key 

print('should be true: ', Foo('abc') == Foo('abc')) 
print('should be false: ', Foo('abc') == Foo('def'))