Harden PostgreSQL deploy: auto-rollback on SQL errors, clean DB on migrate.

Co-authored-by: Cursor <cursoragent@cursor.com>
2026-07-01 08:13:51 +08:00
parent bc79ad308c
commit a1f22624de
3 changed files with 19 additions and 3 deletions
@@ -197,7 +197,11 @@ class DbConnection:

    def execute(self, sql: str, params: Sequence[Any] | None = None) -> DbCursor:
        cur = self.cursor()
-        return cur.execute(sql, params)
+        try:
+            return cur.execute(sql, params)
+        except Exception:
+            rollback_if_postgres(self)
+            raise

    def cursor(self) -> DbCursor:
        if self._backend == "sqlite":