重写抽象方法时参数类型不同的Python3.9解决方案咨询

现有方案的问题 #

你当前使用Union+typing.cast的写法语法上合法，但确实违背了抽象方法的设计初衷：

• 抽象方法的参数/返回值标注为所有子类枚举的Union，相当于对外承诺"接受任意子类的枚举参数"，但实际每个子类只处理自身枚举，类型标注完全失去了约束作用
• typing.cast仅对静态类型检查器生效，运行时没有任何校验逻辑，如果误传其他子类的枚举，会直接引发业务逻辑错误，没有利用抽象类的契约能力

适配Python 3.9的最优方案：泛型抽象类 #

Python 3.9原生支持typing.Generic，可以通过泛型类型变量为抽象类定义参数和返回值的占位符，让每个子类绑定自己的专属枚举类型，既保证所有子类的结构一致性，又能实现精准的类型安全。

完整实现代码 #

import typing
from abc import ABC, abstractmethod
from enum import IntEnum, IntFlag

# 定义泛型类型变量，限定为IntEnum和IntFlag的子类
ArgType = typing.TypeVar("ArgType", bound=IntEnum)
ResultType = typing.TypeVar("ResultType", bound=IntFlag)

# 子类专属枚举定义（值重叠不受影响）
class Child1Enum(IntEnum):
    A = 1
    B = 2

class Child1Result(IntFlag):
    W = 1
    X = 4

class Child2Enum(IntEnum):
    C = 1
    D = 2

class Child2Result(IntFlag):
    Y = 1
    Z = 2

# 泛型抽象父类，通过类型变量约束子类的参数与返回值类型
class Parent(ABC, typing.Generic[ArgType, ResultType]):
    @abstractmethod
    def abstract_method(self, argument: ArgType) -> ResultType:
        pass

# Child1绑定自身枚举，类型标注精准
class Child1(Parent[Child1Enum, Child1Result]):
    def abstract_method(self, argument: Child1Enum) -> Child1Result:
        # 直接使用Child1Enum，无需cast，静态检查器会自动校验
        # 业务逻辑示例
        if argument == Child1Enum.A:
            return Child1Result.W
        return Child1Result.X

# Child2绑定自身枚举，独立于其他子类
class Child2(Parent[Child2Enum, Child2Result]):
    def abstract_method(self, argument: Child2Enum) -> Child2Result:
        # 业务逻辑示例
        if argument == Child2Enum.C:
            return Child2Result.Y
        return Child2Result.Z

方案核心优势 #

• 精准类型约束：每个子类的abstract_method仅接受自身枚举参数、返回自身结果枚举，静态类型检查器（如mypy）能提前发现传参错误
• 符合抽象设计理念：父类定义了"必须实现一个接受指定类型参数、返回指定类型结果的方法"的统一契约，子类只需绑定具体类型即可，兼顾结构一致性与灵活性
• 无安全隐患：无需typing.cast，静态检查与运行时均能保证类型正确性
• 高扩展性：新增子类时，只需定义专属枚举，再继承Parent[NewArgEnum, NewResultEnum]即可，完全符合开闭原则

可选增强：运行时类型校验 #

如果需要在运行时也确保参数类型正确，可以在父类中添加简单的校验逻辑：

class Parent(ABC, typing.Generic[ArgType, ResultType]):
    @abstractmethod
    def abstract_method(self, argument: ArgType) -> ResultType:
        # 运行时校验参数类型
        expected_arg_type = self.__orig_bases__[0].__args__[0]
        if not isinstance(argument, expected_arg_type):
            raise TypeError(f"Expected {expected_arg_type.__name__}, got {type(argument).__name__}")

注：__orig_bases__是Python 3.9+泛型类用于存储原始带参数基类的属性，可直接获取子类绑定的具体类型。

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