house constants in enums for readability and immutability

Signed-off-by: varun-edachali-dbx <varun.edachali@databricks.com>

Author: varun-edachali-dbx

src/databricks/sql/backend/sea/backend.py

@@ -5,6 +5,10 @@
 
 from databricks.sql.backend.sea.utils.constants import (
     ALLOWED_SESSION_CONF_TO_DEFAULT_VALUES_MAP,
+    ResultFormat,
+    ResultDisposition,
+    ResultCompression,
+    WaitTimeout,
 )
 
 if TYPE_CHECKING:
@@ -405,17 +409,25 @@ def execute_command(
                     )
                 )
 
-        format = "ARROW_STREAM" if use_cloud_fetch else "JSON_ARRAY"
-        disposition = "EXTERNAL_LINKS" if use_cloud_fetch else "INLINE"
-        result_compression = "LZ4_FRAME" if lz4_compression else None
+        format = (
+            ResultFormat.ARROW_STREAM if use_cloud_fetch else ResultFormat.JSON_ARRAY
+        ).value
+        disposition = (
+            ResultDisposition.EXTERNAL_LINKS
+            if use_cloud_fetch
+            else ResultDisposition.INLINE
+        ).value
+        result_compression = (
+            ResultCompression.LZ4_FRAME if lz4_compression else ResultCompression.NONE
+        ).value
 
         request = ExecuteStatementRequest(
             warehouse_id=self.warehouse_id,
             session_id=sea_session_id,
             statement=operation,
             disposition=disposition,
             format=format,
-            wait_timeout="0s" if async_op else "10s",
+            wait_timeout=(WaitTimeout.ASYNC if async_op else WaitTimeout.SYNC).value,
             on_wait_timeout="CONTINUE",
             row_limit=max_rows,
             parameters=sea_parameters if sea_parameters else None,

src/databricks/sql/backend/sea/utils/constants.py

@@ -3,6 +3,7 @@
 """
 
 from typing import Dict
+from enum import Enum
 
 # from https://docs.databricks.com/aws/en/sql/language-manual/sql-ref-parameters
 ALLOWED_SESSION_CONF_TO_DEFAULT_VALUES_MAP: Dict[str, str] = {
@@ -15,3 +16,31 @@
     "TIMEZONE": "UTC",
     "USE_CACHED_RESULT": "true",
 }
+
+
+class ResultFormat(Enum):
+    """Enum for result format values."""
+
+    ARROW_STREAM = "ARROW_STREAM"
+    JSON_ARRAY = "JSON_ARRAY"
+
+
+class ResultDisposition(Enum):
+    """Enum for result disposition values."""
+
+    EXTERNAL_LINKS = "EXTERNAL_LINKS"
+    INLINE = "INLINE"
+
+
+class ResultCompression(Enum):
+    """Enum for result compression values."""
+
+    LZ4_FRAME = "LZ4_FRAME"
+    NONE = None
+
+
+class WaitTimeout(Enum):
+    """Enum for wait timeout values."""
+
+    ASYNC = "0s"
+    SYNC = "10s"