From 8f4eeb65f4bfa01e5eb9d963f5232afb946745f5 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Lubo=C5=A1=20M=C3=A1tl?= Date: Sun, 9 Jul 2017 21:07:48 +0200 Subject: [PATCH] Improved string enum, allowed define different value than key --- chamber/utils/datastructures.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/chamber/utils/datastructures.py b/chamber/utils/datastructures.py index 747cba7..7f52832 100644 --- 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):