Python继承dataclass子类如何预设父类必填参数？

问题根源 #

基类__Sensor的type是必填字段，子类继承后未覆盖该字段的必填性，导致实例化时必须传入type参数；同时原代码中__post_init__的执行顺序错误——父类的__post_init__会先于子类的type赋值执行，即使传入type也会导致unique_id生成逻辑使用错误的type值。

解决方案：子类重定义type字段并固定值 #

通过dataclasses.field在子类中重定义type字段，设置固定默认值并禁止通过构造方法传入，既满足父类对type字段的要求，又无需自定义__init__。

修改后的完整代码：

from typing import Optional
from dataclasses import dataclass, field
import re
       
@dataclass(kw_only=True)
class __Sensor():
    name: str
    type: str
    unique_id: Optional[str] = None

    def cleanName(self):
        return re.sub(r'[^A-Za-z]', '', self.name).lower()
        
    def __post_init__(self):
        if self.unique_id is None:
            self.unique_id = self.type + "_" + self.cleanName()

@dataclass(kw_only=True)
class BinarySensor(__Sensor):
    some_attrib: str
    # 重定义type字段，固定默认值且禁止构造方法传入
    type: str = field(init=False, default="binary_sensor")

@dataclass(kw_only=True)
class PowerSensor(__Sensor):
    some_attrib: str
    type: str = field(init=False, default="power_sensor")

# 基类不建议实例化（符合设计初衷）
# mySensor = __Sensor(name="mySensor", type="Dummy")
# print(mySensor)

myBinSensor = BinarySensor(name="myBinarySensor", some_attrib="a lot of stuff here")
print(myBinSensor)
# 输出：BinarySensor(name='myBinarySensor', type='binary_sensor', unique_id='binary_sensor_mybinarysensor', some_attrib='a lot of stuff here')

说明 #

1. field(init=False)：告诉dataclass生成构造方法时不包含该字段，避免用户传入type参数
2. default="binary_sensor"：给type设置固定默认值，满足父类对该字段的非空要求
3. 无需再在__post_init__中赋值type，字段默认值会在实例创建时自动生效，父类的__post_init__能直接使用正确的type值生成unique_id

如果需要严格禁止基类被实例化，可以给基类添加abc.ABC装饰器，使其成为抽象类：

from abc import ABC

@dataclass(kw_only=True)
class __Sensor(ABC):
    # ... 原有代码不变

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