Merge remote-tracking branch 'migration/main' into bqs-samples

Author: Linchin

bigquery_storage/AUTHORING_GUIDE.md

@@ -0,0 +1 @@
+See https://github.com/GoogleCloudPlatform/python-docs-samples/blob/main/AUTHORING_GUIDE.md

bigquery_storage/CONTRIBUTING.md

@@ -0,0 +1 @@
+See https://github.com/GoogleCloudPlatform/python-docs-samples/blob/main/CONTRIBUTING.md

bigquery_storage/__init__.py

@@ -0,0 +1,45 @@
+# Copyright 2021 Google LLC
+#
+# 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 datetime
+import os
+import random
+
+import pytest
+
+
+@pytest.fixture(scope="session")
+def project_id():
+    return os.environ["GOOGLE_CLOUD_PROJECT"]
+
+
+@pytest.fixture(scope="session")
+def dataset(project_id):
+    from google.cloud import bigquery
+
+    client = bigquery.Client()
+
+    # Add a random suffix to dataset name to avoid conflict, because we run
+    # a samples test on each supported Python version almost at the same time.
+    dataset_time = datetime.datetime.now().strftime("%y%m%d_%H%M%S")
+    suffix = f"_{(random.randint(0, 99)):02d}"
+    dataset_name = "samples_tests_" + dataset_time + suffix
+
+    dataset_id = "{}.{}".format(project_id, dataset_name)
+    dataset = bigquery.Dataset(dataset_id)
+    dataset.location = "us-east7"
+    created_dataset = client.create_dataset(dataset)
+    yield created_dataset
+
+    client.delete_dataset(created_dataset, delete_contents=True)

bigquery_storage/conftest.py

@@ -0,0 +1,53 @@
+# -*- coding: utf-8 -*-
+# Copyright 2025 Google LLC
+#
+# 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.
+#
+# Generated code. DO NOT EDIT!
+#
+# Snippet for CreateReadSession
+# NOTE: This snippet has been automatically generated for illustrative purposes only.
+# It may require modifications to work in your environment.
+
+# To install the latest published package dependency, execute the following:
+#   python3 -m pip install google-cloud-bigquery-storage
+
+
+# [START bigquerystorage_v1_generated_BigQueryRead_CreateReadSession_async]
+# This snippet has been automatically generated and should be regarded as a
+# code template only.
+# It will require modifications to work:
+# - It may require correct/in-range values for request initialization.
+# - It may require specifying regional endpoints when creating the service
+#   client as shown in:
+#   https://googleapis.dev/python/google-api-core/latest/client_options.html
+from google.cloud import bigquery_storage_v1
+
+
+async def sample_create_read_session():
+    # Create a client
+    client = bigquery_storage_v1.BigQueryReadAsyncClient()
+
+    # Initialize request argument(s)
+    request = bigquery_storage_v1.CreateReadSessionRequest(
+        parent="parent_value",
+    )
+
+    # Make the request
+    response = await client.create_read_session(request=request)
+
+    # Handle the response
+    print(response)
+
+
+# [END bigquerystorage_v1_generated_BigQueryRead_CreateReadSession_async]

bigquery_storage/generated_samples/bigquerystorage_v1_generated_big_query_read_create_read_session_async.py

@@ -0,0 +1,53 @@
+# -*- coding: utf-8 -*-
+# Copyright 2025 Google LLC
+#
+# 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.
+#
+# Generated code. DO NOT EDIT!
+#
+# Snippet for CreateReadSession
+# NOTE: This snippet has been automatically generated for illustrative purposes only.
+# It may require modifications to work in your environment.
+
+# To install the latest published package dependency, execute the following:
+#   python3 -m pip install google-cloud-bigquery-storage
+
+
+# [START bigquerystorage_v1_generated_BigQueryRead_CreateReadSession_sync]
+# This snippet has been automatically generated and should be regarded as a
+# code template only.
+# It will require modifications to work:
+# - It may require correct/in-range values for request initialization.
+# - It may require specifying regional endpoints when creating the service
+#   client as shown in:
+#   https://googleapis.dev/python/google-api-core/latest/client_options.html
+from google.cloud import bigquery_storage_v1
+
+
+def sample_create_read_session():
+    # Create a client
+    client = bigquery_storage_v1.BigQueryReadClient()
+
+    # Initialize request argument(s)
+    request = bigquery_storage_v1.CreateReadSessionRequest(
+        parent="parent_value",
+    )
+
+    # Make the request
+    response = client.create_read_session(request=request)
+
+    # Handle the response
+    print(response)
+
+
+# [END bigquerystorage_v1_generated_BigQueryRead_CreateReadSession_sync]

bigquery_storage/generated_samples/bigquerystorage_v1_generated_big_query_read_create_read_session_sync.py

