From 9d8d41620e967c7bb2cef65dd28aaafe975dc533 Mon Sep 17 00:00:00 2001 From: Maxime Beauchemin Date: Wed, 11 Nov 2020 23:09:10 -0800 Subject: [PATCH] fix(DBAPI): keep cursor hot when statement is a script --- google/cloud/bigquery/dbapi/cursor.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/google/cloud/bigquery/dbapi/cursor.py b/google/cloud/bigquery/dbapi/cursor.py index 74f8aec4e..e2819e6d1 100644 --- a/google/cloud/bigquery/dbapi/cursor.py +++ b/google/cloud/bigquery/dbapi/cursor.py @@ -222,7 +222,7 @@ def _try_fetch(self, size=None): is_dml = ( self._query_job.statement_type - and self._query_job.statement_type.upper() != "SELECT" + and self._query_job.statement_type.upper() not in ("SELECT", "SCRIPT") ) if is_dml: self._query_data = iter([])