From 59000afa870987bfddd4a07dcd666179c7152b68 Mon Sep 17 00:00:00 2001
From: Nathan Woodburn <github@nathan.woodburn.au>
Date: Tue, 9 Sep 2025 17:34:36 +1000
Subject: [PATCH] feat: Cleanup function layouts for alerts

---
 main.py | 42 +++++++++++++++++++++++++++---------------
 1 file changed, 27 insertions(+), 15 deletions(-)

diff --git a/main.py b/main.py
index 606de27..941a126 100644
--- a/main.py
+++ b/main.py
@@ -1927,6 +1927,10 @@ def get_alerts(account:str) -> list:
 def add_alert(name:str,output:str,account:str="all"):
     """
     Add an alert to the notifications database.
+    
+    name: Name of the alert
+    output: Message of the alert
+    account: Account to add the alert for (default: all)
     """
     if not os.path.exists("user_data/notifications.db"):
         conn = sqlite3.connect("user_data/notifications.db")
@@ -1945,9 +1949,29 @@ def add_alert(name:str,output:str,account:str="all"):
         print(f"Error adding notification: {e}",flush=True)
         pass
 
+def dismiss_alert(alert_id:int,account:str="all"):
+    """
+    Mark an alert as read.
+
+    alert_id: ID of the alert to dismiss
+    account: Account to dismiss the alert for (default: all)
+    """
+    if not os.path.exists("user_data/notifications.db"):
+        return
+    
+    try:
+        conn = sqlite3.connect("user_data/notifications.db")
+        c = conn.cursor()
+        c.execute("UPDATE notifications SET read=1 WHERE id=?", (alert_id,))
+        conn.commit()
+        conn.close()
+    except Exception as e:
+        print(f"Error dismissing notification: {e}",flush=True)
+        pass
+
 @app.route('/dismiss/<int:alert_id>')
 @app.route('/api/v1/dismiss/<int:alert_id>')
-def dismiss_alert(alert_id):
+def dismiss_alert_route(alert_id):
     # Check if the user is logged in
     if request.cookies.get("account") is None:
         return redirect("/login")
@@ -1956,20 +1980,8 @@ def dismiss_alert(alert_id):
     if not account:
         return redirect("/logout")
 
-    if not os.path.exists("user_data/notifications.db"):
-        return redirect("/")
-
-    try:
-        conn = sqlite3.connect("user_data/notifications.db")
-        c = conn.cursor()
-        c.execute("UPDATE notifications SET read=1 WHERE id=? AND (account=? OR account='all')", (alert_id,account))
-        conn.commit()
-        conn.close()
-    except Exception as e:
-        print(f"Error dismissing notification: {e}",flush=True)
-        pass
-
-    return redirect("/")
+    dismiss_alert(alert_id,account)
+    return redirect(request.referrer or "/")
 
 #endregion