Don't try to rollback if connection was closed

FSX · Aug 26, 2015 · 78175bc · 78175bc
1 parent 5e83907
commit 78175bc
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/momoko/connection.py b/momoko/connection.py
@@ -821,10 +821,10 @@ def exec_statements(future):
                 cursor = future.result()
                 cursors.append(cursor)
             except Exception as error:
-                if not auto_rollback:
-                    transaction_future.set_exc_info(sys.exc_info())
-                else:
+                if auto_rollback and not self.closed:
                     self._rollback(transaction_future, error)
+                else:
+                    transaction_future.set_exc_info(sys.exc_info())
                 return
 
             try: