BigQuery parameterized query sample [(#699)](GoogleCloudPlatform/pyth…

…on-docs-samples#699)

Available in latest version of google-cloud-bigquery (0.22.0), so
upgrades the requirement, too.

samples/snippets/sync_query_params.py

@@ -0,0 +1,120 @@
+#!/usr/bin/env python
+
+# Copyright 2016 Google Inc.
+#
+# 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 app to perform synchronous queries with parameters in BigQuery.
+
+For more information, see the README.md under /bigquery.
+
+Example invocation:
+    $ python sync_query_params.py --use-named-params 'romeoandjuliet' 100
+    $ python sync_query_params.py --use-positional-params 'romeoandjuliet' 100
+"""
+
+import argparse
+
+from google.cloud import bigquery
+
+
+def print_results(query_results):
+    """Print 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 sync_query_positional_params(corpus, min_word_count):
+    client = bigquery.Client()
+    query_results = client.run_sync_query(
+        """SELECT word, word_count
+        FROM `bigquery-public-data.samples.shakespeare`
+        WHERE corpus = ?
+        AND word_count >= ?
+        ORDER BY word_count DESC;
+        """,
+        query_parameters=(
+            bigquery.ScalarQueryParameter(
+                # Set the name to None to use positional parameters (? symbol
+                # in the query).  Note that you cannot mix named and positional
+                # parameters.
+                None,
+                'STRING',
+                corpus),
+            bigquery.ScalarQueryParameter(None, 'INT64', min_word_count)))
+
+    # Only standard SQL syntax supports parameters in queries.
+    # See: https://cloud.google.com/bigquery/sql-reference/
+    query_results.use_legacy_sql = False
+    query_results.run()
+    print_results(query_results)
+
+
+def sync_query_named_params(corpus, min_word_count):
+    client = bigquery.Client()
+    query_results = client.run_sync_query(
+        """SELECT word, word_count
+        FROM `bigquery-public-data.samples.shakespeare`
+        WHERE corpus = @corpus
+        AND word_count >= @min_word_count
+        ORDER BY word_count DESC;
+        """,
+        query_parameters=(
+            bigquery.ScalarQueryParameter('corpus', 'STRING', corpus),
+            bigquery.ScalarQueryParameter(
+                'min_word_count',
+                'INT64',
+                min_word_count)))
+    query_results.use_legacy_sql = False
+    query_results.run()
+    print_results(query_results)
+
+
+if __name__ == '__main__':
+    parser = argparse.ArgumentParser(
+        description=__doc__,
+        formatter_class=argparse.RawDescriptionHelpFormatter)
+    parser.add_argument(
+        'corpus',
+        help='Corpus to search from Shakespeare dataset.')
+    parser.add_argument(
+        'min_word_count',
+        help='Minimum count of words to query.',
+        type=int)
+
+    params_type_parser = parser.add_mutually_exclusive_group(required=False)
+    params_type_parser.add_argument(
+        '--use-named-params',
+        dest='use_named_params',
+        action='store_true')
+    params_type_parser.add_argument(
+        '--use-positional-params',
+        dest='use_named_params',
+        action='store_false')
+    parser.set_defaults(use_named_params=False)
+    args = parser.parse_args()
+
+    if args.use_named_params:
+        sync_query_named_params(args.corpus, args.min_word_count)
+    else:
+        sync_query_positional_params(args.corpus, args.min_word_count)

samples/snippets/sync_query_params_test.py

@@ -0,0 +1,31 @@
+# Copyright 2016 Google Inc.
+#
+# 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.
+
+import sync_query_params
+
+
+def test_sync_query_named_params(cloud_config, capsys):
+    sync_query_params.sync_query_named_params(
+        corpus='romeoandjuliet',
+        min_word_count=100)
+    out, _ = capsys.readouterr()
+    assert 'love' in out
+
+
+def test_sync_query_positional_params(cloud_config, capsys):
+    sync_query_params.sync_query_positional_params(
+        corpus='romeoandjuliet',
+        min_word_count=100)
+    out, _ = capsys.readouterr()
+    assert 'love' in out