How do I enumerate/declare some constants in Python?

if my program have three states:
0 FETCH
1 PROCESS
2 SEND

I can set variable like state=2
but how can I make it work with something more verbose, like state=SEND ?

basically: what is C would look like:
#define SEND 2
or
enum StateTypes {
FETCH,
PROCESS,
SEND,
};

https://docs.python.org/3/library/enum.html#creating-an-enum

from enum import Enum

class StateTypes(Enum):
     FETCH = 0
     PROCESS = 1
     SEND = 2

state = StateTypes.SEND

or using automatic values:

from enum import Enum, auto

class StateTypes(Enum):
     FETCH = auto()
     PROCESS = auto()
     SEND = auto()

state = StateTypes.SEND

or simply:

FETCH = 0
PROCESS = 1
SEND = 2

state = SEND
1 Like

Thank you. :slight_smile:

1 Like