如何在Python父类中实现返回子类实例并保留子类属性的方法

我编写了Python父类A，其中包含square()方法，返回一个x属性被平方的新类实例：

class A:
    def __init__(self, x):
        self.x = x

    def square(self):
        return self.__class__(self.x**2)

但在子类B中使用该方法时，由于B的构造函数需要额外参数y，直接调用父类的square()会报错：

class B(A):
    def __init__(self, x, y):
        super().__init__(x)
        self.y = y

# 测试代码
b = B(4, 5)
b2 = b.square()  # 报错：TypeError: __init__() takes exactly 3 arguments (2 given)

虽然重写子类的square()方法可以解决问题，但如果父类有多个类似方法（如cube()、double()），且存在多层子类/孙类，重复重写会非常繁琐。我希望在父类中实现通用的square()方法，使其返回当前子类的新实例，同时保留所有额外属性，无需子类逐个重写。

解决方案 #

方法1：利用实例属性字典传递构造参数 #

如果子类的构造参数名与实例属性名完全一致，可以通过vars(self)获取实例的属性字典，修改x后传递给构造函数：

class A:
    def __init__(self, x):
        self.x = x

    def square(self):
        # 复制当前实例的所有属性
        instance_attrs = vars(self).copy()
        # 将x替换为平方后的值
        instance_attrs['x'] = instance_attrs['x'] ** 2
        # 用属性字典作为关键字参数创建新实例
        return self.__class__(**instance_attrs)

class B(A):
    def __init__(self, x, y):
        super().__init__(x)
        self.y = y

# 测试
a = A(3)
a2 = a.square()
print(a2.x)  # 输出：9

b = B(4, 5)
b2 = b.square()
print(b2.x, b2.y)  # 输出：16 5

这种方法无需子类做任何修改，只要构造参数和实例属性一一对应即可正常工作。

方法2：使用dataclass简化属性与构造参数的映射 #

如果使用Python的dataclasses模块，会自动处理构造参数与实例属性的对应关系，代码更简洁：

from dataclasses import dataclass, asdict

@dataclass
class A:
    x: int

    def square(self):
        # 获取实例的属性字典
        attrs = asdict(self)
        attrs['x'] = attrs['x'] ** 2
        return self.__class__(**attrs)

@dataclass
class B(A):
    y: int

# 测试
a = A(3)
a2 = a.square()
print(a2.x)  # 输出：9

b = B(4, 5)
b2 = b.square()
print(b2.x, b2.y)  # 输出：16 5

方法3：定义钩子方法适配构造参数与属性不匹配的场景 #

如果子类的构造参数名与实例属性名不一致（例如构造参数是y_val，但实例属性是self.y），可以在父类定义一个钩子方法，让子类返回构造所需的参数字典：

class A:
    def __init__(self, x):
        self.x = x

    # 钩子方法：子类可重写以返回构造参数字典
    def get_constructor_args(self):
        return {'x': self.x}

    def square(self):
        args = self.get_constructor_args()
        args['x'] = args['x'] ** 2
        return self.__class__(**args)

class B(A):
    def __init__(self, x, y_val):
        super().__init__(x)
        self.y = y_val

    # 重写钩子方法，返回构造所需的参数
    def get_constructor_args(self):
        return {'x': self.x, 'y_val': self.y}

# 测试
b = B(4, 5)
b2 = b.square()
print(b2.x, b2.y)  # 输出：16 5

这种方式下，即使有多个类似square()的方法，子类也只需要重写一次get_constructor_args()即可，避免了重复代码。

内容的提问来源于stack exchange，提问作者nieswand