From 52fdc8c2120e459772e221c68b25bf6e7060f6f8 Mon Sep 17 00:00:00 2001
From: rubenwardy <rw@rubenwardy.com>
Date: Tue, 29 May 2018 17:20:11 +0100
Subject: [PATCH] Add clear all button to notifications page

---
 app/templates/notifications/list.html | 6 ++++++
 app/views/notifications.py            | 9 ++++++++-
 2 files changed, 14 insertions(+), 1 deletion(-)

diff --git a/app/templates/notifications/list.html b/app/templates/notifications/list.html
index fb133e8..d6de54e 100644
--- a/app/templates/notifications/list.html
+++ b/app/templates/notifications/list.html
@@ -5,6 +5,12 @@ Notifications
 {% endblock %}
 
 {% block content %}
+	{% if current_user.notifications %}
+		<form method="post" action="{{ url_for('clear_notifications_page') }}">
+			<input type="hidden" name="csrf_token" value="{{ csrf_token() }}" />
+			<input type="submit" value="Clear All" />
+		</form>
+	{% endif %}
 	<ul>
 		{% for n in current_user.notifications %}
 			<li><a href="{{ n.url }}">
diff --git a/app/views/notifications.py b/app/views/notifications.py
index 4515ae5..23dbb31 100644
--- a/app/views/notifications.py
+++ b/app/views/notifications.py
@@ -23,4 +23,11 @@ from app.models import *
 @app.route("/notifications/")
 @login_required
 def notifications_page():
-    return render_template("notifications/list.html")
+	return render_template("notifications/list.html")
+
+@app.route("/notifications/clear/", methods=["POST"])
+@login_required
+def clear_notifications_page():
+	current_user.notifications.clear()
+	db.session.commit()
+	return redirect(url_for("notifications_page"))