Improved string enum, allowed define different value than key

druids · Jul 9, 2017 · 8f4eeb6 · 8f4eeb6
1 parent b39126b
commit 8f4eeb6
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/chamber/utils/datastructures.py b/chamber/utils/datastructures.py
@@ -21,7 +21,10 @@ def __getattr__(self, name):
 class Enum(AbstractEnum):
 
     def __init__(self, *items):
-        self.container = OrderedDict(((item, item) for item in items))
+        self.container = OrderedDict((
+            item if isinstance(item, (list, tuple)) else (item, item)
+            for item in items
+        ))
         super(Enum, self).__init__()
 
     def _get_attr_val(self, name):