Python - 在创建过程中扩展枚举字段-解网

问：

是否可以在创建过程中扩展枚举？例：

class MyEnum(enum.StrEnum):
    ID = "id"
    NAME = "NAME

创建此枚举后，我需要包含以下字段：

ID = "id"
NAME = "name"
ID_DESC = "-id"
NAME_DESC = "-name"

我需要它来为 FastAPI 项目创建自定义排序枚举

现在我有了创建新枚举的下一种方法

NewEnum = enum.StrEnum(
    f"{name.title()}OrderingEnum",
    [
        (
            f"{ordering_field.upper()}_DESC"
            if ordering_field.startswith("-")
            else ordering_field.upper(),
            ordering_field,
        )
        for ordering_field in itertools.chain(
            values,
            [f"-{field}" for field in values],
        )
    ],
)

但我需要自动执行此操作，因为每个具有模型的模块都有相似的枚举。也许这可以解决我的问题，用于我的枚举类或覆盖方法，但我还没有找到可行的解决方案MetaClass__new__

Python 枚举元类

from enum import EnumMeta, StrEnum

class IDEnumMeta(EnumMeta):
    def __new__(metacls, cls, bases, classdict, **kwds):
        # add new entries to classdict
        for name in list(classdict._member_names):
            classdict[f'{name}_DESC'] = f'-{classdict[name]}'
        return super().__new__(metacls, cls, bases, classdict, **kwds)

class IDEnum(StrEnum, metaclass=IDEnumMeta):
    pass

并在使用中：

>>> class MyEnum(IDEnum):
...     ID = 'id'
...     NAME = 'name'

>>> list(MyEnum)
[<MyEnum.ID: 'id'>, <MyEnum.NAME: 'name'>, <MyEnum.ID_DESC: '-id'>, <MyEnum.NAME_DESC: '-name'>]

此答案目前使用 private 及其属性，但它们将在 3.13 中公开。_EnumDict_member_names

披露：我是 Python stdlib 枚举、enum34 向后移植和高级枚举（aenum）库的作者。

Python - 在创建过程中扩展枚举字段

Python - extend enum fields during creation

评论

评论