Upload reminder_system.py

backend/reminder_system.py

@@ -0,0 +1,73 @@
+import json
+import os
+from datetime import datetime
+from typing import List, Dict, Any
+
+DB_PATH = os.path.join(os.path.dirname(__file__), "database.json")
+
+def _load_db() -> Dict[str, Any]:
+    """Safely load JSON database."""
+    try:
+        with open(DB_PATH, "r") as f:
+            return json.load(f)
+    except Exception:
+        return {"expenses": [], "reminders": []}
+
+def _save_db(data: Dict[str, Any]):
+    """Safely write JSON database."""
+    with open(DB_PATH, "w") as f:
+        json.dump(data, f, indent=4)
+
+def add_reminder(reminder: Dict[str, Any]) -> None:
+    """
+    Add a new reminder.
+    Expected keys: user_id, name, date (YYYY-MM-DD), amount (optional), notes (optional)
+    """
+    db = _load_db()
+    if "reminders" not in db:
+        db["reminders"] = []
+
+    if "date" in reminder and reminder["date"]:
+        try:
+            datetime.strptime(reminder["date"], "%Y-%m-%d")
+        except ValueError:
+            reminder["date"] = datetime.now().strftime("%Y-%m-%d")
+    else:
+        reminder["date"] = datetime.now().strftime("%Y-%m-%d")
+
+    db["reminders"].append(reminder)
+    _save_db(db)
+
+def list_reminders(user_id: str, include_expired: bool = True) -> List[Dict[str, Any]]:
+    """
+    List all reminders for a user.
+    If include_expired=False, only future reminders are returned.
+    Adds 'status' field: 'due', 'upcoming', 'expired'.
+    """
+    db = _load_db()
+    reminders = [r for r in db.get("reminders", []) if r.get("user_id") == user_id]
+
+    today = datetime.now().date()
+    for r in reminders:
+        r_date = datetime.strptime(r.get("date", ""), "%Y-%m-%d").date()
+        if r_date < today:
+            r["status"] = "expired"
+        elif r_date == today:
+            r["status"] = "due"
+        else:
+            r["status"] = "upcoming"
+
+    if not include_expired:
+        reminders = [r for r in reminders if r["status"] != "expired"]
+    reminders_sorted = sorted(reminders, key=lambda x: x.get("date", ""))
+    return reminders_sorted
+
+def due_reminders(user_id: str) -> List[Dict[str, Any]]:
+    """Return reminders that are due today."""
+    all_reminders = list_reminders(user_id)
+    return [r for r in all_reminders if r.get("status") == "due"]
+
+def expired_reminders(user_id: str) -> List[Dict[str, Any]]:
+    """Return reminders that are past due."""
+    all_reminders = list_reminders(user_id)
+    return [r for r in all_reminders if r.get("status") == "expired"]