Update src/utils/sqlite_retry.py

src/utils/sqlite_retry.py

@@ -1,33 +1,67 @@
-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
+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