feat: enable "rest" transport in Python for services supporting numer…

…ic enums (#163) * feat: enable "rest" transport in Python for services supporting numeric enums PiperOrigin-RevId: 508143576 Source-Link: googleapis/googleapis@7a702a9 Source-Link: googleapis/googleapis-gen@6ad1279 Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiNmFkMTI3OWMwZTdhYTc4N2FjNmI2NmM5ZmQ0YTIxMDY5MmVkZmZjZCJ9 * 🦉 Updates from OwlBot post-processor See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md --------- Co-authored-by: Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>
googleapis · Feb 9, 2023 · 43b46e2 · 43b46e2
1 parent 654a072
commit 43b46e2
Show file tree

Hide file tree

Showing 18 changed files with 7,936 additions and 329 deletions.
diff --git a/google/cloud/orchestration/airflow/service_v1/gapic_metadata.json b/google/cloud/orchestration/airflow/service_v1/gapic_metadata.json
@@ -86,6 +86,46 @@
               ]
             }
           }
+        },
+        "rest": {
+          "libraryClient": "EnvironmentsClient",
+          "rpcs": {
+            "CreateEnvironment": {
+              "methods": [
+                "create_environment"
+              ]
+            },
+            "DeleteEnvironment": {
+              "methods": [
+                "delete_environment"
+              ]
+            },
+            "GetEnvironment": {
+              "methods": [
+                "get_environment"
+              ]
+            },
+            "ListEnvironments": {
+              "methods": [
+                "list_environments"
+              ]
+            },
+            "LoadSnapshot": {
+              "methods": [
+                "load_snapshot"
+              ]
+            },
+            "SaveSnapshot": {
+              "methods": [
+                "save_snapshot"
+              ]
+            },
+            "UpdateEnvironment": {
+              "methods": [
+                "update_environment"
+              ]
+            }
+          }
         }
       }
     },
@@ -110,6 +150,16 @@
               ]
             }
           }
+        },
+        "rest": {
+          "libraryClient": "ImageVersionsClient",
+          "rpcs": {
+            "ListImageVersions": {
+              "methods": [
+                "list_image_versions"
+              ]
+            }
+          }
         }
       }
     }

diff --git a/google/cloud/orchestration/airflow/service_v1/services/environments/client.py b/google/cloud/orchestration/airflow/service_v1/services/environments/client.py
@@ -60,6 +60,7 @@
 from .transports.base import DEFAULT_CLIENT_INFO, EnvironmentsTransport
 from .transports.grpc import EnvironmentsGrpcTransport
 from .transports.grpc_asyncio import EnvironmentsGrpcAsyncIOTransport
+from .transports.rest import EnvironmentsRestTransport
 
 
 class EnvironmentsClientMeta(type):
@@ -73,6 +74,7 @@ class EnvironmentsClientMeta(type):
     _transport_registry = OrderedDict()  # type: Dict[str, Type[EnvironmentsTransport]]
     _transport_registry["grpc"] = EnvironmentsGrpcTransport
     _transport_registry["grpc_asyncio"] = EnvironmentsGrpcAsyncIOTransport
+    _transport_registry["rest"] = EnvironmentsRestTransport
 
     def get_transport_class(
         cls,

diff --git a/google/cloud/orchestration/airflow/service_v1/services/environments/transports/__init__.py b/google/cloud/orchestration/airflow/service_v1/services/environments/transports/__init__.py
@@ -19,14 +19,18 @@
 from .base import EnvironmentsTransport
 from .grpc import EnvironmentsGrpcTransport
 from .grpc_asyncio import EnvironmentsGrpcAsyncIOTransport
+from .rest import EnvironmentsRestInterceptor, EnvironmentsRestTransport
 
 # Compile a registry of transports.
 _transport_registry = OrderedDict()  # type: Dict[str, Type[EnvironmentsTransport]]
 _transport_registry["grpc"] = EnvironmentsGrpcTransport
 _transport_registry["grpc_asyncio"] = EnvironmentsGrpcAsyncIOTransport
+_transport_registry["rest"] = EnvironmentsRestTransport
 
 __all__ = (
     "EnvironmentsTransport",
     "EnvironmentsGrpcTransport",
     "EnvironmentsGrpcAsyncIOTransport",
+    "EnvironmentsRestTransport",
+    "EnvironmentsRestInterceptor",
 )