FilesExpand file tree

 master

/

enum.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

38 lines (35 loc) · 1.59 KB

 master

/

enum.py

Copy path

Top

File metadata and controls

• Code
• Blame

38 lines (35 loc) · 1.59 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

# --------------------------------------------

# http://code.activestate.com/recipes/413486/

# --------------------------------------------

def Enum(*names):

##assert names, "Empty enums are not supported" # <- Don't like empty enums? Uncomment!

class EnumClass(object):

__slots__ = names

def __iter__(self): return iter(constants)

def __len__(self): return len(constants)

def __getitem__(self, i): return constants[i]

def __repr__(self): return 'Enum' + str(names)

def __str__(self): return 'enum ' + str(constants)

class EnumValue(object):

__slots__ = ('__value')

def __init__(self, value): self.__value = value

Value = property(lambda self: self.__value)

EnumType = property(lambda self: EnumType)

def __hash__(self): return hash(self.__value)

def __cmp__(self, other):

# C fans might want to remove the following assertion

# to make all enums comparable by ordinal value {;))

assert self.EnumType is other.EnumType, "Only values from the same enum are comparable"

return cmp(self.__value, other.__value)

def __invert__(self): return constants[maximum - self.__value]

def __nonzero__(self): return bool(self.__value)

def __repr__(self): return str(names[self.__value])

maximum = len(names) - 1

constants = [None] * len(names)

for i, each in enumerate(names):

val = EnumValue(i)

setattr(EnumClass, each, val)

constants[i] = val

constants = tuple(constants)

EnumType = EnumClass()

return EnumType