关键字仅dataclass子类：继承字段的受限类型提示（不重定义默认值）

当使用kw_only=True的dataclass定义基类字段并被子类继承时，我们希望在子类中为字段添加更严格的类型提示（比如限定枚举值范围），同时避免重复编写基类已有的默认值，还要解决pylint报"Missing mandatory keyword argument"的错误。

以下是两种可行的解决方案：

方法1：使用dataclasses.field复用基类默认值 #

通过field(default=Base.字段名)的方式直接引用基类的默认值，同时更新字段的类型注解。这样既避免了重复定义默认值，又能让类型检查器和pylint识别到字段有默认值。

from dataclasses import dataclass, field
from enum import auto, StrEnum
from typing import Literal

class Direction(StrEnum):
    DOWN = auto()
    LEFT = auto()
    UP = auto()
    RIGHT = auto()
    UNKNOWN = auto()

@dataclass(kw_only=True)
class Base:
    direction: Direction = Direction.UNKNOWN

@dataclass(kw_only=True)
class Horizontal(Base):
    # 复用基类默认值，同时限定类型范围
    direction: Literal[Direction.LEFT, Direction.RIGHT, Direction.UNKNOWN] = field(default=Base.direction)

@dataclass(kw_only=True)
class Vertical(Base):
    direction: Literal[Direction.DOWN, Direction.UP, Direction.UNKNOWN] = field(default=Base.direction)

print(Horizontal())  # Horizontal(direction=<Direction.UNKNOWN: 'UNKNOWN'>)
print(Vertical())    # Vertical(direction=<Direction.UNKNOWN: 'UNKNOWN'>)

方法2：直接覆盖子类的__annotations__字典 #

如果不想重新定义字段，可以直接修改子类的__annotations__，仅更新目标字段的类型注解，基类的字段默认值会被自动继承。如果子类有其他自定义字段，记得合并基类的注解，避免丢失。

from dataclasses import dataclass
from enum import auto, StrEnum
from typing import Literal

class Direction(StrEnum):
    DOWN = auto()
    LEFT = auto()
    UP = auto()
    RIGHT = auto()
    UNKNOWN = auto()

@dataclass(kw_only=True)
class Base:
    direction: Direction = Direction.UNKNOWN

@dataclass(kw_only=True)
class Horizontal(Base):
    # 仅覆盖direction字段的类型注解
    __annotations__ = {**Base.__annotations__, "direction": Literal[Direction.LEFT, Direction.RIGHT, Direction.UNKNOWN]}

@dataclass(kw_only=True)
class Vertical(Base):
    __annotations__ = {**Base.__annotations__, "direction": Literal[Direction.DOWN, Direction.UP, Direction.UNKNOWN]}

print(Horizontal())  # 正常实例化，类型提示生效
print(Vertical())

关键说明 #

• 两种方法都仅作用于类型检查阶段，运行时不会限制字段的实际取值（比如仍可给Horizontal传入Direction.DOWN，但类型检查器会报错）。
• 原代码中Horizontal类报错的原因：dataclass子类重写字段时，如果未显式指定默认值，类型检查器会认为该字段没有默认值，必须显式传入，即使基类有默认值——因为字段重写会覆盖基类的字段定义，默认值不会自动继承。

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