Enum questions.
jmp
jeanmichel at sequans.com
Wed Apr 13 07:59:09 EDT 2016
On 04/13/2016 12:12 PM, Antoon Pardon wrote:
> I have been looking at the enum documentation and it
> seems enums are missing two features I rather find
> important.
>
> 1) Given an Enum value, someway to get the next/previous
> one
>
> 2) Given two Enum values, iterate over the values between
> them.
>
> Did I miss those in the documentation or are they really
> missing?
>
From the doc :
"While Enum and IntEnum are expected to cover the majority of use-cases,
they cannot cover them all. Here are recipes for some different types of
enumerations that can be used directly, or as examples for creating
one’s own."
I would disagree with you when you state that the features you mentioned
are important. They could be useful, in certain cases but most of your
code would better be agnostic to the enum value.
Now it is possible that you specifically work with a system where those
features would be really useful. As mentioned by the documentation,
subclassing Enum is possible:
tested with python 2.7
from enum import IntEnum
class UltimateEnum(IntEnum):
@classmethod
def range(cls, *args):
enumToVal = lambda e: e.value if isinstance(e, cls) else e
return (i for i in cls if i.value in range(*map(enumToVal, args)))
@property
def next(self):
it = iter(self.__class__)
try:
for e in it:
if e is self: return next(it)
except StopIteration:
return None
return None
class Foo(UltimateEnum):
A = 1
B = 4
C = 9
D = 28
print "first to C:"
for f in Foo.range(Foo.C):
print f
print "B to D :"
for f in Foo.range(Foo.B, Foo.D):
print f
print "A to D+1 with step 2 : "
for f in Foo.range(Foo.A, Foo.D.value+1, 2):
print f
print "next property"
print Foo.A.next
print Foo.C.next
print Foo.D.next
In [1]: run test.py
first to C:
Foo.A
Foo.B
B to D :
Foo.B
Foo.C
A to D+1 with step 2 :
Foo.A
Foo.C
next property
Foo.B
Foo.D
None
Hope it helps,
jm
More information about the Python-list
mailing list