-
Notifications
You must be signed in to change notification settings - Fork 6.4k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Add veneer samples for BigQuery. (#444)
Put them in their own directory, since they duplicate the content from the existing samples that use the generated libraries.
- Loading branch information
Showing
7 changed files
with
211 additions
and
2 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,5 @@ | ||
# BigQuery Google Cloud Client Library Samples | ||
|
||
<!-- auto-doc-link --> | ||
<!-- end-auto-doc-link --> | ||
|
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,83 @@ | ||
#!/usr/bin/env python | ||
|
||
# Copyright 2016 Google Inc. All Rights Reserved. | ||
# | ||
# Licensed under the Apache License, Version 2.0 (the "License"); | ||
# you may not use this file except in compliance with the License. | ||
# You may obtain a copy of the License at | ||
# | ||
# http://www.apache.org/licenses/LICENSE-2.0 | ||
# | ||
# Unless required by applicable law or agreed to in writing, software | ||
# distributed under the License is distributed on an "AS IS" BASIS, | ||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | ||
# See the License for the specific language governing permissions and | ||
# limitations under the License. | ||
|
||
"""Command-line application to perform asynchronous queries in BigQuery. | ||
For more information, see the README.md under /bigquery. | ||
Example invocation: | ||
$ python async_query.py \ | ||
'SELECT corpus FROM `publicdata.samples.shakespeare` GROUP BY corpus' | ||
""" | ||
|
||
import argparse | ||
import time | ||
import uuid | ||
|
||
from gcloud import bigquery | ||
|
||
|
||
def async_query(query): | ||
client = bigquery.Client() | ||
query_job = client.run_async_query(str(uuid.uuid4()), query) | ||
query_job.use_legacy_sql = False | ||
query_job.begin() | ||
|
||
wait_for_job(query_job) | ||
|
||
# Manually construct the QueryResults. | ||
# TODO: The client library will provide a helper method that does this. | ||
# https://github.com/GoogleCloudPlatform/gcloud-python/issues/2083 | ||
query_results = bigquery.query.QueryResults('', client) | ||
query_results._properties['jobReference'] = { | ||
'jobId': query_job.name, | ||
'projectId': query_job.project | ||
} | ||
|
||
# Drain the query results by requesting a page at a time. | ||
page_token = None | ||
|
||
while True: | ||
rows, total_rows, page_token = query_results.fetch_data( | ||
max_results=10, | ||
page_token=page_token) | ||
|
||
for row in rows: | ||
print(row) | ||
|
||
if not page_token: | ||
break | ||
|
||
|
||
def wait_for_job(job): | ||
while True: | ||
job.reload() # Refreshes the state via a GET request. | ||
if job.state == 'DONE': | ||
if job.error_result: | ||
raise RuntimeError(job.error_result) | ||
return | ||
time.sleep(1) | ||
|
||
|
||
if __name__ == '__main__': | ||
parser = argparse.ArgumentParser( | ||
description=__doc__, | ||
formatter_class=argparse.RawDescriptionHelpFormatter) | ||
parser.add_argument('query', help='BigQuery SQL Query.') | ||
|
||
args = parser.parse_args() | ||
|
||
async_query(args.query) |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,27 @@ | ||
# Copyright 2016 Google Inc. All Rights Reserved. | ||
# | ||
# Licensed under the Apache License, Version 2.0 (the "License"); | ||
# you may not use this file except in compliance with the License. | ||
# You may obtain a copy of the License at | ||
# | ||
# http://www.apache.org/licenses/LICENSE-2.0 | ||
# | ||
# Unless required by applicable law or agreed to in writing, software | ||
# distributed under the License is distributed on an "AS IS" BASIS, | ||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | ||
# See the License for the specific language governing permissions and | ||
# limitations under the License. | ||
|
||
from async_query import async_query | ||
|
||
|
||
def test_async_query(cloud_config, capsys): | ||
query = ( | ||
'SELECT corpus FROM `publicdata.samples.shakespeare` ' | ||
'GROUP BY corpus;') | ||
|
||
async_query(query) | ||
|
||
out, _ = capsys.readouterr() | ||
|
||
assert 'romeoandjuliet' in out |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1 @@ | ||
gcloud==0.18.1 |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,66 @@ | ||
#!/usr/bin/env python | ||
|
||
# Copyright 2016 Google Inc. All Rights Reserved. | ||
# | ||
# Licensed under the Apache License, Version 2.0 (the "License"); | ||
# you may not use this file except in compliance with the License. | ||
# You may obtain a copy of the License at | ||
# | ||
# http://www.apache.org/licenses/LICENSE-2.0 | ||
# | ||
# Unless required by applicable law or agreed to in writing, software | ||
# distributed under the License is distributed on an "AS IS" BASIS, | ||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | ||
# See the License for the specific language governing permissions and | ||
# limitations under the License. | ||
|
||
"""Command-line application to perform synchronous queries in BigQuery. | ||
For more information, see the README.md under /bigquery. | ||
Example invocation: | ||
$ python sync_query.py \ | ||
'SELECT corpus FROM `publicdata.samples.shakespeare` GROUP BY corpus' | ||
""" | ||
|
||
import argparse | ||
|
||
# [START sync_query] | ||
from gcloud import bigquery | ||
|
||
|
||
def sync_query(query): | ||
client = bigquery.Client() | ||
query_results = client.run_sync_query(query) | ||
|
||
# Use standard SQL syntax for queries. | ||
# See: https://cloud.google.com/bigquery/sql-reference/ | ||
query_results.use_legacy_sql = False | ||
|
||
query_results.run() | ||
|
||
# Drain the query results by requesting a page at a time. | ||
page_token = None | ||
|
||
while True: | ||
rows, total_rows, page_token = query_results.fetch_data( | ||
max_results=10, | ||
page_token=page_token) | ||
|
||
for row in rows: | ||
print(row) | ||
|
||
if not page_token: | ||
break | ||
# [END sync_query] | ||
|
||
|
||
if __name__ == '__main__': | ||
parser = argparse.ArgumentParser( | ||
description=__doc__, | ||
formatter_class=argparse.RawDescriptionHelpFormatter) | ||
parser.add_argument('query', help='BigQuery SQL Query.') | ||
|
||
args = parser.parse_args() | ||
|
||
sync_query(args.query) |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,27 @@ | ||
# Copyright 2016 Google Inc. All Rights Reserved. | ||
# | ||
# Licensed under the Apache License, Version 2.0 (the "License"); | ||
# you may not use this file except in compliance with the License. | ||
# You may obtain a copy of the License at | ||
# | ||
# http://www.apache.org/licenses/LICENSE-2.0 | ||
# | ||
# Unless required by applicable law or agreed to in writing, software | ||
# distributed under the License is distributed on an "AS IS" BASIS, | ||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | ||
# See the License for the specific language governing permissions and | ||
# limitations under the License. | ||
|
||
from sync_query import sync_query | ||
|
||
|
||
def test_sync_query(cloud_config, capsys): | ||
query = ( | ||
'SELECT corpus FROM `publicdata.samples.shakespeare` ' | ||
'GROUP BY corpus;') | ||
|
||
sync_query(query) | ||
|
||
out, _ = capsys.readouterr() | ||
|
||
assert 'romeoandjuliet' in out |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters