feat: add context manager support in client

google/cloud/gaming_v1/services/game_server_clusters_service/async_client.py

@@ -822,6 +822,12 @@ async def preview_update_game_server_cluster(
         # 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/gaming_v1/services/game_server_clusters_service/client.py

@@ -356,10 +356,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_game_server_clusters(
@@ -1000,6 +997,19 @@ def preview_update_game_server_cluster(
         # 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/gaming_v1/services/game_server_clusters_service/transports/base.py

@@ -243,6 +243,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/gaming_v1/services/game_server_clusters_service/transports/grpc.py

@@ -484,5 +484,8 @@ def preview_update_game_server_cluster(
             )
         return self._stubs["preview_update_game_server_cluster"]
 
+    def close(self):
+        self.grpc_channel.close()
+
 
 __all__ = ("GameServerClustersServiceGrpcTransport",)

google/cloud/gaming_v1/services/game_server_clusters_service/transports/grpc_asyncio.py

@@ -492,5 +492,8 @@ def preview_update_game_server_cluster(
             )
         return self._stubs["preview_update_game_server_cluster"]
 
+    def close(self):
+        return self.grpc_channel.close()
+
 
 __all__ = ("GameServerClustersServiceGrpcAsyncIOTransport",)

google/cloud/gaming_v1/services/game_server_configs_service/async_client.py

@@ -539,6 +539,12 @@ async def delete_game_server_config(
         # 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/gaming_v1/services/game_server_configs_service/client.py

@@ -355,10 +355,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_game_server_configs(
@@ -709,6 +706,19 @@ def delete_game_server_config(
         # 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/gaming_v1/services/game_server_configs_service/transports/base.py

@@ -196,6 +196,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/gaming_v1/services/game_server_configs_service/transports/grpc.py

@@ -364,5 +364,8 @@ def delete_game_server_config(
             )
         return self._stubs["delete_game_server_config"]
 
+    def close(self):
+        self.grpc_channel.close()
+
 
 __all__ = ("GameServerConfigsServiceGrpcTransport",)

google/cloud/gaming_v1/services/game_server_configs_service/transports/grpc_asyncio.py

@@ -371,5 +371,8 @@ def delete_game_server_config(
             )
         return self._stubs["delete_game_server_config"]
 
+    def close(self):
+        return self.grpc_channel.close()
+
 
 __all__ = ("GameServerConfigsServiceGrpcAsyncIOTransport",)

google/cloud/gaming_v1/services/game_server_deployments_service/async_client.py

@@ -951,6 +951,12 @@ async def fetch_deployment_state(
         # 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/gaming_v1/services/game_server_deployments_service/client.py

@@ -378,10 +378,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_game_server_deployments(
@@ -1158,6 +1155,19 @@ def fetch_deployment_state(
         # 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/gaming_v1/services/game_server_deployments_service/transports/base.py

@@ -248,6 +248,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/gaming_v1/services/game_server_deployments_service/transports/grpc.py

@@ -531,5 +531,8 @@ def fetch_deployment_state(
             )
         return self._stubs["fetch_deployment_state"]
 
+    def close(self):
+        self.grpc_channel.close()
+
 
 __all__ = ("GameServerDeploymentsServiceGrpcTransport",)

google/cloud/gaming_v1/services/game_server_deployments_service/transports/grpc_asyncio.py

@@ -538,5 +538,8 @@ def fetch_deployment_state(
             )
         return self._stubs["fetch_deployment_state"]
 
+    def close(self):
+        return self.grpc_channel.close()
+
 
 __all__ = ("GameServerDeploymentsServiceGrpcAsyncIOTransport",)

google/cloud/gaming_v1/services/realms_service/async_client.py

@@ -687,6 +687,12 @@ async def preview_realm_update(
         # 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/gaming_v1/services/realms_service/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 list_realms(
@@ -848,6 +845,19 @@ def preview_realm_update(
         # 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/gaming_v1/services/realms_service/transports/base.py

@@ -209,6 +209,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/gaming_v1/services/realms_service/transports/grpc.py

@@ -399,5 +399,8 @@ def preview_realm_update(
             )
         return self._stubs["preview_realm_update"]
 
+    def close(self):
+        self.grpc_channel.close()
+
 
 __all__ = ("RealmsServiceGrpcTransport",)

google/cloud/gaming_v1/services/realms_service/transports/grpc_asyncio.py

@@ -404,5 +404,8 @@ def preview_realm_update(
             )
         return self._stubs["preview_realm_update"]
 
+    def close(self):
+        return self.grpc_channel.close()
+
 
 __all__ = ("RealmsServiceGrpcAsyncIOTransport",)