run nox -s format

Author: parthea

packages/google-cloud-bigtable/google/cloud/bigtable/__init__.py

@@ -14,9 +14,8 @@
 
 """Google Cloud Bigtable API package."""
 
-from google.cloud.bigtable.client import Client
-
 from google.cloud.bigtable import gapic_version as package_version
+from google.cloud.bigtable.client import Client
 
 __version__: str
 

packages/google-cloud-bigtable/google/cloud/bigtable/app_profile.py

@@ -14,13 +14,13 @@
 
 """User-friendly container for Google Cloud Bigtable AppProfile."""
 
-
 import re
 
+from google.api_core.exceptions import NotFound
+from google.protobuf import field_mask_pb2
+
 from google.cloud.bigtable.enums import RoutingPolicyType
 from google.cloud.bigtable_admin_v2.types import instance
-from google.protobuf import field_mask_pb2
-from google.api_core.exceptions import NotFound
 
 _APP_PROFILE_NAME_RE = re.compile(
     r"^projects/(?P<project>[^/]+)/"
@@ -165,7 +165,7 @@ def from_pb(cls, app_profile_pb, instance):
         match_app_profile_name = _APP_PROFILE_NAME_RE.match(app_profile_pb.name)
         if match_app_profile_name is None:
             raise ValueError(
-                "AppProfile protobuf name was not in the " "expected format.",
+                "AppProfile protobuf name was not in the expected format.",
                 app_profile_pb.name,
             )
         if match_app_profile_name.group("instance") != instance.instance_id:
@@ -175,8 +175,7 @@ def from_pb(cls, app_profile_pb, instance):
             )
         if match_app_profile_name.group("project") != instance._client.project:
             raise ValueError(
-                "Project ID on app_profile does not match the "
-                "project ID on the client"
+                "Project ID on app_profile does not match the project ID on the client"
             )
         app_profile_id = match_app_profile_name.group("app_profile_id")
 

packages/google-cloud-bigtable/google/cloud/bigtable/backup.py

@@ -17,13 +17,14 @@
 import re
 
 from google.cloud._helpers import _datetime_to_pb_timestamp  # type: ignore
-from google.cloud.bigtable_admin_v2 import BaseBigtableTableAdminClient
-from google.cloud.bigtable_admin_v2.types import table
-from google.cloud.bigtable.encryption_info import EncryptionInfo
-from google.cloud.bigtable.policy import Policy
 from google.cloud.exceptions import NotFound  # type: ignore
 from google.protobuf import field_mask_pb2
 
