feat: add context manager support in client

google/cloud/datacatalog_v1/services/data_catalog/async_client.py

@@ -2911,6 +2911,12 @@ async def test_iam_permissions(
         # 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/datacatalog_v1/services/data_catalog/client.py

@@ -454,10 +454,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 search_catalog(
@@ -3120,6 +3117,19 @@ def test_iam_permissions(
         # 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/datacatalog_v1/services/data_catalog/transports/base.py

@@ -344,6 +344,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 search_catalog(
         self,

google/cloud/datacatalog_v1/services/data_catalog/transports/grpc.py

@@ -1157,5 +1157,8 @@ def test_iam_permissions(
             )
         return self._stubs["test_iam_permissions"]
 
+    def close(self):
+        self.grpc_channel.close()
+
 
 __all__ = ("DataCatalogGrpcTransport",)

google/cloud/datacatalog_v1/services/data_catalog/transports/grpc_asyncio.py

@@ -1188,5 +1188,8 @@ def test_iam_permissions(
             )
         return self._stubs["test_iam_permissions"]
 
+    def close(self):
+        return self.grpc_channel.close()
+
 
 __all__ = ("DataCatalogGrpcAsyncIOTransport",)

google/cloud/datacatalog_v1/services/policy_tag_manager/async_client.py

@@ -1238,6 +1238,12 @@ async def test_iam_permissions(
         # 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/datacatalog_v1/services/policy_tag_manager/client.py

@@ -374,10 +374,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_taxonomy(
@@ -1443,6 +1440,19 @@ def test_iam_permissions(
         # 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/datacatalog_v1/services/policy_tag_manager/transports/base.py

@@ -200,6 +200,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 create_taxonomy(
         self,

google/cloud/datacatalog_v1/services/policy_tag_manager/transports/grpc.py

@@ -597,5 +597,8 @@ def test_iam_permissions(
             )
         return self._stubs["test_iam_permissions"]
 
+    def close(self):
+        self.grpc_channel.close()
+
 
 __all__ = ("PolicyTagManagerGrpcTransport",)

google/cloud/datacatalog_v1/services/policy_tag_manager/transports/grpc_asyncio.py

@@ -610,5 +610,8 @@ def test_iam_permissions(
             )
         return self._stubs["test_iam_permissions"]
 
+    def close(self):
+        return self.grpc_channel.close()
+
 
 __all__ = ("PolicyTagManagerGrpcAsyncIOTransport",)

google/cloud/datacatalog_v1/services/policy_tag_manager_serialization/async_client.py

@@ -357,6 +357,12 @@ async def export_taxonomies(
         # 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/datacatalog_v1/services/policy_tag_manager_serialization/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 replace_taxonomy(
@@ -558,6 +555,19 @@ def export_taxonomies(
         # 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/datacatalog_v1/services/policy_tag_manager_serialization/transports/base.py

@@ -166,6 +166,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 replace_taxonomy(
         self,

google/cloud/datacatalog_v1/services/policy_tag_manager_serialization/transports/grpc.py

@@ -343,5 +343,8 @@ def export_taxonomies(
             )
         return self._stubs["export_taxonomies"]
 
+    def close(self):
+        self.grpc_channel.close()
+
 
 __all__ = ("PolicyTagManagerSerializationGrpcTransport",)

google/cloud/datacatalog_v1/services/policy_tag_manager_serialization/transports/grpc_asyncio.py

@@ -346,5 +346,8 @@ def export_taxonomies(
             )
         return self._stubs["export_taxonomies"]
 
+    def close(self):
+        return self.grpc_channel.close()
+
 
 __all__ = ("PolicyTagManagerSerializationGrpcAsyncIOTransport",)

google/cloud/datacatalog_v1/types/bigquery.py

@@ -28,6 +28,7 @@
 
 class BigQueryConnectionSpec(proto.Message):
     r"""Specification for the BigQuery connection.
+
     Attributes:
         connection_type (google.cloud.datacatalog_v1.types.BigQueryConnectionSpec.ConnectionType):
             The type of the BigQuery connection.
@@ -81,6 +82,7 @@ class DatabaseType(proto.Enum):
 
 class BigQueryRoutineSpec(proto.Message):
     r"""Fields specific for BigQuery routines.
+
     Attributes:
         imported_libraries (Sequence[str]):
             Paths of the imported libraries.

google/cloud/datacatalog_v1/types/data_source.py

@@ -23,6 +23,7 @@
 
 class DataSource(proto.Message):
     r"""Physical location of an entry.
+
     Attributes:
         service (google.cloud.datacatalog_v1.types.DataSource.Service):
             Service that physically stores the data.

google/cloud/datacatalog_v1/types/datacatalog.py

@@ -786,6 +786,7 @@ class RoutineType(proto.Enum):
 
     class Argument(proto.Message):
         r"""Input or output argument of a function or stored procedure.
+
         Attributes:
             name (str):
                 The name of the argument. A return argument

google/cloud/datacatalog_v1/types/gcs_fileset_spec.py

@@ -25,6 +25,7 @@
 
 class GcsFilesetSpec(proto.Message):
     r"""Describes a Cloud Storage fileset entry.
+
     Attributes:
         file_patterns (Sequence[str]):
             Required. Patterns to identify a set of files in Google
@@ -75,6 +76,7 @@ class GcsFilesetSpec(proto.Message):
 
 class GcsFileSpec(proto.Message):
     r"""Specification of a single file in Cloud Storage.
+
     Attributes:
         file_path (str):
             Required. Full file path. Example:

google/cloud/datacatalog_v1/types/policytagmanagerserialization.py

@@ -137,6 +137,7 @@ class ImportTaxonomiesRequest(proto.Message):
 
 class InlineSource(proto.Message):
     r"""Inline source containing taxonomies to import.
+
     Attributes:
         taxonomies (Sequence[google.cloud.datacatalog_v1.types.SerializedTaxonomy]):
             Required. Taxonomies to import.

google/cloud/datacatalog_v1/types/table_spec.py

@@ -38,6 +38,7 @@ class TableSourceType(proto.Enum):
 
 class BigQueryTableSpec(proto.Message):
     r"""Describes a BigQuery table.
+
     Attributes:
         table_source_type (google.cloud.datacatalog_v1.types.TableSourceType):
             Output only. The table source type.
@@ -60,6 +61,7 @@ class BigQueryTableSpec(proto.Message):
 
 class ViewSpec(proto.Message):
     r"""Table view specification.
+
     Attributes:
         view_query (str):
             Output only. The query that defines the table
@@ -71,6 +73,7 @@ class ViewSpec(proto.Message):
 
 class TableSpec(proto.Message):
     r"""Normal BigQuery table specification.
+
     Attributes:
         grouped_entry (str):
             Output only. If the table is date-sharded, that is, it

google/cloud/datacatalog_v1/types/tags.py

@@ -113,6 +113,7 @@ class TagField(proto.Message):
 
     class EnumValue(proto.Message):
         r"""An enum value.
+
         Attributes:
             display_name (str):
                 The display name of the enum value.
@@ -216,6 +217,7 @@ class TagTemplate(proto.Message):
 
 class TagTemplateField(proto.Message):
     r"""The template for an individual field within a tag template.
+
     Attributes:
         name (str):
             Output only. The resource name of the tag template field in
@@ -266,6 +268,7 @@ class TagTemplateField(proto.Message):
 
 class FieldType(proto.Message):
     r"""
+
     Attributes:
         primitive_type (google.cloud.datacatalog_v1.types.FieldType.PrimitiveType):
             Primitive types, such as string, boolean,
@@ -285,6 +288,7 @@ class PrimitiveType(proto.Enum):
 
     class EnumType(proto.Message):
         r"""
+
         Attributes:
             allowed_values (Sequence[google.cloud.datacatalog_v1.types.FieldType.EnumType.EnumValue]):
                 The set of allowed values for this enum.
@@ -302,6 +306,7 @@ class EnumType(proto.Message):
 
         class EnumValue(proto.Message):
             r"""
+
             Attributes:
                 display_name (str):
                     Required. The display name of the enum value. Must not be an

google/cloud/datacatalog_v1beta1/services/data_catalog/async_client.py

@@ -2796,6 +2796,12 @@ async def test_iam_permissions(
         # 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(