使用Python：在类的定义中使用自己的类型

20 浏览
0 Comments

使用Python：在类的定义中使用自己的类型

此问题已在其他地方有答案：

Python中类型注释的自引用或前向引用[重复]

如何使用封闭类的类型提示方法?

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

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 
    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中都有效？

admin 更改状态以发布 2023年5月25日
0

1 答案

0 Comments

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

从Python 3.7+开始，您可以通过在模块顶部添加此导入来推迟注释的评估：https://www.python.org/dev/peps/pep-0563/

from __future__ import annotations

对于Python < 3.7，您可以使用字符串文字来延迟类型的评估：

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'))

0