如何让Python Enum的int值动态匹配对应数据库记录ID？

我有一张存储特定公司信息的数据库表，代码中需要以“静态”方式引用这些公司（比如用Enum的auto()生成枚举值）。我希望Enum的int值能和对应公司的数据库ID匹配，但不想在Enum定义里硬编码ID——因为数据库重建或恢复时ID可能会变化。目前我通过Peewee结合lru_cache实现了动态获取ID的属性，请问怎么让这个动态解析的ID成为Enum的实际int值？

示例代码：

from enum import Enum, auto

class CompanyEnum(Enum):
     APPLE = auto()
     AMAZON = auto()
     META = auto()

当前实现：

from functools import lru_cache
from enum import Enum, auto
from models import Company

class CompanyEnum(Enum):
     APPLE = auto()
     AMAZON = auto()
     META = auto()

     @lru_cache
     @property
     def id(self):
         company = Company.get(Company.name.upper() == self.name)
         return company.id

Enum的成员值在类定义时就会被固定，直接用auto()后动态修改行不通。可以通过自定义Enum元类或者动态生成枚举类的方式，让枚举值直接从数据库加载对应的ID。

方法一：自定义Enum元类（推荐，支持IDE自动补全） #

通过自定义元类，在枚举类创建时自动查询数据库，将auto()占位符替换为对应的公司ID：

from enum import Enum, EnumMeta, auto
from models import Company

class DynamicCompanyEnumMeta(EnumMeta):
    def __new__(metacls, cls, bases, classdict):
        # 筛选出所有用auto()定义的枚举成员
        auto_members = [name for name, val in classdict.items() if isinstance(val, auto)]
        if auto_members:
            # 批量查询数据库中的目标公司，减少IO次数
            company_map = {
                c.name.upper(): c.id 
                for c in Company.select().where(Company.name.upper() << auto_members)
            }
            
            # 替换auto()为数据库ID
            for name in auto_members:
                if name not in company_map:
                    raise ValueError(f"数据库中未找到公司：{name}")
                classdict[name] = company_map[name]
        
        return super().__new__(metacls, cls, bases, classdict)

class CompanyEnum(Enum, metaclass=DynamicCompanyEnumMeta):
    APPLE = auto()
    AMAZON = auto()
    META = auto()

# 使用示例
print(CompanyEnum.APPLE.value)  # 直接输出数据库中的APPLE ID
print(int(CompanyEnum.AMAZON))  # 转int也能得到ID

优点 #

• 保留了枚举类的显式定义，IDE能识别成员并提供自动补全
• 类初始化时仅查询一次数据库，性能高效
• 枚举值直接对应数据库ID，无需额外调用id属性

方法二：动态生成枚举类（代码更简洁） #

如果不需要IDE的自动补全提示，可以直接从数据库查询后动态生成枚举类：

from enum import Enum
from functools import lru_cache
from models import Company

@lru_cache(maxsize=None)
def get_company_enum():
    # 指定需要加载的公司名称列表
    target_companies = ['APPLE', 'AMAZON', 'META']
    # 查询数据库并构建名称-ID映射
    company_data = {
        c.name.upper(): c.id 
        for c in Company.select().where(Company.name.upper() << target_companies)
    }
    # 动态生成枚举类
    return Enum('CompanyEnum', company_data)

# 使用时先获取枚举类
CompanyEnum = get_company_enum()

# 示例调用
print(CompanyEnum.META.value)

优点 #

• 代码更简洁，无需定义冗长的元类
• 可以灵活调整需要加载的公司列表

注意事项 #

1. 数据库初始化时机：必须确保数据库连接已初始化、表已创建后，再加载枚举类，否则会抛出数据库连接错误
2. 数据一致性：如果数据库重建导致ID变化，需要重启应用重新加载枚举类（因为枚举值在初始化时就固定了）
3. 异常处理：可以根据业务需求调整缺失公司的处理逻辑，比如设置默认ID或跳过该成员

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