ensure error exists before extracting err details

varun-edachali-dbx · varun-edachali-dbx · commit 6a4faede7ae0 · 2025-07-22T10:20:29.000+05:30
Signed-off-by: varun-edachali-dbx &lt;varun.edachali@databricks.com&gt;
diff --git a/src/databricks/sql/backend/sea/backend.py b/src/databricks/sql/backend/sea/backend.py
@@ -410,8 +410,10 @@ def _check_command_not_in_failed_or_closed_state(
             )
         if state == CommandState.FAILED:
             error = status.error
+            error_code = error.error_code if error else "UNKNOWN_ERROR_CODE"
+            error_message = error.message if error else "UNKNOWN_ERROR_MESSAGE"
             raise ServerOperationError(
-                "Command failed: {} {}".format(error.error_code, error.message),
+                "Command failed: {} {}".format(error_code, error_message),
                 {
                     "operation-id": command_id,
                 },

Original file line number	Diff line number	Diff line change
`@@ -410,8 +410,10 @@ def _check_command_not_in_failed_or_closed_state(`
`410`	`410`	`)`
`411`	`411`	`if state == CommandState.FAILED:`
`412`	`412`	`error = status.error`
	`413`	`+ error_code = error.error_code if error else "UNKNOWN_ERROR_CODE"`
	`414`	`+ error_message = error.message if error else "UNKNOWN_ERROR_MESSAGE"`
`413`	`415`	`raise ServerOperationError(`
`414`		`- "Command failed: {} {}".format(error.error_code, error.message),`
	`416`	`+ "Command failed: {} {}".format(error_code, error_message),`
`415`	`417`	`{`
`416`	`418`	`"operation-id": command_id,`
`417`	`419`	`},`