From 809e6abb29f82a7fbe6587d606e8d75283f2a2fe Mon Sep 17 00:00:00 2001
From: Ilya Gurov <ilya.faer@mail.ru>
Date: Fri, 10 Sep 2021 13:45:19 +0300
Subject: [PATCH] fix: rollback failed exception log (#106)

---
 .../cloud/sqlalchemy_spanner/sqlalchemy_spanner.py  | 13 ++++++-------
 1 file changed, 6 insertions(+), 7 deletions(-)

diff --git a/google/cloud/sqlalchemy_spanner/sqlalchemy_spanner.py b/google/cloud/sqlalchemy_spanner/sqlalchemy_spanner.py
index 0e372fa4..45993354 100644
--- a/google/cloud/sqlalchemy_spanner/sqlalchemy_spanner.py
+++ b/google/cloud/sqlalchemy_spanner/sqlalchemy_spanner.py
@@ -800,13 +800,12 @@ def do_rollback(self, dbapi_connection):
         To prevent rollback exception, don't rollback
         committed/rolled back transactions.
         """
-        if (
-            not isinstance(dbapi_connection, spanner_dbapi.Connection)
-            and dbapi_connection.connection._transaction
-            and (
-                dbapi_connection.connection._transaction.rolled_back
-                or dbapi_connection.connection._transaction.committed
-            )
+        if not isinstance(dbapi_connection, spanner_dbapi.Connection):
+            dbapi_connection = dbapi_connection.connection
+
+        if dbapi_connection._transaction and (
+            dbapi_connection._transaction.rolled_back
+            or dbapi_connection._transaction.committed
         ):
             pass
         else: