run nox -s generate_sync

Author: parthea

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

@@ -16,12 +16,9 @@
 # This file is automatically generated by CrossSync. Do not edit manually.
 
 from __future__ import annotations
-
 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.data.exceptions as bt_exceptions
 import google.cloud.bigtable_v2.types.bigtable as types_pb
 from google.cloud.bigtable.data._cross_sync import CrossSync
@@ -35,10 +32,10 @@
 )
 
 if TYPE_CHECKING:
+    from google.cloud.bigtable.data.mutations import RowMutationEntry
     from google.cloud.bigtable.data._sync_autogen.client import (
         _DataApiTarget as TargetType,
     )
-    from google.cloud.bigtable.data.mutations import RowMutationEntry
     from google.cloud.bigtable_v2.services.bigtable.client import (
         BigtableClient as GapicClientType,
     )
@@ -137,8 +134,7 @@ def _run_attempt(self):
             GoogleAPICallError: if the gapic rpc fails"""
         request_entries = [self.mutations[idx].proto for idx in self.remaining_indices]
         active_request_indices = {
-            req_idx: orig_idx
-            for (req_idx, orig_idx) in enumerate(self.remaining_indices)
+            req_idx: orig_idx for req_idx, orig_idx in enumerate(self.remaining_indices)
         }
         self.remaining_indices = []
         if not request_entries:

packages/google-cloud-bigtable/google/cloud/bigtable/data/_sync_autogen/_read_rows.py

@@ -17,12 +17,9 @@
 # This file is automatically generated by CrossSync. Do not edit manually.
 
 from __future__ import annotations
-
 from typing import TYPE_CHECKING, Sequence
-
 from google.api_core import retry as retries
 from google.api_core.retry import exponential_sleep_generator
-
 from google.cloud.bigtable.data._cross_sync import CrossSync
 from google.cloud.bigtable.data._helpers import (
     _attempt_timeout_generator,

packages/google-cloud-bigtable/google/cloud/bigtable/data/_sync_autogen/_swappable_channel.py

@@ -16,10 +16,9 @@
 # This file is automatically generated by CrossSync. Do not edit manually.
 
 from __future__ import annotations
-
 from typing import Callable
-
-from grpc import Channel, ChannelConnectivity
+from grpc import ChannelConnectivity
+from grpc import Channel
 
 
 class _WrappedChannel(Channel):

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

@@ -17,16 +17,14 @@
 # This file is automatically generated by CrossSync. Do not edit manually.
 
 from __future__ import annotations
-
 import abc
 import concurrent.futures
 import os
 import random
 import time
 import warnings
 from functools import partial
-from typing import TYPE_CHECKING, Any, Callable, Iterable, Optional, Sequence, Set, cast
-
+from typing import TYPE_CHECKING, Any, Callable, Optional, Sequence, Set, cast
 import google.auth._default
 import google.auth.credentials
 from google.api_core import client_options as client_options_lib
@@ -41,8 +39,7 @@
 from google.cloud.environment_vars import BIGTABLE_EMULATOR
 from google.protobuf.internal.enum_type_wrapper import EnumTypeWrapper
 from google.protobuf.message import Message
-from grpc import Channel, insecure_channel, intercept_channel
-
+from grpc import Channel
 from google.cloud.bigtable.client import _DEFAULT_BIGTABLE_EMULATOR_CLIENT
 from google.cloud.bigtable.data._cross_sync import CrossSync
 from google.cloud.bigtable.data._helpers import (
@@ -58,13 +55,6 @@
     _WarmedInstanceKey,
 )
 from google.cloud.bigtable.data._metrics import BigtableClientSideMetricsController
-from google.cloud.bigtable.data._sync_autogen._swappable_channel import (
-    SwappableChannel as SwappableChannelType,
-)
-from google.cloud.bigtable.data._sync_autogen.metrics_interceptor import (
-    BigtableMetricsInterceptor as MetricsInterceptorType,
-)
-from google.cloud.bigtable.data._sync_autogen.mutations_batcher import _MB_SIZE
 from google.cloud.bigtable.data.exceptions import (
     FailedQueryShardError,
     ShardedReadRowsExceptionGroup,
@@ -88,10 +78,6 @@
     RowFilterChain,
     StripValueTransformerFilter,
 )
-from google.cloud.bigtable_v2.services.bigtable import BigtableClient as GapicClient
-from google.cloud.bigtable_v2.services.bigtable.transports import (
-    BigtableGrpcTransport as TransportType,
-)
 from google.cloud.bigtable_v2.services.bigtable.transports.base import (
     DEFAULT_CLIENT_INFO,
 )
@@ -102,6 +88,19 @@
     ReadModifyWriteRowRequest,
     SampleRowKeysRequest,
 )
+from typing import Iterable
+from grpc import insecure_channel, intercept_channel
+from google.cloud.bigtable.data._sync_autogen._swappable_channel import (
+    SwappableChannel as SwappableChannelType,
+)
+from google.cloud.bigtable.data._sync_autogen.metrics_interceptor import (
+    BigtableMetricsInterceptor as MetricsInterceptorType,
+)
+from google.cloud.bigtable.data._sync_autogen.mutations_batcher import _MB_SIZE
+from google.cloud.bigtable_v2.services.bigtable import BigtableClient as GapicClient
+from google.cloud.bigtable_v2.services.bigtable.transports import (
+    BigtableGrpcTransport as TransportType,
+)
 
 if TYPE_CHECKING:
     from google.cloud.bigtable.data._helpers import RowKeySamples, ShardedQuery
@@ -320,7 +319,7 @@ def _ping_and_warm_instances(
                 ],
                 wait_for_ready=True,
             )
-            for (instance_name, app_profile_id) in instance_list
+            for instance_name, app_profile_id in instance_list
         ]
         result_list = CrossSync._Sync_Impl.gather_partials(
             partial_list, return_exceptions=True, sync_executor=self._executor
@@ -669,7 +668,7 @@ def execute_query(
         prepare_predicate = retries.if_exception_type(
             *[_get_error_type(e) for e in prepare_retryable_errors]
         )
-        (prepare_operation_timeout, prepare_attempt_timeout) = _align_timeouts(
+        prepare_operation_timeout, prepare_attempt_timeout = _align_timeouts(
             prepare_operation_timeout, prepare_attempt_timeout
         )
         prepare_sleep_generator = retries.exponential_sleep_generator(0.01, 2, 60)
@@ -695,7 +694,7 @@ def execute_query(
             "prepared_query": prepare_result.prepared_query,
             "params": pb_params,
         }
-        (operation_timeout, attempt_timeout) = _align_timeouts(
+        operation_timeout, attempt_timeout = _align_timeouts(
             operation_timeout, attempt_timeout
         )
         return CrossSync._Sync_Impl.ExecuteQueryIterator(
@@ -896,7 +895,7 @@ def read_rows_stream(
                 from any retries that failed
             google.api_core.exceptions.GoogleAPIError: raised if the request encounters an unrecoverable error
         """
