Используйте экземпляры enum в другом классе в Python3

Question

У меня есть класс enum с именем 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

Я думаю, что не смог импортировать свой класс enum должным образом, или я не знал, как использовать enum между классами или что-то в этом роде. Итак, что я должен сделать, чтобы использовать мой enum-класс в других классах?

Спасибо!

Answer 1

Если ваш класс enum содержится в файле с именем UsedPlatform.py, то вы должны изменить свой оператор import в test.py на:

from UsedPlatforms import UsedPlatforms