ensure command_id is not None

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

Author: varun-edachali-dbx

src/databricks/sql/backend/thrift_backend.py

@@ -796,6 +796,8 @@ def _results_message_to_execute_response(self, resp, operation_state):
             arrow_queue_opt = None
 
         command_id = CommandId.from_thrift_handle(resp.operationHandle)
+        if command_id is None:
+            raise ValueError(f"Invalid Thrift handle: {resp.operationHandle}")
 
         return ExecuteResponse(
             arrow_queue=arrow_queue_opt,
@@ -1156,6 +1158,8 @@ def get_columns(
 
     def _handle_execute_response(self, resp, cursor):
         command_id = CommandId.from_thrift_handle(resp.operationHandle)
+        if command_id is None:
+            raise ValueError(f"Invalid Thrift handle: {resp.operationHandle}")
 
         cursor.active_command_id = command_id
         self._check_direct_results_for_error(resp.directResults)
@@ -1169,6 +1173,9 @@ def _handle_execute_response(self, resp, cursor):
 
     def _handle_execute_response_async(self, resp, cursor):
         command_id = CommandId.from_thrift_handle(resp.operationHandle)
+        if command_id is None:
+            raise ValueError(f"Invalid Thrift handle: {resp.operationHandle}")
+
         cursor.active_command_id = command_id
         self._check_direct_results_for_error(resp.directResults)