fix(migrate): raise exception even if db is not available (#22922) (#…

…22999)

* fix(migrate): raise exception even if db is not available

* fix: wrap correctly

(cherry picked from commit c354b31)

Co-authored-by: Ankush Menat <ankush@frappe.io>

frappe/migrate.py

@@ -1,6 +1,8 @@
 # Copyright (c) 2022, Frappe Technologies Pvt. Ltd. and Contributors
 # License: MIT. See LICENSE
 
+import contextlib
+import functools
 import json
 import os
 from textwrap import dedent
@@ -36,14 +38,18 @@
 
 
 def atomic(method):
+	@functools.wraps(method)
 	def wrapper(*args, **kwargs):
 		try:
 			ret = method(*args, **kwargs)
 			frappe.db.commit()
 			return ret
-		except Exception:
-			frappe.db.rollback()
-			raise
+		except Exception as e:
+			# database itself can be gone while attempting rollback.
+			# We should preserve original exception in this case.
+			with contextlib.suppress(Exception):
+				frappe.db.rollback()
+			raise e
 
 	return wrapper