🦉 Updates from OwlBot post-processor

See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md

Author: gcf-owl-bot[bot]

docs/conf.py

@@ -24,9 +24,9 @@
 # All configuration values have a default; values that are commented out
 # serve to show the default.
 
+import sys
 import os
 import shlex
-import sys
 
 # If extensions (or modules to document with autodoc) are in another directory,
 # add these directories to sys.path here. If the directory is relative to the

google/cloud/spanner_v1/database.py

@@ -43,7 +43,10 @@
 from google.cloud.spanner_v1.types.spanner import ExecuteSqlRequest
 from google.cloud.spanner_v1.types import RequestOptions
 from google.cloud.spanner_v1.services.spanner import SpannerClient
-from google.cloud.spanner_v1.types.transaction import TransactionOptions, TransactionSelector
+from google.cloud.spanner_v1.types.transaction import (
+    TransactionOptions,
+    TransactionSelector,
+)
 from google.cloud.spanner_v1.types.type import Type, TypeCode
 from google.cloud.spanner_v1._helpers import (
     _merge_query_options,
@@ -466,7 +469,7 @@ def session_options(self) -> SessionOptions:
     @property
     def _pool(self):
         """Backward compatibility property for accessing the session pool.
-        
+
         :rtype: :class:`~google.cloud.spanner_v1.pool.AbstractSessionPool`
         :returns: the session pool from the session manager
         """
@@ -485,14 +488,12 @@ def session(self, labels=None, database_role=None):
         :returns: a new session bound to this database.
         """
         from google.cloud.spanner_v1.session import Session
-        
+
         session_labels = labels or {}
         session_database_role = database_role or self.database_role
-        
+
         return Session(
-            database=self,
-            labels=session_labels,
-            database_role=session_database_role
+            database=self, labels=session_labels, database_role=session_database_role
         )
 
     def metadata_with_request_id(
@@ -1580,12 +1581,13 @@ def _get_session(self):
         from google.cloud.spanner_v1.database import TransactionType
         from google.cloud.spanner_v1.session import Session
 
-        
         if self._session is None:
             database = self._database
-           
+
             if self._session_id is None:
-                session = database._session_manager.get_session(TransactionType.READ_ONLY)
+                session = database._session_manager.get_session(
+                    TransactionType.READ_ONLY
+                )
                 self._session_id = session.session_id
             else:
                 session = Session(database=database)

google/cloud/spanner_v1/keyset.py

@@ -14,7 +14,10 @@
 
 """Wrap representation of Spanner keys / ranges."""
 
-from google.cloud.spanner_v1.types.keys import KeyRange as KeyRangePB, KeySet as KeySetPB
+from google.cloud.spanner_v1.types.keys import (
+    KeyRange as KeyRangePB,
+    KeySet as KeySetPB,
+)
 from google.cloud.spanner_v1._helpers import _make_list_value_pb, _make_list_value_pbs
 
 

google/cloud/spanner_v1/metrics/spanner_metrics_tracer_factory.py

@@ -143,13 +143,14 @@ def _generate_client_hash(client_uid: str) -> str:
         """
         if not client_uid:
             return "000000"
-            
+
         if not HAS_OPENTELEMETRY_INSTALLED:
             # Fallback to a simple hash if mmh3 is not available
             return f"{hash(client_uid) % 0xFFFFFF:06x}"
-            
+
         try:
             import mmh3
+
             hashed_client = mmh3.hash64(client_uid)
 
             # Join the hashes back together since mmh3 splits into high and low 32bits

google/cloud/spanner_v1/param_types.py

@@ -17,7 +17,12 @@
 from google.protobuf.internal.enum_type_wrapper import EnumTypeWrapper
 from google.protobuf.message import Message
 
-from google.cloud.spanner_v1.types.type import StructType, Type, TypeAnnotationCode, TypeCode
+from google.cloud.spanner_v1.types.type import (
+    StructType,
+    Type,
+    TypeAnnotationCode,
+    TypeCode,
+)
 
 # Scalar parameter types
 STRING = Type(code=TypeCode.STRING)

google/cloud/spanner_v1/session.py

@@ -21,7 +21,10 @@
 from google.api_core.exceptions import Aborted, GoogleAPICallError, NotFound
 from google.api_core.gapic_v1 import method
 
-from google.cloud.spanner_v1.types.spanner import CreateSessionRequest, ExecuteSqlRequest
+from google.cloud.spanner_v1.types.spanner import (
+    CreateSessionRequest,
+    ExecuteSqlRequest,
+)
 from google.cloud.spanner_v1._helpers import (
     _delay_until_retry,
     _get_retry_delay,

tests/unit/spanner_dbapi/test_connection.py

@@ -165,15 +165,18 @@ def test__session_checkout(self, mock_database):
         mock_session = mock.MagicMock()
         mock_session_manager.get_session.return_value = mock_session
         mock_database._session_manager = mock_session_manager
-        
+
         connection = Connection(INSTANCE, mock_database)
 
         connection._session_checkout()
-        
+
         # Verify that the session manager's get_session method was called
         # with the correct transaction type (READ_WRITE by default)
         from google.cloud.spanner_v1.session_options import TransactionType
-        mock_session_manager.get_session.assert_called_once_with(TransactionType.READ_WRITE)
+
+        mock_session_manager.get_session.assert_called_once_with(
+            TransactionType.READ_WRITE
+        )
         self.assertEqual(connection._session, mock_session)
 
         connection._session = "db_session"
@@ -191,12 +194,12 @@ def test__release_session(self, mock_database):
         # Mock the session manager and its put_session method
         mock_session_manager = mock.MagicMock()
         mock_database._session_manager = mock_session_manager
-        
+
         connection = Connection(INSTANCE, mock_database)
         connection._session = "session"
 
         connection._release_session()
-        
+
         # Verify that the session manager's put_session method was called
         mock_session_manager.put_session.assert_called_once_with("session")
         self.assertIsNone(connection._session)

tests/unit/test_database.py

@@ -2039,6 +2039,7 @@ def test_list_tables(self):
 
     def session(self, **kwargs):
         from google.cloud.spanner_v1.session import Session
+
         return Session(database=self, **kwargs)
 
     @property
@@ -2469,7 +2470,9 @@ def test_ctor_w_exact_staleness(self):
     def test_from_dict(self):
         klass = self._get_target_class()
         database = self._make_database()
-        session = database._session_manager.get_session.return_value = self._make_session()
+        session = (
+            database._session_manager.get_session.return_value
+        ) = self._make_session()
         snapshot = session.snapshot.return_value = self._make_snapshot()
         api_repr = {
             "session_id": self.SESSION_ID,
@@ -2504,7 +2507,9 @@ def test__get_session_already(self):
 
     def test__get_session_new(self):
         database = self._make_database()
-        session = database._session_manager.get_session.return_value = self._make_session()
+        session = (
+            database._session_manager.get_session.return_value
+        ) = self._make_session()
         batch_txn = self._make_one(database)
         self.assertIs(batch_txn._get_session(), session)
         session.create.assert_called_once_with()
@@ -3476,28 +3481,29 @@ def __init__(self, name, instance=None):
 
         # Create a mock session manager for the new architecture
         self._session_manager = type("MockSessionManager", (), {})()
-        
+
         # Add get_session method to mock session manager
         def mock_get_session(transaction_type=None):
             # Return a session from the pool if available
-            if hasattr(self._session_manager, '_pool') and self._session_manager._pool:
+            if hasattr(self._session_manager, "_pool") and self._session_manager._pool:
                 return self._session_manager._pool.get()
             # Otherwise create a new session
             return _Session(database=self)
-        
+
         def mock_put_session(session):
             # Put session back into the pool if available
-            if hasattr(self._session_manager, '_pool') and self._session_manager._pool:
+            if hasattr(self._session_manager, "_pool") and self._session_manager._pool:
                 try:
                     self._session_manager._pool.put(session)
                 except:
                     pass  # Ignore errors when putting back sessions
-        
+
         self._session_manager.get_session = mock_get_session
         self._session_manager.put_session = mock_put_session
 
     def session(self, **kwargs):
         from google.cloud.spanner_v1.session import Session
+
         return Session(database=self, **kwargs)
 
     @property

tests/unit/test_pool.py

@@ -43,6 +43,7 @@ def _make_database(name="name"):
 
 def _make_session():
     from google.cloud.spanner_v1.session import Session
+
     return mock.create_autospec(Session, instance=True)