Python打字：接受某些类型的通用Enum值

问题描述

我想编写一个接受任何enum.Enum的任何值的函数，只要它的值是str或int。它应与混合int和str值的枚举一起工作。

例如，如果我有：

class E(Enum):
    a = 0
    b = "abc"

我确实希望允许传递诸如E.a和E.b之类的值，而无需实际“知道” E（可以使用任何枚举）。
我不想依靠传递E.a.value或使用字典来代替。因此，仅检查int或str的参数是不够的。

是否可以将其定义为类型提示？

编辑：由于似乎不可能，所以我将仅依赖于str值，并使用Felix建议的解决方案。

解决方法

这可能是可能的，只是没有完全按照您的描述进行。我不会说强制参数为枚举是解决问题的方法。相反，我将使枚举表现像您想要的类型。标准库中已经有一个example。

您希望函数接受字符串或整数。很简单。

def foo(arg: Union[str,int]):
    pass

现在剩下的就是定义枚举了，这些枚举是字符串和整数。如果您不想使用标准的IntEnum，则可以轻松地自己定义。

class IntEnum(int,enum.Enum):
    a = 1
    b = 4

class StrEnum(str,enum.Enum):
    a = 'a'
    b = 'b'

print(IntEnum.a + IntEnum.b) # -> 5
print(StrEnum.a + StrEnum.b) # -> 'ab'

或者，如果您未定义任何实际成员，则可以将它们用作父类。但是，如果您希望在一个枚举中包含字符串和整数，那么最好只调用enum.value。如果您喜欢花样，可以在装饰器中进行示例，但这有点过头了。不一定会推荐它。

from functools import wraps

def enum_value(func):
    @wraps(func)
    def wrapper(arg):
        if isinstance(arg,enum.Enum):
            arg = arg.value
        return func(arg)
    return wrapper

@enum_value
def foo(arg: Union[str,int]):
    print(arg,type(arg))

foo(StrEnum.a) # -> a <class 'str'>

python python-typing typing