循环遍历类中所有类似于枚举的变量

Question

我有一个类似于枚举的类。 我想遍历他的变量（枚举值）

class Demos(object):
    class DemoType(object):
        def __init__(self, name):
            self.name = name

        def __repr__(self):
            return self.name

    VARIABLE1 = DemoType("Car")
    VARIABLE2 = DemoType("Bus")
    VARIABLE3 = DemoType("Example")
    VARIABLE4 = DemoType("Example2")

我考虑过使用Role.__dict__ 或vars(Role)，但它们不仅包含变量，还包含RoleType 类和其他属性，如__module__。 __doc__ 等等...

我也希望它是这样表示的，主要是因为它会给DemoType添加更多变量。 name以外的变量，所以请尝试通过这种方式找到答案。

Answer 1

与其重新发明枚举类型，不如使用 Python 的 Enum 类型（也曾是 backported）。然后你的代码可能看起来像

class Demos(Enum):
    VARIABLE1 = "Car"
    VARIABLE2 = "Bus"
    VARIABLE3 = "Example"
    VARIABLE4 = "Example2"


--> for variable in Demos:
...    print variable

Answer 2

我找到了答案，而且它根本不是 How can I represent an 'Enum' in Python? 的副本。 答案是通过以下list comprehensive创建以下list：

variables = [attr for attr in dir(Demos()) if not attr.startswith("__") and not callable(attr)]
print variables 

我也可以通过这种方式为我创建一个函数：

class Demos(object):
    class DemoType(object):
        def __init__(self, name):
            self.name = name

        def __repr__(self):
            return self.name

    @classmethod
    def get_variables(cls):
        return [getattr(cls, attr) for attr in dir(cls) if not callable(getattr(cls, attr)) and not attr.startswith("__")]

    VARIABLE1 = DemoType("Car")
    VARIABLE2 = DemoType("Bus")
    VARIABLE3 = DemoType("Example")
    VARIABLE4 = DemoType("Example2")


for variable in Demos.get_variables():
    print variable