Python Enum存储函数时通过名称访问触发KeyError问题求助

你碰到的这个KeyError问题，本质是Enum类对函数/方法的特殊处理逻辑导致的：

当你在Enum类中直接赋值一个普通函数时，Enum的元类（EnumMeta）会把这个函数识别为类的方法，而不是Enum成员的value。所以它不会被加入到Enum的成员映射字典里，自然用MyEnum['function']这种键索引的方式就找不到它，触发KeyError；但直接用属性访问MyEnum.function时，其实是在访问类的方法，所以能输出<unbound method MyEnum.f>。

而当你赋值字符串、数值这类非函数值时，Enum会正常把它们注册为Enum成员，所以两种访问方式都能正常工作。

解决办法 #

要让函数能作为Enum成员的value，同时支持键索引访问，你只需要把函数包装成一个非方法类型的对象，让Enum元类把它识别为成员而不是类方法。这里有两种简单的实现方式：

方法1：用staticmethod包装函数 #

staticmethod会把函数转换成一个静态方法对象，Enum不会把它当作类方法处理，而是会将其作为成员的value注册：

from enum import Enum

def f():
    print("Hello from f!")

class MyEnum(Enum):
    function = staticmethod(f)

# 现在两种访问方式都正常了
print(MyEnum.function)          # MyEnum.function
print(MyEnum.function.value)    # <function f at 0x...>
print(MyEnum['function'])       # MyEnum.function
# 调用函数的话，直接取value后执行
MyEnum.function.value()         # 输出: Hello from f!

方法2：自定义包装类封装函数 #

如果需要更灵活的控制，可以写一个简单的包装类，把函数包起来，这样Enum会把包装类的实例当作普通成员value：

from enum import Enum

def f():
    print("Hello from f!")

class WrappedFunction:
    def __init__(self, func):
        self.func = func
    def __call__(self, *args, **kwargs):
        return self.func(*args, **kwargs)

class MyEnum(Enum):
    function = WrappedFunction(f)

# 测试访问
print(MyEnum.function)          # MyEnum.function
print(MyEnum['function'])       # MyEnum.function
# 调用函数
MyEnum.function.value()         # 输出: Hello from f!

补充说明 #

如果你只是需要在Enum里存储函数并能调用，也可以考虑直接用字典来映射名称和函数，但既然想用Enum的特性，上面的包装方式就能完美解决你的问题啦～

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