Skip to content

Commit 4baba27

Browse files
committed
migrate to autogenerated google/cloud/bigquery_storage/__init__.py with post processing
1 parent 8cc4bfb commit 4baba27

File tree

1 file changed

+56
-2
lines changed

1 file changed

+56
-2
lines changed

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

Lines changed: 56 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -127,5 +127,59 @@ replacements:
127127
"pyarrow": ["pyarrow>=0.15.0"],
128128
}
129129
count: 1
130-
131-
130+
- paths: [
131+
packages/google-cloud-bigquery-storage/google/cloud/bigquery_storage/__init__.py,
132+
]
133+
before: |
134+
from google.cloud.bigquery_storage_v1.services.big_query_read.async_client import \(
135+
BigQueryReadAsyncClient,
136+
\)
137+
from google.cloud.bigquery_storage_v1.services.big_query_read.client import \(
138+
BigQueryReadClient,
139+
\)
140+
after: |
141+
from google.cloud.bigquery_storage_v1 import BigQueryReadClient
142+
from google.cloud.bigquery_storage_v1 import gapic_types as types
143+
from google.cloud.bigquery_storage_v1.reader import ReadRowsStream
144+
count: 1
145+
- paths: [
146+
packages/google-cloud-bigquery-storage/google/cloud/bigquery_storage/__init__.py,
147+
]
148+
before: |
149+
from google.cloud.bigquery_storage_v1.types.table import TableFieldSchema, TableSchema\n
150+
__all__ = \(
151+
after: |
152+
from google.cloud.bigquery_storage_v1.types.table import TableFieldSchema, TableSchema
153+
from google.cloud.bigquery_storage_v1.writer import AppendRowsStream\n
154+
__all__ = (
155+
count: 1
156+
- paths: [
157+
packages/google-cloud-bigquery-storage/google/cloud/bigquery_storage/__init__.py,
158+
]
159+
before: |
160+
"BigQueryReadClient",
161+
"BigQueryReadAsyncClient",
162+
"BigQueryWriteClient",
163+
"BigQueryWriteAsyncClient",
164+
"ArrowRecordBatch",
165+
after: |
166+
"BigQueryReadClient",
167+
"BigQueryWriteClient",
168+
"BigQueryWriteAsyncClient",
169+
"__version__",
170+
"types",
171+
"ArrowRecordBatch",
172+
count: 1
173+
count: 1
174+
- paths: [
175+
packages/google-cloud-bigquery-storage/google/cloud/bigquery_storage/__init__.py,
176+
]
177+
before: |
178+
"ThrottleState",
179+
"ReadSession",
180+
after: |
181+
"ThrottleState",
182+
"AppendRowsStream",
183+
"ReadRowsStream",
184+
"ReadSession",
185+
count: 1

0 commit comments

Comments
 (0)