Skip to content
This repository was archived by the owner on Mar 31, 2026. It is now read-only.

Commit 099ae10

Browse files
gcf-owl-bot[bot]surbhigarg92
authored andcommitted
1 parent 20c8e81 commit 099ae10

File tree

6 files changed

+32
-20
lines changed

6 files changed

+32
-20
lines changed

google/cloud/spanner_v1/__init__.py

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -64,10 +64,7 @@
6464
from .types.type import TypeAnnotationCode
6565
from .types.type import TypeCode
6666
from .data_types import JsonObject
67-
from .transaction import (
68-
BatchTransactionId,
69-
DefaultTransactionOptions
70-
)
67+
from .transaction import BatchTransactionId, DefaultTransactionOptions
7168

7269
from google.cloud.spanner_v1 import param_types
7370
from google.cloud.spanner_v1.client import Client
@@ -152,5 +149,5 @@
152149
"SpannerClient",
153150
"SpannerAsyncClient",
154151
"BatchTransactionId",
155-
"DefaultTransactionOptions"
152+
"DefaultTransactionOptions",
156153
)

google/cloud/spanner_v1/client.py

Lines changed: 11 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -170,7 +170,7 @@ class Client(ClientWithProject):
170170
:type default_transaction_options: :class:`~google.cloud.spanner_v1.DefaultTransactionOptions`
171171
or :class:`dict`
172172
:param default_transaction_options: (Optional) Default options to use for all transactions.
173-
173+
174174
:raises: :class:`ValueError <exceptions.ValueError>` if both ``read_only``
175175
and ``admin`` are :data:`True`
176176
"""
@@ -192,7 +192,7 @@ def __init__(
192192
route_to_leader_enabled=True,
193193
directed_read_options=None,
194194
observability_options=None,
195-
default_transaction_options: Optional[DefaultTransactionOptions] =None,
195+
default_transaction_options: Optional[DefaultTransactionOptions] = None,
196196
):
197197
self._emulator_host = _get_spanner_emulator_host()
198198

@@ -257,7 +257,9 @@ def __init__(
257257
if default_transaction_options is None:
258258
default_transaction_options = DefaultTransactionOptions()
259259
elif not isinstance(default_transaction_options, DefaultTransactionOptions):
260-
raise TypeError("default_transaction_options must be an instance of DefaultTransactionOptions")
260+
raise TypeError(
261+
"default_transaction_options must be an instance of DefaultTransactionOptions"
262+
)
261263

262264
self._default_transaction_options = default_transaction_options
263265

@@ -508,7 +510,9 @@ def directed_read_options(self, directed_read_options):
508510
self._directed_read_options = directed_read_options
509511

510512
@default_transaction_options.setter
511-
def default_transaction_options(self, default_transaction_options:DefaultTransactionOptions):
513+
def default_transaction_options(
514+
self, default_transaction_options: DefaultTransactionOptions
515+
):
512516
"""Sets default_transaction_options for the client
513517
:type default_transaction_options: :class:`~google.cloud.spanner_v1.DefaultTransactionOptions`
514518
or :class:`dict`
@@ -517,6 +521,8 @@ def default_transaction_options(self, default_transaction_options:DefaultTransac
517521
if default_transaction_options is None:
518522
default_transaction_options = DefaultTransactionOptions()
519523
elif not isinstance(default_transaction_options, DefaultTransactionOptions):
520-
raise TypeError("default_transaction_options must be an instance of DefaultTransactionOptions")
524+
raise TypeError(
525+
"default_transaction_options must be an instance of DefaultTransactionOptions"
526+
)
521527

522528
self._default_transaction_options = default_transaction_options

google/cloud/spanner_v1/transaction.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -665,6 +665,9 @@ class BatchTransactionId:
665665
session_id: str
666666
read_timestamp: Any
667667

668+
668669
@dataclass
669670
class DefaultTransactionOptions:
670-
isolation_level: TransactionOptions.IsolationLevel = TransactionOptions.IsolationLevel.ISOLATION_LEVEL_UNSPECIFIED
671+
isolation_level: TransactionOptions.IsolationLevel = (
672+
TransactionOptions.IsolationLevel.ISOLATION_LEVEL_UNSPECIFIED
673+
)

tests/unit/test_client.py

Lines changed: 4 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -16,11 +16,7 @@
1616

1717
import os
1818
import mock
19-
from google.cloud.spanner_v1 import (
20-
DirectedReadOptions,
21-
DefaultTransactionOptions
22-
)
23-
19+
from google.cloud.spanner_v1 import DirectedReadOptions, DefaultTransactionOptions
2420

2521

2622
def _make_credentials():
@@ -57,7 +53,9 @@ class TestClient(unittest.TestCase):
5753
"auto_failover_disabled": True,
5854
},
5955
}
60-
DEFAULT_TRANSACTION_OPTIONS = DefaultTransactionOptions(isolation_level="SERIALIZABLE")
56+
DEFAULT_TRANSACTION_OPTIONS = DefaultTransactionOptions(
57+
isolation_level="SERIALIZABLE"
58+
)
6159

6260
def _get_target_class(self):
6361
from google.cloud import spanner

tests/unit/test_database.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,11 @@
2525
from google.api_core.retry import Retry
2626
from google.protobuf.field_mask_pb2 import FieldMask
2727

28-
from google.cloud.spanner_v1 import RequestOptions, DirectedReadOptions, DefaultTransactionOptions
28+
from google.cloud.spanner_v1 import (
29+
RequestOptions,
30+
DirectedReadOptions,
31+
DefaultTransactionOptions,
32+
)
2933

3034
DML_WO_PARAM = """
3135
DELETE FROM citizens

tests/unit/test_session.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -85,7 +85,9 @@ def _make_one(self, *args, **kwargs):
8585

8686
@staticmethod
8787
def _make_database(
88-
name=DATABASE_NAME, database_role=None, default_transaction_options=DefaultTransactionOptions()
88+
name=DATABASE_NAME,
89+
database_role=None,
90+
default_transaction_options=DefaultTransactionOptions(),
8991
):
9092
database = mock.create_autospec(Database, instance=True)
9193
database.name = name
@@ -1776,7 +1778,9 @@ def test_run_in_transaction_w_isolation_level_at_client(self):
17761778
gax_api = self._make_spanner_api()
17771779
gax_api.begin_transaction.return_value = TransactionPB(id=b"FACEDACE")
17781780
database = self._make_database(
1779-
default_transaction_options=DefaultTransactionOptions(isolation_level="SERIALIZABLE")
1781+
default_transaction_options=DefaultTransactionOptions(
1782+
isolation_level="SERIALIZABLE"
1783+
)
17801784
)
17811785
database.spanner_api = gax_api
17821786
session = self._make_one(database)

0 commit comments

Comments
 (0)