Skip to content

Commit 942cfff

Browse files
author
Tobiasz Kędzierski
committed
feat(BigQuery) add support for custom QueryJobConfig in execute cursor method
1 parent ee0f70a commit 942cfff

File tree

1 file changed

+2
-3
lines changed
  • bigquery/google/cloud/bigquery/dbapi

1 file changed

+2
-3
lines changed

bigquery/google/cloud/bigquery/dbapi/cursor.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -116,7 +116,7 @@ def _set_rowcount(self, query_results):
116116
total_rows = num_dml_affected_rows
117117
self.rowcount = total_rows
118118

119-
def execute(self, operation, parameters=None, job_id=None):
119+
def execute(self, operation, parameters=None, job_id=None, config=None):
120120
"""Prepare and execute a database operation.
121121
122122
.. note::
@@ -160,9 +160,8 @@ def execute(self, operation, parameters=None, job_id=None):
160160
formatted_operation = _format_operation(operation, parameters=parameters)
161161
query_parameters = _helpers.to_query_parameters(parameters)
162162

163-
config = job.QueryJobConfig()
163+
config = config or job.QueryJobConfig(use_legacy_sql=False)
164164
config.query_parameters = query_parameters
165-
config.use_legacy_sql = False
166165
self._query_job = client.query(
167166
formatted_operation, job_config=config, job_id=job_id
168167
)

0 commit comments

Comments
 (0)