Use enum instances in another class in Python3

Question

I have an enum class called UsedPlatforms:

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

I want to use this enum in another python class like:

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

But when I run second script, I get an error like:

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

I think I couldn't import my enum class properly or I didn't know how to use enums between classes, or smth. So, what should I do to use my enum class in another classes?

Thanks!

You don't need/want commas after the value or you'll get a tuple value -- i.e. PROD == (1, ). — Ethan Furman
– Ethan Furman, Commented Jul 24, 2019 at 8:06

Mauricio Gracia Gutierrez · Accepted Answer · 2022-03-18 17:11:41Z

If your enum class contained in a file named UsedPlatform.py, then you should change your import statement in test.py to:

from UsedPlatforms import UsedPlatforms

If the file you are trying to import is inside a folder you need :

from folder.FileWithoutExtension import <ClassName/Enum>

Collectives™ on Stack Overflow

Use enum instances in another class in Python3

1 Answer 1

Comments

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Related