-        (operation_timeout, attempt_timeout) = _get_timeouts(
+        operation_timeout, attempt_timeout = _get_timeouts(
             operation_timeout, attempt_timeout, self
         )
         retryable_excs = _get_retryable_errors(retryable_errors, self)
@@ -1042,7 +1041,7 @@ def read_rows_sharded(
             ValueError: if the query_list is empty"""
         if not sharded_query:
             raise ValueError("empty sharded_query")
-        (operation_timeout, attempt_timeout) = _get_timeouts(
+        operation_timeout, attempt_timeout = _get_timeouts(
             operation_timeout, attempt_timeout, self
         )
         rpc_timeout_generator = _attempt_timeout_generator(
@@ -1085,7 +1084,7 @@ def read_rows_with_semaphore(query):
             raise ShardedReadRowsExceptionGroup(
                 [
                     FailedQueryShardError(idx, sharded_query[idx], e)
-                    for (idx, e) in error_dict.items()
+                    for idx, e in error_dict.items()
                 ],
                 results_list,
                 len(sharded_query),
@@ -1175,7 +1174,7 @@ def sample_row_keys(
                 from any retries that failed
             google.api_core.exceptions.GoogleAPIError: raised if the request encounters an unrecoverable error
         """
-        (operation_timeout, attempt_timeout) = _get_timeouts(
+        operation_timeout, attempt_timeout = _get_timeouts(
             operation_timeout, attempt_timeout, self
         )
         attempt_timeout_gen = _attempt_timeout_generator(
@@ -1290,7 +1289,7 @@ def mutate_row(
             google.api_core.exceptions.GoogleAPIError: raised on non-idempotent operations that cannot be
                 safely retried.
             ValueError: if invalid arguments are provided"""
-        (operation_timeout, attempt_timeout) = _get_timeouts(
+        operation_timeout, attempt_timeout = _get_timeouts(
             operation_timeout, attempt_timeout, self
         )
         if not mutations:
@@ -1362,7 +1361,7 @@ def bulk_mutate_rows(
             MutationsExceptionGroup: if one or more mutations fails
                 Contains details about any failed entries in .exceptions
             ValueError: if invalid arguments are provided"""
-        (operation_timeout, attempt_timeout) = _get_timeouts(
+        operation_timeout, attempt_timeout = _get_timeouts(
             operation_timeout, attempt_timeout, self
         )
         retryable_excs = _get_retryable_errors(retryable_errors, self)
@@ -1415,7 +1414,7 @@ def check_and_mutate_row(
             bool indicating whether the predicate was true or false
         Raises:
             google.api_core.exceptions.GoogleAPIError: exceptions from grpc call"""
-        (operation_timeout, _) = _get_timeouts(operation_timeout, None, self)
+        operation_timeout, _ = _get_timeouts(operation_timeout, None, self)
         if true_case_mutations is not None and (
             not isinstance(true_case_mutations, list)
         ):
@@ -1470,7 +1469,7 @@ def read_modify_write_row(
         Raises:
             google.api_core.exceptions.GoogleAPIError: exceptions from grpc call
             ValueError: if invalid arguments are provided"""
-        (operation_timeout, _) = _get_timeouts(operation_timeout, None, self)
+        operation_timeout, _ = _get_timeouts(operation_timeout, None, self)
         if operation_timeout <= 0:
             raise ValueError("operation_timeout must be greater than 0")
         if rules is not None and (not isinstance(rules, list)):

packages/google-cloud-bigtable/google/cloud/bigtable/data/_sync_autogen/metrics_interceptor.py

@@ -15,18 +15,15 @@
 # This file is automatically generated by CrossSync. Do not edit manually.
 
 from __future__ import annotations
-
 import time
 from functools import wraps
 from typing import Sequence
-
-from grpc import UnaryStreamClientInterceptor, UnaryUnaryClientInterceptor
-
 from google.cloud.bigtable.data._metrics.data_model import (
     ActiveOperationMetric,
     OperationState,
     OperationType,
 )
+from grpc import UnaryStreamClientInterceptor, UnaryUnaryClientInterceptor
 
 
 def _with_active_operation(func):
@@ -54,7 +51,7 @@ def _get_metadata(source) -> dict[str, str | bytes] | None:
     try:
         metadata: Sequence[tuple[str, str | bytes]]
         metadata = source.trailing_metadata() + source.initial_metadata()
-        return {k: v for (k, v) in metadata}
+        return {k: v for k, v in metadata}
     except Exception:
         return None
 

packages/google-cloud-bigtable/google/cloud/bigtable/data/_sync_autogen/mutations_batcher.py

@@ -16,13 +16,11 @@
 # This file is automatically generated by CrossSync. Do not edit manually.
 
 from __future__ import annotations
-
 import atexit
 import concurrent.futures
 import warnings
 from collections import deque
 from typing import TYPE_CHECKING, Sequence, cast
-
 from google.cloud.bigtable.data._cross_sync import CrossSync
 from google.cloud.bigtable.data._helpers import (
     TABLE_DEFAULT,
@@ -39,10 +37,10 @@
 )
 
 if TYPE_CHECKING:
+    from google.cloud.bigtable.data.mutations import RowMutationEntry
     from google.cloud.bigtable.data._sync_autogen.client import (
         _DataApiTarget as TargetType,
     )
-    from google.cloud.bigtable.data.mutations import RowMutationEntry
 _MB_SIZE = 1024 * 1024
 
 
@@ -201,7 +199,7 @@ def __init__(
         batch_retryable_errors: Sequence[type[Exception]]
         | TABLE_DEFAULT = TABLE_DEFAULT.MUTATE_ROWS,
     ):
-        (self._operation_timeout, self._attempt_timeout) = _get_timeouts(
+        self._operation_timeout, self._attempt_timeout = _get_timeouts(
             batch_operation_timeout, batch_attempt_timeout, table
         )
         self._retryable_errors: list[type[Exception]] = _get_retryable_errors(
@@ -210,7 +208,7 @@ def __init__(
         self._closed = CrossSync._Sync_Impl.Event()
         self._target = table
         self._staged_entries: list[RowMutationEntry] = []
-        (self._staged_count, self._staged_bytes) = (0, 0)
+        self._staged_count, self._staged_bytes = (0, 0)
         self._flow_control = CrossSync._Sync_Impl._FlowControl(
             flow_control_max_mutation_count, flow_control_max_bytes
         )
@@ -291,8 +289,8 @@ def _schedule_flush(self) -> CrossSync._Sync_Impl.Future[None] | None:
             Future[None] | None:
                 future representing the background task, if started"""
         if self._staged_entries:
-            (entries, self._staged_entries) = (self._staged_entries, [])
-            (self._staged_count, self._staged_bytes) = (0, 0)
+            entries, self._staged_entries = (self._staged_entries, [])
+            self._staged_count, self._staged_bytes = (0, 0)
             new_task = CrossSync._Sync_Impl.create_task(
                 self._flush_internal, entries, sync_executor=self._sync_flush_executor
             )
@@ -371,14 +369,14 @@ def _raise_exceptions(self):
         Raises:
             MutationsExceptionGroup: exception group with all unreported exceptions"""
         if self._oldest_exceptions or self._newest_exceptions:
-            (oldest, self._oldest_exceptions) = (self._oldest_exceptions, [])
+            oldest, self._oldest_exceptions = (self._oldest_exceptions, [])
             newest = list(self._newest_exceptions)
             self._newest_exceptions.clear()
-            (entry_count, self._entries_processed_since_last_raise) = (
+            entry_count, self._entries_processed_since_last_raise = (
                 self._entries_processed_since_last_raise,
                 0,
             )
-            (exc_count, self._exceptions_since_last_raise) = (
+            exc_count, self._exceptions_since_last_raise = (
                 self._exceptions_since_last_raise,
                 0,
             )

packages/google-cloud-bigtable/google/cloud/bigtable/data/execute_query/_sync_autogen/execute_query_iterator.py

@@ -16,13 +16,10 @@
 # This file is automatically generated by CrossSync. Do not edit manually.
 
 from __future__ import annotations
-
 from typing import TYPE_CHECKING, Any, Dict, Optional, Sequence, Tuple
-
 from google.api_core import retry as retries
 from google.protobuf.internal.enum_type_wrapper import EnumTypeWrapper
 from google.protobuf.message import Message
-
 from google.cloud.bigtable.data._cross_sync import CrossSync
 from google.cloud.bigtable.data._helpers import (
     _attempt_timeout_generator,
@@ -42,9 +39,7 @@
 from google.cloud.bigtable_v2.types.bigtable import (
     ExecuteQueryRequest as ExecuteQueryRequestPB,
 )
-from google.cloud.bigtable_v2.types.bigtable import (
-    ExecuteQueryResponse,
-)
+from google.cloud.bigtable_v2.types.bigtable import ExecuteQueryResponse
 
 if TYPE_CHECKING:
     from google.cloud.bigtable.data import BigtableDataClient as DataClientType

packages/google-cloud-bigtable/test_proxy/handlers/client_handler_data_sync_autogen.py

@@ -190,9 +190,7 @@ async def ExecuteQuery(self, request, **kwargs):
         app_profile_id = self.app_profile_id or request.get("app_profile_id", None)
         query = request.get("query")
         params = request.get("params") or {}
-        (formatted_params, parameter_types) = sql_encoding_helpers.convert_params(
-            params
-        )
+        formatted_params, parameter_types = sql_encoding_helpers.convert_params(params)
         operation_timeout = (
             kwargs.get("operation_timeout", self.per_operation_timeout) or 20
         )