cache enum value

tinylambda · tinylambda · commit 21ed7c6069cd · 2021-12-15T16:48:58.000+08:00
diff --git a/module_enum/enum_get_by_value.py b/module_enum/enum_get_by_value.py
@@ -0,0 +1,33 @@
+import enum
+
+CACHE_CODE_TO_CATEGORY: dict = {}
+
+
+class Category(enum.Enum):
+    HISTORY = {
+        'code': 10001,
+        'name': 'history',
+    }
+    MATH = {
+        'code': 10002,
+        'name': 'math',
+    }
+    PHYSICS = {
+        'code': 10003,
+        'name': 'physics',
+    }
+
+    def __init__(self, value: dict):
+        self.code = value['code']
+        self.category_name = value['name']
+        CACHE_CODE_TO_CATEGORY[self.code] = self
+
+    @classmethod
+    def get_by_code(cls, code: int):
+        return CACHE_CODE_TO_CATEGORY.get(code)
+
+
+if __name__ == '__main__':
+    print(Category.get_by_code(10001))
+    print(Category.get_by_code(10002))
+
diff --git a/module_enum/enum_intenum.py b/module_enum/enum_intenum.py
@@ -14,3 +14,6 @@ class BugStatus(enum.IntEnum):
 print('Ordered by value: ')
 print('\n'.join(' ' + s.name for s in sorted(BugStatus)))
 
+print(BugStatus.new == 7)
+print(BugStatus.incomplete == 6)
+
