fix(DBAPI): keep cursor hot when statement is a script

mistercrunch · mistercrunch · commit 9d8d41620e96 · 2020-11-11T23:12:36.000-08:00
diff --git 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([])

Original file line number	Diff line number	Diff line change
`@@ -222,7 +222,7 @@ def _try_fetch(self, size=None):`
`222`	`222`
`223`	`223`	`is_dml = (`
`224`	`224`	`self._query_job.statement_type`
`225`		`- and self._query_job.statement_type.upper() != "SELECT"`
	`225`	`+ and self._query_job.statement_type.upper() not in ("SELECT", "SCRIPT")`
`226`	`226`	`)`
`227`	`227`	`if is_dml:`
`228`	`228`	`self._query_data = iter([])`