如何在python中检查变量是否为枚举类型

Question

我有一个这样的枚举

@enum.unique
class TransactionTypes(enum.IntEnum):
    authorisation = 1
    balance_adjustment = 2
    chargeback = 3
    auth_reversal = 4

现在我用这个枚举分配一个变量，像这样

a = TransactionTypes

我想检查 'a' 的类型，如果它是枚举，则做一些事情，如果不是枚举，则做其他事情

我试过这样的

if type(a) == enum:
    print "do enum related stuff"
else:
    print "do something else"

问题是它不能正常工作。

Answer 1

现在我用这个枚举分配一个变量，像这样

a = TransactionTypes

我希望你不是，因为你刚刚分配给a 的是整个枚举，而不是它的一个成员（例如TransactionTypes.chargeback）如果这真的是你想做的，那么正确的测试应该是：

if issubclass(a, enum.Enum)

但是，如果你的意思是这样的：

a = TransactionTypes.authorisation

那么你需要的测试是：

# for any Enum member
if isinstance(a, Enum):

或

# for a TransactionTypes Enum
if isinstance(a, TransactionTypes):

Answer 2

如前所述，使用isinstance 方法检查天气实例是否为enum.Enum 类型。

用于演示其用法的小工作代码：

import enum


class STATUS(enum.Enum):
    FINISHED = enum.auto()
    DELETED = enum.auto()
    CANCELLED = enum.auto()
    PENDING = enum.auto()


if __name__ == "__main__":
    instance = STATUS.CANCELLED

    if isinstance(instance, enum.Enum):
        print('name : ', instance.name, ' value : ', instance.value)
    else:
        print(str(instance))

输出：

name :  CANCELLED  value :  3

Answer 3

可靠的解决方案：

from enum import IntEnum
from collections import Iterable

def is_IntEnum(obj):
    try:
        return isinstance(obj, Iterable) and isinstance (next(iter(obj)), IntEnum)
    except:
        return False # Handle StopIteration, if obj has no elements

Answer 4

我以为我有一个丑陋的方法。例如：

print(o.__class__.__class__) 

输出：

<enum.EnumMeta>