SQLAlchemy动态组合共享与服务专属枚举类的实现方案咨询

看起来你遇到了Python枚举无法继承的经典问题，同时还要兼顾SQLAlchemy数据库存储的可扩展性，以及业务层需要区分不同应用操作集合的需求。你的现有方案把所有操作值合并成大列表存入数据库，但业务层枚举没法包含基础操作，确实会导致Steam路由里拿不到TURBINE的问题。下面我给你一套更优雅的解决方案，覆盖枚举扩展、数据库存储和业务层使用的三个核心需求：

1. 解决枚举无法继承的问题：动态合并枚举类 #

Python标准库的Enum确实不支持继承，但我们可以通过动态生成枚举类的方式，把基础枚举和应用专属枚举的成员合并，生成每个应用的完整操作枚举。这样既能保留枚举的类型安全，又能实现“扩展”的效果。

实现方式： #

先定义基础枚举和各应用的专属枚举，再写一个工具函数合并它们（这里用整数作为枚举值，满足你"Operations cannot be String"的要求，也可以替换为UUID等其他非字符串类型）：

import enum
from sqlalchemy import Column, Integer, Enum
from sqlalchemy.ext.declarative import declarative_base

# 共享的基础操作枚举
class BaseOperation(enum.Enum):
    TURBINE = 1

# Steam应用专属操作枚举
class SteamExclusiveOperation(enum.Enum):
    TRAIN = 2

# Gasoline应用专属操作枚举
class GasolineExclusiveOperation(enum.Enum):
    CAR = 3

def combine_enums(base_enum: type[enum.Enum], *exclusive_enums: type[enum.Enum]) -> type[enum.Enum]:
    """
    动态合并基础枚举与专属枚举，生成包含所有操作的新枚举类
    :param base_enum: 共享的基础操作枚举
    :param exclusive_enums: 应用专属的操作枚举
    :return: 合并后的完整枚举类
    """
    combined_members = {}
    # 先添加基础操作的成员，保留原枚举的成员名和值
    for member in base_enum:
        combined_members[member.name] = member.value
    # 再添加各专属操作的成员
    for enum_cls in exclusive_enums:
        for member in enum_cls:
            combined_members[member.name] = member.value
    # 动态生成新的枚举类，命名规则为「基础枚举名_Combined」
    return enum.Enum(f"{base_enum.__name__}_Combined", combined_members)

# 生成Steam应用的完整操作枚举（含基础+Steam专属）
SteamOperation = combine_enums(BaseOperation, SteamExclusiveOperation)
# 生成Gasoline应用的完整操作枚举（含基础+Gasoline专属）
GasolineOperation = combine_enums(BaseOperation, GasolineExclusiveOperation)

这样SteamOperation就同时包含TURBINE和TRAIN两个成员，完全解决了你在Steam路由里拿不到基础操作的问题。

2. SQLAlchemy数据库存储：统一枚举类型+可扩展设计 #

为了让数据库能存储所有可能的操作值，我们需要生成包含所有操作值的枚举类型，同时保证后续可扩展：

Base = declarative_base()

# 收集所有操作值，用于生成数据库枚举
ALL_OPERATION_VALUES = (
    [member.value for member in BaseOperation]
    + [member.value for member in SteamExclusiveOperation]
    + [member.value for member in GasolineExclusiveOperation]
)

class OPEntry(Base):
    __tablename__ = "operation_entries"
    id = Column(Integer, primary_key=True, autoincrement=True)
    # 定义数据库枚举，指定name参数方便后续迁移识别
    operation = Column(
        Enum(*ALL_OPERATION_VALUES, name="operation_type_enum"),
        nullable=False,
        index=True
    )

可扩展性说明： #

• 新增基础操作：只需在BaseOperation里添加成员，更新ALL_OPERATION_VALUES后，用Alembic等迁移工具更新数据库枚举即可。
• 新增应用专属操作：只需在对应专属枚举里添加成员，同样更新ALL_OPERATION_VALUES并执行迁移。
• 数据库适配：PostgreSQL可通过ALTER TYPE operation_type_enum ADD VALUE 4;（假设新值为4）扩展枚举；SQLite会自动将Enum转为String类型加CHECK约束，迁移时更新CHECK约束即可。

3. 业务层路由使用：直接用合并后的枚举 #

现在在Steam的路由里，你可以直接使用SteamOperation枚举，它已经包含了基础操作和专属操作：

from fastapi import APIRouter

steam_router = APIRouter(prefix="/steam")

@steam_router.post("/op-entry")
def add_steam_operation(operation: SteamOperation):
    # operation可以是SteamOperation.TURBINE或SteamOperation.TRAIN
    db_entry = OPEntry(operation=operation.value)
    # 执行数据库保存逻辑...
    return {
        "operation_name": operation.name,
        "operation_value": operation.value,
        "message": "Operation saved successfully"
    }

方案优势对比你的现有方案 #

1. 业务层枚举完整性：每个应用的枚举都包含基础操作和专属操作，解决了原方案中Steam路由拿不到TURBINE的问题。
2. 类型安全：全程使用枚举类型，完全符合你"Operations cannot be String"的要求，避免了直接使用字符串带来的类型错误。
3. 可扩展性：新增操作只需修改对应的枚举类，无需手动维护大列表，降低了出错概率。
4. 数据库兼容性：适配不同数据库的枚举特性，同时支持平滑迁移扩展。

备注：内容来源于stack exchange，提问作者hhilal