小编ase*_*aya的帖子

我有一个名为UsedPlatforms的枚举类：

from enum import Enum

class UsedPlatforms(Enum):
    PROD = 1,
    TEST = 2

我想在另一个 python 类中使用这个枚举，例如：

import UsedPlatforms

def foo(platform):
    if platform == UsedPlatforms.PROD:
        print("Did it!")

foo(platform=UsedPlatforms.PROD)

但是当我运行第二个脚本时，我收到如下错误：

Traceback (most recent call last):
  File "/home/user/Projects/EnumTest/test.py", line 9, in <module>
    foo(platform=UsedPlatforms.PROD)
AttributeError: module 'UsedPlatforms' has no attribute 'PROD'

Process finished with exit code 1

我认为我无法正确导入我的枚举类，或者我不知道如何在类之间使用枚举，或者其他什么。那么，我应该怎么做才能在其他类中使用我的枚举类呢？

谢谢！