Update asyncdb/drivers/pg.py

Co-authored-by: sourcery-ai[bot] <58596630+sourcery-ai[bot]@users.noreply.github.com>
phenobarbital · Oct 24, 2024 · 0d657ad · 0d657ad
1 parent 3c70420
commit 0d657ad
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/asyncdb/drivers/pg.py b/asyncdb/drivers/pg.py
@@ -919,8 +919,10 @@ async def transaction(self):
 
     async def commit(self):
         if self._transaction:
-            await self._transaction.commit()
-        self._transaction = None
+            try:
+                await self._transaction.commit()
+            finally:
+                self._transaction = None
 
     async def rollback(self):
         if self._transaction: