Skip to content

Commit dad9288

Browse files
committed
Change parameter to kwargs
1 parent 55bf05c commit dad9288

File tree

1 file changed

+12
-14
lines changed

1 file changed

+12
-14
lines changed

pandas/io/gbq.py

Lines changed: 12 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -375,7 +375,7 @@ def process_insert_errors(self, insert_errors):
375375

376376
raise StreamingInsertError
377377

378-
def run_query(self, query, udf_resource_uri=None):
378+
def run_query(self, query, **kwargs):
379379
try:
380380
from googleapiclient.errors import HttpError
381381
except:
@@ -395,13 +395,9 @@ def run_query(self, query, udf_resource_uri=None):
395395
}
396396
}
397397
}
398-
399-
if udf_resource_uri is not None:
400-
if not isinstance(udf_resource_uri, list):
401-
udf_resource_uri = [udf_resource_uri]
402-
403-
job_data['configuration']['query']['userDefinedFunctionResources'] = \
404-
[{'resourceUri': uri} for uri in udf_resource_uri]
398+
query_config = kwargs.get('query_config')
399+
if query_config is not None:
400+
job_data['configuration']['query'].update(query_config)
405401

406402

407403
self._start_timer()
@@ -629,8 +625,9 @@ def _parse_entry(field_value, field_type):
629625
return field_value
630626

631627

632-
def read_gbq(query, project_id=None, index_col=None, col_order=None,
633-
reauth=False, verbose=True, private_key=None, dialect='legacy', udf_resource_uri=None):
628+
def read_gbq(query, project_id=None, index_col=None, col_order=None,
629+
reauth=False, verbose=True, private_key=None, dialect='legacy',
630+
**kwargs):
634631
"""Load data from Google BigQuery.
635632
636633
THIS IS AN EXPERIMENTAL LIBRARY
@@ -690,9 +687,10 @@ def read_gbq(query, project_id=None, index_col=None, col_order=None,
690687
691688
.. versionadded:: 0.19.0
692689
693-
udf_resource_uri : list(str) or str (optional)
694-
A code resource to load from a Google Cloud Storage URI.
695-
Describes user-defined function resources used in the query.
690+
**kwargs: Arbitrary keyword arguments
691+
query_config (dict): query configuration parameters for job processing.
692+
For more information see `BigQuery SQL Reference
693+
<https://cloud.google.com/bigquery/docs/reference/rest/v2/jobs#configuration.query>`
696694
697695
.. versionadded:: 0.19.0
698696
@@ -712,7 +710,7 @@ def read_gbq(query, project_id=None, index_col=None, col_order=None,
712710
connector = GbqConnector(project_id, reauth=reauth, verbose=verbose,
713711
private_key=private_key,
714712
dialect=dialect)
715-
schema, pages = connector.run_query(query, udf_resource_uri)
713+
schema, pages = connector.run_query(query, **kwargs)
716714
dataframe_list = []
717715
while len(pages) > 0:
718716
page = pages.pop()

0 commit comments

Comments
 (0)