Change Category from dataclass to enum (#1227)

Author: khyatimahendru

.gencode_hash.txt

@@ -157,7 +157,7 @@ f9d90861e568b27445bef241f04cce64cc44731c95c8bd9e3f65cef79d42dab0  gencode/python
 9f27d0a731ced4b83c9769661fed2656814983394532aac9e2a97c71dd185e2a  gencode/python/udmi/schema/bucket.py
 b4f4a394ce4049fafe267a146458d5b1725e2532788c5811b4b0b96f84715e31  gencode/python/udmi/schema/building_config_entity.py
 6289f299f33104a14be1c67762c8bcb494c008d34a2ff906ef3760164538fc82  gencode/python/udmi/schema/building_translation.py
-d874c058eea491c9fe723d0b1c7bae06a008984a049bcb31ab5fea8de71f71a9  gencode/python/udmi/schema/category.py
+9839ac82da64c8ad1b9ae144fa6be9d819a68f8e22336403321cad9af0d0033f  gencode/python/udmi/schema/category.py
 e92d040fe84cc1bbc6e6b2d81a39dae156e1501e1c07a0b6cfc42e1599403e64  gencode/python/udmi/schema/commands_discovery.py
 8468bc35ad108e4618555cab711bdee700b64780e0d8c18f641c6a0ed21e2338  gencode/python/udmi/schema/commands_mapping.py
 9e15cea83255f018e6e85fdbf8051211561c10ea34caa926a21eb140de31d82c  gencode/python/udmi/schema/common.py

bin/gencode_py_categories

@@ -12,22 +12,41 @@ REGEX = re.compile(
 
 def generate_header():
   return [
-    "from dataclasses import dataclass",
-    "from typing import ClassVar, Dict",
-    "",
-    "from ._base import DataModel",
-    "from .level import Level",
-    "",
-    "@dataclass",
-    "class Category(DataModel):",
-    "  \"\"\"",
-    "  This class is manually curated, auto-generated, and then copied into "
-    "the directory.",
-    "  Look for the proper source and don't be fooled! Ultimately sourced "
-    "from docs/specs/categories.md",
-    "  \"\"\"",
-    "  LEVEL: ClassVar[Dict[str, Level]] = {}",
-    ""
+      "\"\"\"",
+      "This module defines the Category enum, representing system event "
+      "categories",
+      "and their default logging levels.",
+      "\"\"\"",
+      "from enum import Enum",
+      "from .level import Level",
+      "",
+      "class Category(Enum):",
+      "    \"\"\"",
+      "    This class is manually curated, auto-generated, and then copied "
+      "into the directory.",
+      "    Look for the proper source and don't be fooled! Ultimately sourced"
+      " from docs/specs/categories.md",
+      "    \"\"\"",
+      ""
+  ]
+
+def generate_footer():
+  return [
+      "",
+      "    def __init__(self, value_str: str, default_level: Level):",
+      "        self._value_str = value_str",
+      "        self.default_level = default_level",
+      "",
+      "    @property",
+      "    def value(self) -> str:",
+      "        return self._value_str",
+      "",
+      "    @property",
+      "    def level(self) -> int:",
+      "        return self.default_level.value",
+      "",
+      "    def __str__(self) -> str:",
+      "        return self._value_str"
   ]
 
 def process_file():
@@ -61,10 +80,10 @@ def process_file():
         const_name = full_path.upper().replace(".", "_")
 
         entries.append({
-          "const": const_name,
-          "val": full_path,
-          "level": level,
-          "desc": description
+            "const": const_name,
+            "val": full_path,
+            "level": level,
+            "desc": description
         })
 
   return entries
@@ -82,12 +101,12 @@ def write_output(entries):
       lvl = entry["level"]
       desc = entry["desc"]
 
-      f.write(f"  # {desc}\n")
-      f.write(f"  {c_name}: ClassVar[str] = \"{c_val}\"\n")
-      f.write(f"  {c_name}_LEVEL: ClassVar[Level] = Level.{lvl}\n")
-      f.write(f"  {c_name}_VALUE: ClassVar[int] = Level.{lvl}.value\n")
-      f.write(f"  LEVEL[{c_name}] = Level.{lvl}\n")
-      f.write("\n")
+      f.write(f"    # {desc}\n")
+      # Enum member definition: NAME = ("value", Level.CONST)
+      f.write(f"    {c_name} = (\"{c_val}\", Level.{lvl})\n\n")
+
+    for line in generate_footer():
+      f.write(line + "\n")
 
   print(f"Generated {len(entries)} categories into {OUTPUT_FILE}")