migrate to autogenerated google/cloud/bigquery_storage_v1beta2/__init__.py with post processing

Author: parthea

packages/google-cloud-bigquery-storage/google/cloud/bigquery_storage_v1beta2/__init__.py

@@ -1,21 +1,18 @@
 # -*- coding: utf-8 -*-
-#
-# Copyright 2020 Google LLC
+# 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
 #
-#     https://www.apache.org/licenses/LICENSE-2.0
+#     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 __future__ import absolute_import
-
+#
 from google.cloud.bigquery_storage_v1beta2 import gapic_version as package_version
 
 __version__ = package_version.__version__

scripts/client-post-processing/bigquery-storage-integration.yaml

@@ -343,3 +343,90 @@ replacements:
     after: |
       __version__ = package_version.__version__
     count: 2
+    # Given that this file is mostly handwritten, we could omit the file completely when we migrate to librarian
+    # See `preserve_regex` in https://github.com/googleapis/librarian/blob/main/doc/language-onboarding.md#generate
+  - paths: [
+      packages/google-cloud-bigquery-storage/google/cloud/bigquery_storage_v1beta2/__init__.py,
+    ]
+    before: |
+      __version__ = package_version.__version__\n\n
+      from .services.big_query_read import BigQueryReadAsyncClient, BigQueryReadClient
+      from .services.big_query_write import BigQueryWriteAsyncClient, BigQueryWriteClient
+      from .types.arrow import ArrowRecordBatch, ArrowSchema, ArrowSerializationOptions
+      from .types.avro import AvroRows, AvroSchema
+      from .types.protobuf import ProtoRows, ProtoSchema
+      from .types.storage import \(
+          AppendRowsRequest,
+          AppendRowsResponse,
+          BatchCommitWriteStreamsRequest,
+          BatchCommitWriteStreamsResponse,
+          CreateReadSessionRequest,
+          CreateWriteStreamRequest,
+          FinalizeWriteStreamRequest,
+          FinalizeWriteStreamResponse,
+          FlushRowsRequest,
+          FlushRowsResponse,
+          GetWriteStreamRequest,
+          ReadRowsRequest,
+          ReadRowsResponse,
+          SplitReadStreamRequest,
+          SplitReadStreamResponse,
+          StorageError,
+          StreamStats,
+          ThrottleState,
+      \)
+      from .types.stream import DataFormat, ReadSession, ReadStream, WriteStream
+      from .types.table import TableFieldSchema, TableSchema\n
+      __all__ = \(
+          "BigQueryReadAsyncClient",
+          "BigQueryWriteAsyncClient",
+          "AppendRowsRequest",
+          "AppendRowsResponse",
+          "ArrowRecordBatch",
+          "ArrowSchema",
+          "ArrowSerializationOptions",
+          "AvroRows",
+          "AvroSchema",
+          "BatchCommitWriteStreamsRequest",
+          "BatchCommitWriteStreamsResponse",
+          "BigQueryReadClient",
+          "BigQueryWriteClient",
+          "CreateReadSessionRequest",
+          "CreateWriteStreamRequest",
+          "DataFormat",
+          "FinalizeWriteStreamRequest",
+          "FinalizeWriteStreamResponse",
+          "FlushRowsRequest",
+          "FlushRowsResponse",
+          "GetWriteStreamRequest",
+          "ProtoRows",
+          "ProtoSchema",
+          "ReadRowsRequest",
+          "ReadRowsResponse",
+          "ReadSession",
+          "ReadStream",
+          "SplitReadStreamRequest",
+          "SplitReadStreamResponse",
+          "StorageError",
+          "StreamStats",
+          "TableFieldSchema",
+          "TableSchema",
+          "ThrottleState",
+          "WriteStream",
+      \)
+    after: |
+      __version__ = package_version.__version__\n
+      from google.cloud.bigquery_storage_v1beta2 import client, types\n\n
+      class BigQueryReadClient(client.BigQueryReadClient):
+          __doc__ = client.BigQueryReadClient.__doc__\n\n
+      class BigQueryWriteClient(client.BigQueryWriteClient):
+          __doc__ = client.BigQueryWriteClient.__doc__\n\n
+      __all__ = (
+          # google.cloud.bigquery_storage_v1beta2
+          "__version__",
+          "types",
+          # google.cloud.bigquery_storage_v1beta2.client
+          "BigQueryReadClient",
+          "BigQueryWriteClient",
+      )
+    count: 1