+from google.cloud.bigtable.encryption_info import EncryptionInfo
+from google.cloud.bigtable.policy import Policy
+from google.cloud.bigtable_admin_v2 import BaseBigtableTableAdminClient
+from google.cloud.bigtable_admin_v2.types import table
+
 _BACKUP_NAME_RE = re.compile(
     r"^projects/(?P<project>[^/]+)/"
     r"instances/(?P<instance_id>[a-z][-a-z0-9]*)/"

packages/google-cloud-bigtable/google/cloud/bigtable/batcher.py

@@ -13,15 +13,14 @@
 # limitations under the License.
 
 """User friendly container for Google Cloud Bigtable MutationBatcher."""
-import threading
-import queue
-import concurrent.futures
-import atexit
-
 
-from google.api_core.exceptions import from_grpc_status
+import atexit
+import concurrent.futures
+import queue
+import threading
 from dataclasses import dataclass
 
+from google.api_core.exceptions import from_grpc_status
 
 FLUSH_COUNT = 100  # after this many elements, send out the batch
 

packages/google-cloud-bigtable/google/cloud/bigtable/client.py

@@ -27,33 +27,27 @@
 * a :class:`~google.cloud.bigtable.table.Table` owns a
   :class:`~google.cloud.bigtable.row.Row` (and all the cells in the row)
 """
+
 import os
 import warnings
-import grpc  # type: ignore
 
+import grpc  # type: ignore
 from google.api_core.gapic_v1 import client_info as client_info_lib
 from google.auth.credentials import AnonymousCredentials  # type: ignore
+from google.cloud.client import ClientWithProject  # type: ignore
+from google.cloud.environment_vars import BIGTABLE_EMULATOR  # type: ignore
 
-from google.cloud import bigtable_v2
-from google.cloud import bigtable_admin_v2
-from google.cloud.bigtable_v2.services.bigtable.transports import BigtableGrpcTransport
+from google.cloud import bigtable, bigtable_admin_v2, bigtable_v2
+from google.cloud.bigtable.cluster import _CLUSTER_NAME_RE, Cluster
+from google.cloud.bigtable.instance import Instance
 from google.cloud.bigtable_admin_v2.services.bigtable_instance_admin.transports import (
     BigtableInstanceAdminGrpcTransport,
 )
 from google.cloud.bigtable_admin_v2.services.bigtable_table_admin.transports import (
     BigtableTableAdminGrpcTransport,
 )
-
-from google.cloud import bigtable
-from google.cloud.bigtable.instance import Instance
-from google.cloud.bigtable.cluster import Cluster
-
-from google.cloud.client import ClientWithProject  # type: ignore
-
 from google.cloud.bigtable_admin_v2.types import instance
-from google.cloud.bigtable.cluster import _CLUSTER_NAME_RE
-from google.cloud.environment_vars import BIGTABLE_EMULATOR  # type: ignore
-
+from google.cloud.bigtable_v2.services.bigtable.transports import BigtableGrpcTransport
 
 INSTANCE_TYPE_PRODUCTION = instance.Instance.Type.PRODUCTION
 INSTANCE_TYPE_DEVELOPMENT = instance.Instance.Type.DEVELOPMENT
@@ -164,7 +158,7 @@ def __init__(
             )
         if read_only and admin:
             raise ValueError(
-                "A read-only client cannot also perform" "administrative actions."
+                "A read-only client cannot also performadministrative actions."
             )
 
         # NOTE: We set the scopes **before** calling the parent constructor.

packages/google-cloud-bigtable/google/cloud/bigtable/cluster.py

@@ -14,12 +14,12 @@
 
 """User friendly container for Google Cloud Bigtable Cluster."""
 
-
 import re
-from google.cloud.bigtable_admin_v2.types import instance
+
 from google.api_core.exceptions import NotFound
 from google.protobuf import field_mask_pb2
 
+from google.cloud.bigtable_admin_v2.types import instance
 
 _CLUSTER_NAME_RE = re.compile(
     r"^projects/(?P<project>[^/]+)/"
@@ -166,16 +166,16 @@ def from_pb(cls, cluster_pb, instance):
         match_cluster_name = _CLUSTER_NAME_RE.match(cluster_pb.name)
         if match_cluster_name is None:
             raise ValueError(
-                "Cluster protobuf name was not in the " "expected format.",
+                "Cluster protobuf name was not in the expected format.",
                 cluster_pb.name,
             )
         if match_cluster_name.group("instance") != instance.instance_id:
             raise ValueError(
-                "Instance ID on cluster does not match the " "instance ID on the client"
+                "Instance ID on cluster does not match the instance ID on the client"
             )
         if match_cluster_name.group("project") != instance._client.project:
             raise ValueError(
-                "Project ID on cluster does not match the " "project ID on the client"
+                "Project ID on cluster does not match the project ID on the client"
             )
         cluster_id = match_cluster_name.group("cluster_id")
 
@@ -191,15 +191,9 @@ def _update_from_pb(self, cluster_pb):
         self.location_id = cluster_pb.location.split("/")[-1]
         self.serve_nodes = cluster_pb.serve_nodes
 
-        self.min_serve_nodes = (
-            cluster_pb.cluster_config.cluster_autoscaling_config.autoscaling_limits.min_serve_nodes
-        )
-        self.max_serve_nodes = (
-            cluster_pb.cluster_config.cluster_autoscaling_config.autoscaling_limits.max_serve_nodes
-        )
-        self.cpu_utilization_percent = (
-            cluster_pb.cluster_config.cluster_autoscaling_config.autoscaling_targets.cpu_utilization_percent
-        )
+        self.min_serve_nodes = cluster_pb.cluster_config.cluster_autoscaling_config.autoscaling_limits.min_serve_nodes
+        self.max_serve_nodes = cluster_pb.cluster_config.cluster_autoscaling_config.autoscaling_limits.max_serve_nodes
+        self.cpu_utilization_percent = cluster_pb.cluster_config.cluster_autoscaling_config.autoscaling_targets.cpu_utilization_percent
 
         self.default_storage_type = cluster_pb.default_storage_type
         if cluster_pb.encryption_config:
@@ -528,16 +522,10 @@ def _to_pb(self):
             )
 
         if self.min_serve_nodes:
-            cluster_pb.cluster_config.cluster_autoscaling_config.autoscaling_limits.min_serve_nodes = (
-                self.min_serve_nodes
-            )
+            cluster_pb.cluster_config.cluster_autoscaling_config.autoscaling_limits.min_serve_nodes = self.min_serve_nodes
         if self.max_serve_nodes:
-            cluster_pb.cluster_config.cluster_autoscaling_config.autoscaling_limits.max_serve_nodes = (
-                self.max_serve_nodes
-            )
+            cluster_pb.cluster_config.cluster_autoscaling_config.autoscaling_limits.max_serve_nodes = self.max_serve_nodes
         if self.cpu_utilization_percent:
-            cluster_pb.cluster_config.cluster_autoscaling_config.autoscaling_targets.cpu_utilization_percent = (
-                self.cpu_utilization_percent
-            )
+            cluster_pb.cluster_config.cluster_autoscaling_config.autoscaling_targets.cpu_utilization_percent = self.cpu_utilization_percent
 
         return cluster_pb

packages/google-cloud-bigtable/google/cloud/bigtable/column_family.py

@@ -14,13 +14,13 @@
 
 """User friendly container for Google Cloud Bigtable Column Family."""
 
+from google.api_core.gapic_v1.method import DEFAULT
 
 from google.cloud import _helpers
-from google.cloud.bigtable_admin_v2.types import table as table_v2_pb2
 from google.cloud.bigtable_admin_v2.types import (
     bigtable_table_admin as table_admin_v2_pb2,
 )
-from google.api_core.gapic_v1.method import DEFAULT
+from google.cloud.bigtable_admin_v2.types import table as table_v2_pb2
 
 
 class GarbageCollectionRule(object):

packages/google-cloud-bigtable/google/cloud/bigtable/data/__init__.py

@@ -14,56 +14,56 @@
 # limitations under the License.
 #
 from google.cloud.bigtable import gapic_version as package_version
-
-from google.cloud.bigtable.data._async.client import BigtableDataClientAsync
-from google.cloud.bigtable.data._async.client import TableAsync
-from google.cloud.bigtable.data._async.client import AuthorizedViewAsync
+from google.cloud.bigtable.data._async._mutate_rows import _MutateRowsOperationAsync
+from google.cloud.bigtable.data._async._read_rows import _ReadRowsOperationAsync
+from google.cloud.bigtable.data._async.client import (
+    AuthorizedViewAsync,
+    BigtableDataClientAsync,
+    TableAsync,
+)
 from google.cloud.bigtable.data._async.mutations_batcher import MutationsBatcherAsync
-from google.cloud.bigtable.data._sync_autogen.client import BigtableDataClient
-from google.cloud.bigtable.data._sync_autogen.client import Table
-from google.cloud.bigtable.data._sync_autogen.client import AuthorizedView
+from google.cloud.bigtable.data._cross_sync import CrossSync
+from google.cloud.bigtable.data._helpers import (
+    TABLE_DEFAULT,
+    RowKeySamples,
+    ShardedQuery,
+)
+from google.cloud.bigtable.data._sync_autogen._mutate_rows import _MutateRowsOperation
+from google.cloud.bigtable.data._sync_autogen._read_rows import _ReadRowsOperation
+from google.cloud.bigtable.data._sync_autogen.client import (
+    AuthorizedView,
+    BigtableDataClient,
+    Table,
+)
 from google.cloud.bigtable.data._sync_autogen.mutations_batcher import MutationsBatcher
-
-from google.cloud.bigtable.data.read_rows_query import ReadRowsQuery
-from google.cloud.bigtable.data.read_rows_query import RowRange
-from google.cloud.bigtable.data.row import Row
-from google.cloud.bigtable.data.row import Cell
-
-from google.cloud.bigtable.data.mutations import Mutation
-from google.cloud.bigtable.data.mutations import RowMutationEntry
-from google.cloud.bigtable.data.mutations import AddToCell
-from google.cloud.bigtable.data.mutations import SetCell
-from google.cloud.bigtable.data.mutations import DeleteRangeFromColumn
-from google.cloud.bigtable.data.mutations import DeleteAllFromFamily
-from google.cloud.bigtable.data.mutations import DeleteAllFromRow
-
-from google.cloud.bigtable.data.exceptions import InvalidChunk
-from google.cloud.bigtable.data.exceptions import FailedMutationEntryError
-from google.cloud.bigtable.data.exceptions import FailedQueryShardError
-
-from google.cloud.bigtable.data.exceptions import RetryExceptionGroup
-from google.cloud.bigtable.data.exceptions import MutationsExceptionGroup
-from google.cloud.bigtable.data.exceptions import ShardedReadRowsExceptionGroup
-from google.cloud.bigtable.data.exceptions import ParameterTypeInferenceFailed
-
-from google.cloud.bigtable.data._helpers import TABLE_DEFAULT
-from google.cloud.bigtable.data._helpers import RowKeySamples
-from google.cloud.bigtable.data._helpers import ShardedQuery
+from google.cloud.bigtable.data.exceptions import (
+    FailedMutationEntryError,
+    FailedQueryShardError,
+    InvalidChunk,
+    MutationsExceptionGroup,
+    ParameterTypeInferenceFailed,
+    RetryExceptionGroup,
+    ShardedReadRowsExceptionGroup,
+)
+from google.cloud.bigtable.data.mutations import (
+    AddToCell,
+    DeleteAllFromFamily,
+    DeleteAllFromRow,
+    DeleteRangeFromColumn,
+    Mutation,
+    RowMutationEntry,
+    SetCell,
+)
+from google.cloud.bigtable.data.read_rows_query import ReadRowsQuery, RowRange
+from google.cloud.bigtable.data.row import Cell, Row
 
 # setup custom CrossSync mappings for library
 from google.cloud.bigtable_v2.services.bigtable.async_client import (
     BigtableAsyncClient,
 )
-from google.cloud.bigtable.data._async._read_rows import _ReadRowsOperationAsync
-from google.cloud.bigtable.data._async._mutate_rows import _MutateRowsOperationAsync
-
 from google.cloud.bigtable_v2.services.bigtable.client import (
     BigtableClient,
 )
-from google.cloud.bigtable.data._sync_autogen._read_rows import _ReadRowsOperation
-from google.cloud.bigtable.data._sync_autogen._mutate_rows import _MutateRowsOperation
-
-from google.cloud.bigtable.data._cross_sync import CrossSync
 
 CrossSync.add_mapping("GapicClient", BigtableAsyncClient)
 CrossSync._Sync_Impl.add_mapping("GapicClient", BigtableClient)

packages/google-cloud-bigtable/google/cloud/bigtable/data/_async/__init__.py

@@ -12,12 +12,9 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from google.cloud.bigtable.data._async.client import BigtableDataClientAsync
-from google.cloud.bigtable.data._async.client import TableAsync
-
+from google.cloud.bigtable.data._async.client import BigtableDataClientAsync, TableAsync
 from google.cloud.bigtable.data._async.mutations_batcher import MutationsBatcherAsync
 
-
 __all__ = [
     "BigtableDataClientAsync",
     "TableAsync",

packages/google-cloud-bigtable/google/cloud/bigtable/data/_async/_mutate_rows.py

@@ -14,38 +14,42 @@
 #
 from __future__ import annotations
 
-from typing import Sequence, TYPE_CHECKING
+from typing import TYPE_CHECKING, Sequence
 
 from google.api_core import exceptions as core_exceptions
 from google.api_core import retry as retries
-import google.cloud.bigtable_v2.types.bigtable as types_pb
+
 import google.cloud.bigtable.data.exceptions as bt_exceptions
-from google.cloud.bigtable.data._helpers import _attempt_timeout_generator
-from google.cloud.bigtable.data._helpers import _retry_exception_factory
+import google.cloud.bigtable_v2.types.bigtable as types_pb
+from google.cloud.bigtable.data._cross_sync import CrossSync
+from google.cloud.bigtable.data._helpers import (
+    _attempt_timeout_generator,
+    _retry_exception_factory,
+)
 
 # mutate_rows requests are limited to this number of mutations
-from google.cloud.bigtable.data.mutations import _MUTATE_ROWS_REQUEST_MUTATION_LIMIT
-from google.cloud.bigtable.data.mutations import _EntryWithProto
-
-from google.cloud.bigtable.data._cross_sync import CrossSync
+from google.cloud.bigtable.data.mutations import (
+    _MUTATE_ROWS_REQUEST_MUTATION_LIMIT,
+    _EntryWithProto,
+)
 
 if TYPE_CHECKING:
     from google.cloud.bigtable.data.mutations import RowMutationEntry
 
     if CrossSync.is_async:
-        from google.cloud.bigtable_v2.services.bigtable.async_client import (
-            BigtableAsyncClient as GapicClientType,
-        )
         from google.cloud.bigtable.data._async.client import (  # type: ignore
             _DataApiTargetAsync as TargetType,
         )
-    else:
-        from google.cloud.bigtable_v2.services.bigtable.client import (  # type: ignore
-            BigtableClient as GapicClientType,
+        from google.cloud.bigtable_v2.services.bigtable.async_client import (
+            BigtableAsyncClient as GapicClientType,
         )
+    else:
         from google.cloud.bigtable.data._sync_autogen.client import (  # type: ignore
             _DataApiTarget as TargetType,
         )
+        from google.cloud.bigtable_v2.services.bigtable.client import (  # type: ignore
+            BigtableClient as GapicClientType,
+        )
 
 __CROSS_SYNC_OUTPUT__ = "google.cloud.bigtable.data._sync_autogen._mutate_rows"