fix cursor

src/database.py

@@ -85,25 +85,63 @@ class DB:
                 # Map transaction control statements
                 if sql_strip.startswith('BEGIN'):
                     self._conn.autocommit = False
-                    with self._conn.cursor() as c:
-                        c.execute('BEGIN;')
-                    # Return dummy wrapper
-                    return _CursorWrapper(c)
+                    cur = self._conn.cursor()
+                    try:
+                        cur.execute('BEGIN;')
+                    finally:
+                        cur.close()
+                    # Return a dummy cursor wrapper
+
+                    class _DummyCursor:
+                        def fetchone(self):
+                            return None
+
+                        def fetchall(self):
+                            return []
+
+                        @property
+                        def rowcount(self):
+                            return 0
+
+                    return _DummyCursor()
+
                 if sql_strip == 'COMMIT':
                     try:
                         self._conn.commit()
                     finally:
                         self._conn.autocommit = True
-                    # Return a cursor with no data
-                    with self._conn.cursor() as c:
-                        return _CursorWrapper(c)
+
+                    class _DummyCursor:
+                        def fetchone(self):
+                            return None
+
+                        def fetchall(self):
+                            return []
+
+                        @property
+                        def rowcount(self):
+                            return 0
+
+                    return _DummyCursor()
+
                 if sql_strip == 'ROLLBACK':
                     try:
                         self._conn.rollback()
                     finally:
                         self._conn.autocommit = True
-                    with self._conn.cursor() as c:
-                        return _CursorWrapper(c)
+
+                    class _DummyCursor:
+                        def fetchone(self):
+                            return None
+
+                        def fetchall(self):
+                            return []
+
+                        @property
+                        def rowcount(self):
+                            return 0
+
+                    return _DummyCursor()
 
                 # Regular statement
                 with self._conn.cursor() as c: