Commitbf6d41c

committed

ensure command_id is not None

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

1 parent726abe7 commitbf6d41cCopy full SHA for bf6d41c

File tree

-0

lines changed

-0

lines changed

Lines changed: 7 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -796,6 +796,8 @@ def _results_message_to_execute_response(self, resp, operation_state):`
`796`	`796`	`arrow_queue_opt=None`
`797`	`797`
`798`	`798`	`command_id=CommandId.from_thrift_handle(resp.operationHandle)`
	`799`	`+ifcommand_idisNone:`
	`800`	`+raiseValueError(f"Invalid Thrift handle:{resp.operationHandle}")`
`799`	`801`
`800`	`802`	`returnExecuteResponse(`
`801`	`803`	`arrow_queue=arrow_queue_opt,`
`@@ -1156,6 +1158,8 @@ def get_columns(`
`1156`	`1158`
`1157`	`1159`	`def_handle_execute_response(self,resp,cursor):`
`1158`	`1160`	`command_id=CommandId.from_thrift_handle(resp.operationHandle)`
	`1161`	`+ifcommand_idisNone:`
	`1162`	`+raiseValueError(f"Invalid Thrift handle:{resp.operationHandle}")`
`1159`	`1163`
`1160`	`1164`	`cursor.active_command_id=command_id`
`1161`	`1165`	`self._check_direct_results_for_error(resp.directResults)`
`@@ -1169,6 +1173,9 @@ def _handle_execute_response(self, resp, cursor):`
`1169`	`1173`
`1170`	`1174`	`def_handle_execute_response_async(self,resp,cursor):`
`1171`	`1175`	`command_id=CommandId.from_thrift_handle(resp.operationHandle)`
	`1176`	`+ifcommand_idisNone:`
	`1177`	`+raiseValueError(f"Invalid Thrift handle:{resp.operationHandle}")`
	`1178`	`+`
`1172`	`1179`	`cursor.active_command_id=command_id`
`1173`	`1180`	`self._check_direct_results_for_error(resp.directResults)`
`1174`	`1181`

Comments

(0)