added section to owlbot.py to preserve pooled transport

googleapis · Mar 18, 2024 · 84585a8 · 84585a8
1 parent 663f35e
commit 84585a8
Showing 1 changed file with 27 additions and 0 deletions.
diff --git a/owlbot.py b/owlbot.py
@@ -218,6 +218,33 @@ def lint_setup_py(session):
 ''',
 )
 
+
+# ----------------------------------------------------------------------------
+# Customize gapics to include PooledBigtableGrpcAsyncIOTransport
+# ----------------------------------------------------------------------------
+
+place_before("google/cloud/bigtable_v2/services/bigtable/client.py",
+    "from .transports.rest import BigtableRestTransport",
+    "from .transports.pooled_grpc_asyncio import PooledBigtableGrpcAsyncIOTransport",
+)
+place_before("google/cloud/bigtable_v2/services/bigtable/client.py",
+    "_transport_registry["rest"] = BigtableRestTransport",
+    "_transport_registry["pooled_grpc_asyncio"] = PooledBigtableGrpcAsyncIOTransport",
+)
+place_before("google/cloud/bigtable_v2/services/bigtable/transports/__init__.py",
+    "from .rest import BigtableRestTransport",
+    "from .pooled_grpc_asyncio import PooledBigtableGrpcAsyncIOTransport",
+)
+place_before("google/cloud/bigtable_v2/services/bigtable/transports/__init__.py",
+    "_transport_registry["rest"] = BigtableRestTransport",
+    "_transport_registry["pooled_grpc_asyncio"] = PooledBigtableGrpcAsyncIOTransport",
+)
+place_before("google/cloud/bigtable_v2/services/bigtable/transports/__init__.py",
+    '"BigtableRestTransport",',
+    '"PooledBigtableGrpcAsyncIOTransport",',
+)
+
+
 # ----------------------------------------------------------------------------
 # Samples templates
 # ----------------------------------------------------------------------------