Convert enum to int in python

Question

I have an enum Nationality:

class Nationality:         Poland='PL'         Germany='DE'         France='FR' 

How can I convert this some enum to int in this or similar way:

position_of_enum = int(Nationality.Poland)  # here I want to get 0 

I know that I can do it if I had code by:

counter=0 for member in dir(Nationality):     if getattr(Nationality, member) == code:         lookFor = member         counter += 1 return counter 

but I don't have, and this way looks too big for python. I'm sure that there is something much simpler .

Answer 1

Please use IntEnum

from enum import IntEnum  class loggertype(IntEnum):     Info = 0     Warning = 1     Error = 2     Fatal = 3  int(loggertype.Info) 0 

Answer 2

Using either the enum34 backport or aenum¹ you can create a specialized Enum:

# using enum34 from enum import Enum  class Nationality(Enum):      PL = 0, 'Poland'     DE = 1, 'Germany'     FR = 2, 'France'      def __new__(cls, value, name):         member = object.__new__(cls)         member._value_ = value         member.fullname = name         return member      def __int__(self):         return self.value 

and in use:

>>> print(Nationality.PL) Nationality.PL >>> print(int(Nationality.PL)) 0 >>> print(Nationality.PL.fullname) 'Poland' 

The above is more easily written using aenum¹:

# using aenum from aenum import Enum, MultiValue  class Nationality(Enum):     _init_ = 'value fullname'     _settings_ = MultiValue      PL = 0, 'Poland'     DE = 1, 'Germany'     FR = 2, 'France'      def __int__(self):         return self.value 

which has the added functionality of:

>>> Nationality('Poland') <Nationality.PL: 0> 

---

¹ Disclosure: I am the author of the Python stdlib Enum, the enum34 backport, and the Advanced Enumeration (aenum) library.