类的类型提示可以在类函数中使用吗？[副本]

Question

是否可以在同一类的类的方法中使用类型注释？假设我有一个向量类，这个类支持__add__。

像这样的东西：

class Vector:
    def __add__(self, other: Vector):
        #blah blah

这当然行不通。是否有其他类似的方法或解决方法？

Answer 1

您可以检查other是否为向量类型：

def __add__(self, other):
    if isinstance(other, Vector):
        ##do the add
    else:
        raise TypeError('other is not a Vector Type')

Answer 2

类型注释旨在帮助人们阅读代码、静态分析工具和文档工具。即使可以执行代码（我的意思是没有namererror），它实际上也不会强制执行任何内容。

在python中，通常认为强制执行特定类是一种不好的风格，您通常需要一个特定的接口（比如具有特定的属性）：

class Vector(object):
    def __init__(self, x, y):
        self.x, self.y = x, y
    def __add__(self, other):
        try:
            return Vector(self.x + other.x, self.y + other.y)
        except AttributeError:  # the other had no x or y attribute
            raise TypeError('cannot add Vector to {}'.format(type(other)))

或者只需省略try和except，然后让它在属性错误下失败：

class Vector(object):
    def __init__(self, x, y):
        self.x, self.y = x, y
    def __add__(self, other):
        return Vector(self.x + other.x, self.y + other.y)

没有信息异常消息（您需要一个吗？）但是当其他没有x或y属性时，它仍然会失败。

如果我误解了你的问题，它只是关于“如何添加打字提示”，那么你就把它包装成字符串（称为“向前引用”）：

class Vector(object):
    def __add__(self, other: 'Vector'):
        ...

他们甚至在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