如何让子类dataclass初始化并隐藏父类指定字段？

这个需求完全可以实现，下面提供两种符合Python dataclass规范的写法：

方法一：重写字段并使用__post_init__自动赋值 #

这种方法利用dataclass的字段配置和后置初始化方法，既保留自动生成的初始化逻辑，又能隐藏父类的a参数：

from dataclasses import dataclass, field

@dataclass
class A:
    a: str
    b: str

@dataclass
class B(A):
    c: str
    # 重写父类的a字段，设置init=False排除出初始化参数列表
    a: str = field(init=False)

    def __post_init__(self):
        # 在初始化完成后自动给a赋值
        self.a = f"extended-{self.c}"

# 测试调用，无需传入a
test = B(b="foo", c="bar")
print(test.a)  # 输出: extended-bar
print(test.b)  # 输出: foo
print(test.c)  # 输出: bar

方法二：自定义__init__方法 #

如果需要更灵活的初始化逻辑，可以手动编写子类的构造函数，直接控制父类的初始化参数：

from dataclasses import dataclass

@dataclass
class A:
    a: str
    b: str

# 禁用dataclass自动生成__init__，我们自己写
@dataclass(init=False)
class B(A):
    c: str

    def __init__(self, b: str, c: str):
        # 调用父类构造函数，直接传入自动生成的a值
        super().__init__(a=f"extended-{c}", b=b)
        self.c = c

test = B(b="foo", c="bar")
print(test.a)  # 输出: extended-bar
print(test.b)  # 输出: foo
print(test.c)  # 输出: bar

原理说明 #

默认情况下，dataclass子类会继承父类的所有字段作为初始化参数，所以原代码中的B会要求传入a、b、c三个参数。通过重写父类字段并设置init=False，或者禁用自动生成构造函数后自定义__init__，就能实现隐藏a参数并自动赋值的效果。

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