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

Commit 457117d

Browse files
gcf-owl-bot[bot]surbhigarg92
authored andcommitted
1 parent 4f8af2f commit 457117d

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
@@ -166,7 +166,7 @@ class Client(ClientWithProject):
166166
:type default_transaction_options: :class:`~google.cloud.spanner_v1.DefaultTransactionOptions`
167167
or :class:`dict`
168168
:param default_transaction_options: (Optional) Default options to use for all transactions.
169-
169+
170170
:raises: :class:`ValueError <exceptions.ValueError>` if both ``read_only``
171171
and ``admin`` are :data:`True`
172172
"""
@@ -188,7 +188,7 @@ def __init__(
188188
route_to_leader_enabled=True,
189189
directed_read_options=None,
190190
observability_options=None,
191-
default_transaction_options: Optional[DefaultTransactionOptions] =None,
191+
default_transaction_options: Optional[DefaultTransactionOptions] = None,
192192
):
193193
self._emulator_host = _get_spanner_emulator_host()
194194

@@ -253,7 +253,9 @@ def __init__(
253253
if default_transaction_options is None:
254254
default_transaction_options = DefaultTransactionOptions()
255255
elif not isinstance(default_transaction_options, DefaultTransactionOptions):
256-
raise TypeError("default_transaction_options must be an instance of DefaultTransactionOptions")
256+
raise TypeError(
257+
"default_transaction_options must be an instance of DefaultTransactionOptions"
258+
)
257259

258260
self._default_transaction_options = default_transaction_options
259261

@@ -504,7 +506,9 @@ def directed_read_options(self, directed_read_options):
504506
self._directed_read_options = directed_read_options
505507

506508
@default_transaction_options.setter
507-
def default_transaction_options(self, default_transaction_options:DefaultTransactionOptions):
509+
def default_transaction_options(
510+
self, default_transaction_options: DefaultTransactionOptions
511+
):
508512
"""Sets default_transaction_options for the client
509513
:type default_transaction_options: :class:`~google.cloud.spanner_v1.DefaultTransactionOptions`
510514
or :class:`dict`
@@ -513,6 +517,8 @@ def default_transaction_options(self, default_transaction_options:DefaultTransac
513517
if default_transaction_options is None:
514518
default_transaction_options = DefaultTransactionOptions()
515519
elif not isinstance(default_transaction_options, DefaultTransactionOptions):
516-
raise TypeError("default_transaction_options must be an instance of DefaultTransactionOptions")
520+
raise TypeError(
521+
"default_transaction_options must be an instance of DefaultTransactionOptions"
522+
)
517523

518524
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
@@ -656,6 +656,9 @@ class BatchTransactionId:
656656
session_id: str
657657
read_timestamp: Any
658658

659+
659660
@dataclass
660661
class DefaultTransactionOptions:
661-
isolation_level: TransactionOptions.IsolationLevel = TransactionOptions.IsolationLevel.ISOLATION_LEVEL_UNSPECIFIED
662+
isolation_level: TransactionOptions.IsolationLevel = (
663+
TransactionOptions.IsolationLevel.ISOLATION_LEVEL_UNSPECIFIED
664+
)

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)