Update src/utils/sqlite_retry.py

src/utils/sqlite_retry.py

@@ -1,67 +1,33 @@
-from datetime import datetime, timedelta
-import pytz
-
-# Hotel timezone
-OMAN_TZ = pytz.timezone("Asia/Muscat")
-
-
-# -------------------------------------------------
-# Core time helpers
-# -------------------------------------------------
-
-def utc_now_iso() -> str:
-    """
-    Returns current UTC time in ISO-8601 format.
-    Used for all audit-safe timestamps.
-    """
-    return datetime.utcnow().replace(microsecond=0).isoformat() + "Z"
-
-
-def muscat_now() -> datetime:
-    """
-    Returns current time in Asia/Muscat timezone.
-    """
-    return datetime.now(OMAN_TZ)
-
-
-# Backward-compatible alias
-local_now = muscat_now
-
-
-def local_date_str() -> str:
-    """
-    Returns local date (Asia/Muscat) as YYYY-MM-DD.
-    """
-    return muscat_now().strftime("%Y-%m-%d")
-
-
-# -------------------------------------------------
-# Business day & closing logic
-# -------------------------------------------------
-
-def business_day_for_close(now: datetime | None = None) -> str:
-    """
-    Business day closes at 06:00 Asia/Muscat.
-
-    - Before 06:00 → business day = yesterday
-    - After 06:00  → business day = today
-    """
-    if now is None:
-        now = muscat_now()
-
-    if now.hour < 6:
-        return (now.date() - timedelta(days=1)).isoformat()
-    return now.date().isoformat()
-
-
-# Backward-compatible alias
-business_date_for_closing = business_day_for_close
-
-
-def should_auto_close(now: datetime | None = None) -> bool:
-    """
-    Returns True if auto-closing logic should run.
-    """
-    if now is None:
-        now = muscat_now()
-    return now.hour >= 6
+import sqlite3
+import time
+from typing import Callable, TypeVar
+
+T = TypeVar("T")
+
+
+def run_tx(
+    conn: sqlite3.Connection,
+    fn: Callable[[sqlite3.Connection], T],
+    retries: int = 5,
+    delay: float = 0.2,
+) -> T:
+    """
+    Runs a SQLite transaction with retry logic.
+    Handles 'database is locked' errors safely.
+    """
+    last_exc = None
+
+    for attempt in range(retries):
+        try:
+            result = fn(conn)
+            conn.commit()
+            return result
+        except sqlite3.OperationalError as e:
+            conn.rollback()
+            last_exc = e
+            if "locked" in str(e).lower():
+                time.sleep(delay * (attempt + 1))
+                continue
+            raise
+
+    raise last_exc