在__init__中使用继承的类方法

时间:2015-11-10 14:24:24

标签: python inheritance class-method

我有一个由几个孩子继承的父类。我想使用父@classmethod个初始值设定项初始化其中一个孩子。我怎样才能做到这一点?我试过了:

class Point(object):
    def __init__(self,x,y):
        self.x = x
        self.y = y

    @classmethod
    def from_mag_angle(cls,mag,angle):
        x = mag*cos(angle)
        y = mag*sin(angle)
        return cls(x=x,y=y)


class PointOnUnitCircle(Point):
    def __init__(self,angle):
        Point.from_mag_angle(mag=1,angle=angle)


p1 = Point(1,2)
p2 = Point.from_mag_angle(2,pi/2)
p3 = PointOnUnitCircle(pi/4)
p3.x #fail

2 个答案:

答案 0 :(得分:3)

如果您尝试这样写__init__PointOnUnitCirclePoint的界面不同(因为它需要angle而不是x, y)因此,它不应该是它的子类。怎么样:

class PointOnUnitCircle(Point):

    def __init__(self, x, y):
        if not self._on_unit_circle(x, y):
            raise ValueError('({}, {}) not on unit circle'.format(x, y))
        super(PointOnUnitCircle, self).__init__(x, y)

    @staticmethod
    def _on_unit_circle(x, y):
        """Whether the point x, y lies on the unit circle."""
        raise NotImplementedError

    @classmethod
    def from_angle(cls, angle):
        return cls.from_mag_angle(1, angle)

    @classmethod
    def from_mag_angle(cls, mag, angle):  
        # note that switching these parameters would allow a default mag=1
        if mag != 1:
            raise ValueError('magnitude must be 1 for unit circle')
        return super(PointOnUnitCircle, cls).from_mag_angle(1, angle)

这使接口保持相同,添加了检查子类输入的逻辑(一旦你编写了它!)并提供了一个新的类方法,可以从{{轻松构造一个新的PointOnUnitCircle 1}}。而不是

angle

你必须写

p3 = PointOnUnitCircle(pi/4)

答案 1 :(得分:0)

您可以覆盖子类的__new__方法,从超类的备用构造函数构造实例,如下所示。

import math


class Point(object):

    def __init__(self, x, y):
        self.x = x
        self.y = y

    @classmethod
    def from_polar(cls, radius, angle):
        x = radius * math.cos(angle)
        y = radius * math.sin(angle)
        return cls(x, y)


class PointOnUnitCircle(Point):

    def __new__(cls, angle):
        point = Point.from_polar(1, angle)
        point.__class__ = cls
        return point

    def __init__(self, angle):
        pass

请注意,在__new__中,行point = Point.from_polar(1, angle)不能被point = super().from_polar(1, angle)替换,因为Point将自身作为备用构造函数的第一个参数发送,{{1}将子类super()发送到备用构造函数,该构造函数循环调用调用它的子类PointOnUnitCircle,依此类推,直到__new__发生。另请注意,即使RecursionError在子类中为空,但未在子类中覆盖__init__,超级类__init__也会在__init__之后立即自动调用,撤消备用构造函数。

或者,某些对象设计的组合比继承更简单。例如,您可以使用以下类替换上述__new__类而不覆盖PointOnUnitCircle

__new__