feat: add context manager support in client (#32)

- [ ] Regenerate this pull request now.

chore: fix docstring for first attribute of protos

committer: @busunkim96
PiperOrigin-RevId: 401271153

Source-Link: googleapis/googleapis@787f8c9

Source-Link: googleapis/googleapis-gen@81decff
Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiODFkZWNmZmU5ZmM3MjM5NmE4MTUzZTc1NmQxZDY3YTZlZWNmZDYyMCJ9

google/cloud/orchestration/airflow/service_v1/services/environments/async_client.py

@@ -759,6 +759,12 @@ async def delete_environment(
         # Done; return the response.
         return response
 
+    async def __aenter__(self):
+        return self
+
+    async def __aexit__(self, exc_type, exc, tb):
+        await self.transport.close()
+
 
 try:
     DEFAULT_CLIENT_INFO = gapic_v1.client_info.ClientInfo(

google/cloud/orchestration/airflow/service_v1/services/environments/client.py

@@ -348,10 +348,7 @@ def __init__(
                 client_cert_source_for_mtls=client_cert_source_func,
                 quota_project_id=client_options.quota_project_id,
                 client_info=client_info,
-                always_use_jwt_access=(
-                    Transport == type(self).get_transport_class("grpc")
-                    or Transport == type(self).get_transport_class("grpc_asyncio")
-                ),
+                always_use_jwt_access=True,
             )
 
     def create_environment(
@@ -950,6 +947,19 @@ def delete_environment(
         # Done; return the response.
         return response
 
+    def __enter__(self):
+        return self
+
+    def __exit__(self, type, value, traceback):
+        """Releases underlying transport's resources.
+
+        .. warning::
+            ONLY use as a context manager if the transport is NOT shared
+            with other clients! Exiting the with block will CLOSE the transport
+            and may cause errors in other clients!
+        """
+        self.transport.close()
+
 
 try:
     DEFAULT_CLIENT_INFO = gapic_v1.client_info.ClientInfo(

google/cloud/orchestration/airflow/service_v1/services/environments/transports/base.py

@@ -173,6 +173,15 @@ def _prep_wrapped_messages(self, client_info):
             ),
         }
 
+    def close(self):
+        """Closes resources associated with the transport.
+
+       .. warning::
+            Only call this method if the transport is NOT shared
+            with other clients - this may cause errors in other clients!
+        """
+        raise NotImplementedError()
+
     @property
     def operations_client(self) -> operations_v1.OperationsClient:
         """Return the client designed to process long-running operations."""

google/cloud/orchestration/airflow/service_v1/services/environments/transports/grpc.py

@@ -374,5 +374,8 @@ def delete_environment(
             )
         return self._stubs["delete_environment"]
 
+    def close(self):
+        self.grpc_channel.close()
+
 
 __all__ = ("EnvironmentsGrpcTransport",)

google/cloud/orchestration/airflow/service_v1/services/environments/transports/grpc_asyncio.py

@@ -388,5 +388,8 @@ def delete_environment(
             )
         return self._stubs["delete_environment"]
 
+    def close(self):
+        return self.grpc_channel.close()
+
 
 __all__ = ("EnvironmentsGrpcAsyncIOTransport",)

google/cloud/orchestration/airflow/service_v1/services/image_versions/async_client.py

@@ -240,6 +240,12 @@ async def list_image_versions(
         # Done; return the response.
         return response
 
+    async def __aenter__(self):
+        return self
+
+    async def __aexit__(self, exc_type, exc, tb):
+        await self.transport.close()
+
 
 try:
     DEFAULT_CLIENT_INFO = gapic_v1.client_info.ClientInfo(

google/cloud/orchestration/airflow/service_v1/services/image_versions/client.py

@@ -326,10 +326,7 @@ def __init__(
                 client_cert_source_for_mtls=client_cert_source_func,
                 quota_project_id=client_options.quota_project_id,
                 client_info=client_info,
-                always_use_jwt_access=(
-                    Transport == type(self).get_transport_class("grpc")
-                    or Transport == type(self).get_transport_class("grpc_asyncio")
-                ),
+                always_use_jwt_access=True,
             )
 
     def list_image_versions(
@@ -413,6 +410,19 @@ def list_image_versions(
         # Done; return the response.
         return response
 
+    def __enter__(self):
+        return self
+
+    def __exit__(self, type, value, traceback):
+        """Releases underlying transport's resources.
+
+        .. warning::
+            ONLY use as a context manager if the transport is NOT shared
+            with other clients! Exiting the with block will CLOSE the transport
+            and may cause errors in other clients!
+        """
+        self.transport.close()
+
 
 try:
     DEFAULT_CLIENT_INFO = gapic_v1.client_info.ClientInfo(

google/cloud/orchestration/airflow/service_v1/services/image_versions/transports/base.py

@@ -159,6 +159,15 @@ def _prep_wrapped_messages(self, client_info):
             ),
         }
 
+    def close(self):
+        """Closes resources associated with the transport.
+
+       .. warning::
+            Only call this method if the transport is NOT shared
+            with other clients - this may cause errors in other clients!
+        """
+        raise NotImplementedError()
+
     @property
     def list_image_versions(
         self,

google/cloud/orchestration/airflow/service_v1/services/image_versions/transports/grpc.py

@@ -254,5 +254,8 @@ def list_image_versions(
             )
         return self._stubs["list_image_versions"]
 
+    def close(self):
+        self.grpc_channel.close()
+
 
 __all__ = ("ImageVersionsGrpcTransport",)

google/cloud/orchestration/airflow/service_v1/services/image_versions/transports/grpc_asyncio.py

@@ -257,5 +257,8 @@ def list_image_versions(
             )
         return self._stubs["list_image_versions"]
 
+    def close(self):
+        return self.grpc_channel.close()
+
 
 __all__ = ("ImageVersionsGrpcAsyncIOTransport",)

google/cloud/orchestration/airflow/service_v1/types/environments.py

@@ -380,6 +380,7 @@ class WebServerNetworkAccessControl(proto.Message):
 
     class AllowedIpRange(proto.Message):
         r"""Allowed IP range with user-provided description.
+
         Attributes:
             value (str):
                 IP address or range, defined using CIDR notation, of

google/cloud/orchestration/airflow/service_v1/types/image_versions.py

@@ -71,7 +71,7 @@ def raw_page(self):
 
 
 class ImageVersion(proto.Message):
-    r"""ImageVersion informatio.
+    r"""ImageVersion information
 
     Attributes:
         image_version_id (str):

google/cloud/orchestration/airflow/service_v1beta1/services/environments/async_client.py

@@ -904,6 +904,12 @@ async def check_upgrade(
         # Done; return the response.
         return response
 
+    async def __aenter__(self):
+        return self
+
+    async def __aexit__(self, exc_type, exc, tb):
+        await self.transport.close()
+
 
 try:
     DEFAULT_CLIENT_INFO = gapic_v1.client_info.ClientInfo(

google/cloud/orchestration/airflow/service_v1beta1/services/environments/client.py

@@ -350,10 +350,7 @@ def __init__(
                 client_cert_source_for_mtls=client_cert_source_func,
                 quota_project_id=client_options.quota_project_id,
                 client_info=client_info,
-                always_use_jwt_access=(
-                    Transport == type(self).get_transport_class("grpc")
-                    or Transport == type(self).get_transport_class("grpc_asyncio")
-                ),
+                always_use_jwt_access=True,
             )
 
     def create_environment(
@@ -1097,6 +1094,19 @@ def check_upgrade(
         # Done; return the response.
         return response
 
+    def __enter__(self):
+        return self
+
+    def __exit__(self, type, value, traceback):
+        """Releases underlying transport's resources.
+
+        .. warning::
+            ONLY use as a context manager if the transport is NOT shared
+            with other clients! Exiting the with block will CLOSE the transport
+            and may cause errors in other clients!
+        """
+        self.transport.close()
+
 
 try:
     DEFAULT_CLIENT_INFO = gapic_v1.client_info.ClientInfo(

google/cloud/orchestration/airflow/service_v1beta1/services/environments/transports/base.py

@@ -179,6 +179,15 @@ def _prep_wrapped_messages(self, client_info):
             ),
         }
 
+    def close(self):
+        """Closes resources associated with the transport.
+
+       .. warning::
+            Only call this method if the transport is NOT shared
+            with other clients - this may cause errors in other clients!
+        """
+        raise NotImplementedError()
+
     @property
     def operations_client(self) -> operations_v1.OperationsClient:
         """Return the client designed to process long-running operations."""

google/cloud/orchestration/airflow/service_v1beta1/services/environments/transports/grpc.py

@@ -429,5 +429,8 @@ def check_upgrade(
             )
         return self._stubs["check_upgrade"]
 
+    def close(self):
+        self.grpc_channel.close()
+
 
 __all__ = ("EnvironmentsGrpcTransport",)

google/cloud/orchestration/airflow/service_v1beta1/services/environments/transports/grpc_asyncio.py

@@ -447,5 +447,8 @@ def check_upgrade(
             )
         return self._stubs["check_upgrade"]
 
+    def close(self):
+        return self.grpc_channel.close()
+
 
 __all__ = ("EnvironmentsGrpcAsyncIOTransport",)

google/cloud/orchestration/airflow/service_v1beta1/services/image_versions/async_client.py

@@ -242,6 +242,12 @@ async def list_image_versions(
         # Done; return the response.
         return response
 
+    async def __aenter__(self):
+        return self
+
+    async def __aexit__(self, exc_type, exc, tb):
+        await self.transport.close()
+
 
 try:
     DEFAULT_CLIENT_INFO = gapic_v1.client_info.ClientInfo(

google/cloud/orchestration/airflow/service_v1beta1/services/image_versions/client.py

@@ -328,10 +328,7 @@ def __init__(
                 client_cert_source_for_mtls=client_cert_source_func,
                 quota_project_id=client_options.quota_project_id,
                 client_info=client_info,
-                always_use_jwt_access=(
-                    Transport == type(self).get_transport_class("grpc")
-                    or Transport == type(self).get_transport_class("grpc_asyncio")
-                ),
+                always_use_jwt_access=True,
             )
 
     def list_image_versions(
@@ -415,6 +412,19 @@ def list_image_versions(
         # Done; return the response.
         return response
 
+    def __enter__(self):
+        return self
+
+    def __exit__(self, type, value, traceback):
+        """Releases underlying transport's resources.
+
+        .. warning::
+            ONLY use as a context manager if the transport is NOT shared
+            with other clients! Exiting the with block will CLOSE the transport
+            and may cause errors in other clients!
+        """
+        self.transport.close()
+
 
 try:
     DEFAULT_CLIENT_INFO = gapic_v1.client_info.ClientInfo(

google/cloud/orchestration/airflow/service_v1beta1/services/image_versions/transports/base.py

@@ -159,6 +159,15 @@ def _prep_wrapped_messages(self, client_info):
             ),
         }
 
+    def close(self):
+        """Closes resources associated with the transport.
+
+       .. warning::
+            Only call this method if the transport is NOT shared
+            with other clients - this may cause errors in other clients!
+        """
+        raise NotImplementedError()
+
     @property
     def list_image_versions(
         self,

google/cloud/orchestration/airflow/service_v1beta1/services/image_versions/transports/grpc.py

@@ -254,5 +254,8 @@ def list_image_versions(
             )
         return self._stubs["list_image_versions"]
 
+    def close(self):
+        self.grpc_channel.close()
+
 
 __all__ = ("ImageVersionsGrpcTransport",)

google/cloud/orchestration/airflow/service_v1beta1/services/image_versions/transports/grpc_asyncio.py

@@ -257,5 +257,8 @@ def list_image_versions(
             )
         return self._stubs["list_image_versions"]
 
+    def close(self):
+        return self.grpc_channel.close()
+
 
 __all__ = ("ImageVersionsGrpcAsyncIOTransport",)

google/cloud/orchestration/airflow/service_v1beta1/types/environments.py

@@ -462,6 +462,7 @@ class WebServerNetworkAccessControl(proto.Message):
 
     class AllowedIpRange(proto.Message):
         r"""Allowed IP range with user-provided description.
+
         Attributes:
             value (str):
                 IP address or range, defined using CIDR notation, of
@@ -964,6 +965,7 @@ class WorkloadsConfig(proto.Message):
 
     class SchedulerResource(proto.Message):
         r"""Configuration for resources used by Airflow schedulers.
+
         Attributes:
             cpu (float):
                 Optional. CPU request and limit for a single
@@ -985,6 +987,7 @@ class SchedulerResource(proto.Message):
 
     class WebServerResource(proto.Message):
         r"""Configuration for resources used by Airflow web server.
+
         Attributes:
             cpu (float):
                 Optional. CPU request and limit for Airflow
@@ -1003,6 +1006,7 @@ class WebServerResource(proto.Message):
 
     class WorkerResource(proto.Message):
         r"""Configuration for resources used by Airflow workers.
+
         Attributes:
             cpu (float):
                 Optional. CPU request and limit for a single

google/cloud/orchestration/airflow/service_v1beta1/types/image_versions.py

@@ -71,7 +71,7 @@ def raw_page(self):
 
 
 class ImageVersion(proto.Message):
-    r"""Image Version informatio.
+    r"""Image Version information
 
     Attributes:
         image_version_id (str):