在__add__运算符中返回相同子类的对象

Question

我正在为自己的解释器开发一个简单的类型系统。我正在写这样的东西：

class Base(object):

    def __init__(self, content):

        self.__content = content

    @property

    def content(self):

        return self.__content

    @content.setter

    def content(self, value):

        self.__content = value

class Number(Base):

    def __init__(self, content):

        super(Number, self).__init__(content)

    def __add__(self, other):

        return Number(self.content + other.content)

    ...and so on

class Float(Number):

    def __init__(self, content):

        super(Float, self).__init__(content)

class Integer(Number):

    def __init__(self, content):

        super(Integer, self).__init__(content)

我的问题是，如果我做这样的事情，显然：

if __name__ == '__main__':

    f1 = Float(3.5)

    f2 = Float(2.3)

    f3 = f1 + f2

    type(f3)

我已经将f1和f2相加，它们是Float类型，但是我已经获得了f3，这是Number类型，但是我希望f3是Float类型。如何仅在我的Number超类中一次定义我的add运算符，以返回与f1和f2相同的类型？我有使用isinstance吗？有没有更清洁的方法可以做到这一点？

Answer 1

您可以使用__class__：def __add__(self, other):&nbsp; &nbsp; return self.__class__(self.content + other.content)正如@Eric指出的那样，您可能想要做类似的事情if self.__class__ == other.__class__:&nbsp; &nbsp; <use __class__>else:&nbsp; &nbsp; <use Number>以确保可预测的行为（如果类不匹配，则采取其他措施）。__radd__ 在这里也值得覆盖：__radd__ = __add__这将使 Number(1) + Float(1) == Float(1) + Number(1) == Float(2)

Answer 2

我认为您想要{(1, 0): self.__class__, (0, 1): other.__class__}.get((issubclass(self.__class__, other.__class__), issubclass(other.__class__, self.__class__)), Number)。如果我不必在评论中加上它，那将不太那么钝。