@@ -0,0 +1,54 @@
+# -*- coding: utf-8 -*-
+# Copyright 2025 Google LLC
+#
+# 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.
+#
+# Generated code. DO NOT EDIT!
+#
+# Snippet for ReadRows
+# NOTE: This snippet has been automatically generated for illustrative purposes only.
+# It may require modifications to work in your environment.
+
+# To install the latest published package dependency, execute the following:
+#   python3 -m pip install google-cloud-bigquery-storage
+
+
+# [START bigquerystorage_v1_generated_BigQueryRead_ReadRows_async]
+# This snippet has been automatically generated and should be regarded as a
+# code template only.
+# It will require modifications to work:
+# - It may require correct/in-range values for request initialization.
+# - It may require specifying regional endpoints when creating the service
+#   client as shown in:
+#   https://googleapis.dev/python/google-api-core/latest/client_options.html
+from google.cloud import bigquery_storage_v1
+
+
+async def sample_read_rows():
+    # Create a client
+    client = bigquery_storage_v1.BigQueryReadAsyncClient()
+
+    # Initialize request argument(s)
+    request = bigquery_storage_v1.ReadRowsRequest(
+        read_stream="read_stream_value",
+    )
+
+    # Make the request
+    stream = await client.read_rows(request=request)
+
+    # Handle the response
+    async for response in stream:
+        print(response)
+
+
+# [END bigquerystorage_v1_generated_BigQueryRead_ReadRows_async]

bigquery_storage/generated_samples/bigquerystorage_v1_generated_big_query_read_read_rows_async.py

@@ -0,0 +1,54 @@
+# -*- coding: utf-8 -*-
+# Copyright 2025 Google LLC
+#
+# 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.
+#
+# Generated code. DO NOT EDIT!
+#
+# Snippet for ReadRows
+# NOTE: This snippet has been automatically generated for illustrative purposes only.
+# It may require modifications to work in your environment.
+
+# To install the latest published package dependency, execute the following:
+#   python3 -m pip install google-cloud-bigquery-storage
+
+
+# [START bigquerystorage_v1_generated_BigQueryRead_ReadRows_sync]
+# This snippet has been automatically generated and should be regarded as a
+# code template only.
+# It will require modifications to work:
+# - It may require correct/in-range values for request initialization.
+# - It may require specifying regional endpoints when creating the service
+#   client as shown in:
+#   https://googleapis.dev/python/google-api-core/latest/client_options.html
+from google.cloud import bigquery_storage_v1
+
+
+def sample_read_rows():
+    # Create a client
+    client = bigquery_storage_v1.BigQueryReadClient()
+
+    # Initialize request argument(s)
+    request = bigquery_storage_v1.ReadRowsRequest(
+        read_stream="read_stream_value",
+    )
+
+    # Make the request
+    stream = client.read_rows(request=request)
+
+    # Handle the response
+    for response in stream:
+        print(response)
+
+
+# [END bigquerystorage_v1_generated_BigQueryRead_ReadRows_sync]

bigquery_storage/generated_samples/bigquerystorage_v1_generated_big_query_read_read_rows_sync.py

@@ -0,0 +1,53 @@
+# -*- coding: utf-8 -*-
+# Copyright 2025 Google LLC
+#
+# 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.
+#
+# Generated code. DO NOT EDIT!
+#
+# Snippet for SplitReadStream
+# NOTE: This snippet has been automatically generated for illustrative purposes only.
+# It may require modifications to work in your environment.
+
+# To install the latest published package dependency, execute the following:
+#   python3 -m pip install google-cloud-bigquery-storage
+
+
+# [START bigquerystorage_v1_generated_BigQueryRead_SplitReadStream_async]
+# This snippet has been automatically generated and should be regarded as a
+# code template only.
+# It will require modifications to work:
+# - It may require correct/in-range values for request initialization.
+# - It may require specifying regional endpoints when creating the service
+#   client as shown in:
+#   https://googleapis.dev/python/google-api-core/latest/client_options.html
+from google.cloud import bigquery_storage_v1
+
+
+async def sample_split_read_stream():
+    # Create a client
+    client = bigquery_storage_v1.BigQueryReadAsyncClient()
+
+    # Initialize request argument(s)
+    request = bigquery_storage_v1.SplitReadStreamRequest(
+        name="name_value",
+    )
+
+    # Make the request
+    response = await client.split_read_stream(request=request)
+
+    # Handle the response
+    print(response)
+
+
+# [END bigquerystorage_v1_generated_BigQueryRead_SplitReadStream_async]

bigquery_storage/generated_samples/bigquerystorage_v1_generated_big_query_read_split_read_stream_async.py

@@ -0,0 +1,53 @@
+# -*- coding: utf-8 -*-
+# Copyright 2025 Google LLC
+#
+# 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.
+#
+# Generated code. DO NOT EDIT!
+#
+# Snippet for SplitReadStream
+# NOTE: This snippet has been automatically generated for illustrative purposes only.
+# It may require modifications to work in your environment.
+
+# To install the latest published package dependency, execute the following:
+#   python3 -m pip install google-cloud-bigquery-storage
+
+
+# [START bigquerystorage_v1_generated_BigQueryRead_SplitReadStream_sync]
+# This snippet has been automatically generated and should be regarded as a
+# code template only.
+# It will require modifications to work:
+# - It may require correct/in-range values for request initialization.
+# - It may require specifying regional endpoints when creating the service
+#   client as shown in:
+#   https://googleapis.dev/python/google-api-core/latest/client_options.html
+from google.cloud import bigquery_storage_v1
+
+
+def sample_split_read_stream():
+    # Create a client
+    client = bigquery_storage_v1.BigQueryReadClient()
+
+    # Initialize request argument(s)
+    request = bigquery_storage_v1.SplitReadStreamRequest(
+        name="name_value",
+    )
+
+    # Make the request
+    response = client.split_read_stream(request=request)
+
+    # Handle the response
+    print(response)
+
+
+# [END bigquerystorage_v1_generated_BigQueryRead_SplitReadStream_sync]