shifted EnumEncoder to telemetry/utils.py

Signed-off-by: Sai Shree Pradhan <saishree.pradhan@databricks.com>

Author: saishreeeee

src/databricks/sql/telemetry/models/event.py

@@ -9,20 +9,7 @@
     ExecutionResultFormat,
 )
 from typing import Optional
-from enum import Enum
-
-
-class EnumEncoder(json.JSONEncoder):
-    """
-    Custom JSON encoder to handle Enum values.
-    This is used to convert Enum values to their string representations.
-    Default JSON encoder raises a TypeError for Enums.
-    """
-
-    def default(self, obj):
-        if isinstance(obj, Enum):
-            return obj.value
-        return super().default(obj)
+from databricks.sql.telemetry.utils import EnumEncoder
 
 
 @dataclass

src/databricks/sql/telemetry/models/frontend_logs.py

@@ -1,6 +1,7 @@
 import json
 from dataclasses import dataclass, asdict
-from databricks.sql.telemetry.models.event import TelemetryEvent, EnumEncoder
+from databricks.sql.telemetry.models.event import TelemetryEvent
+from databricks.sql.telemetry.utils import EnumEncoder
 from typing import Optional
 
 

src/databricks/sql/telemetry/utils.py

@@ -0,0 +1,15 @@
+import json
+from enum import Enum
+
+
+class EnumEncoder(json.JSONEncoder):
+    """
+    Custom JSON encoder to handle Enum values.
+    This is used to convert Enum values to their string representations.
+    Default JSON encoder raises a TypeError for Enums.
+    """
+
+    def default(self, obj):
+        if isinstance(obj, Enum):
+            return obj.value
+        return super().default(obj)