From de9a0666a8856e42d227754f6dc3915f6928af4c Mon Sep 17 00:00:00 2001
From: Sawyer Bergeron <sawyerbergeron@gmail.com>
Date: Mon, 28 Jan 2019 13:07:05 -0500
Subject: Update Inbox Featureset

Inbox now supports marking unread, marking for deletion,
and has a new UI

Change-Id: Ia684056f70ecca31bbd71634f4d1d5415f7e5428
Signed-off-by: Sawyer Bergeron <sawyerbergeron@gmail.com>
---
 src/notifier/views.py | 39 ++++++++++++++++++++++++++++++++-------
 1 file changed, 32 insertions(+), 7 deletions(-)

(limited to 'src/notifier/views.py')

diff --git a/src/notifier/views.py b/src/notifier/views.py
index 4ee757f..3a85eda 100644
--- a/src/notifier/views.py
+++ b/src/notifier/views.py
@@ -7,27 +7,52 @@
 # http://www.apache.org/licenses/LICENSE-2.0
 ##############################################################################
 
-from notifier.models import Notification
 from django.shortcuts import render
+from notifier.models import Notification
+from django.db.models import Q
 
 
 def InboxView(request):
     if request.user.is_authenticated:
         user = request.user
     else:
-        return render(request, "dashboard/login.html", {'title': 'Authentication Required'})
+        return render(request, "dashboard/login.html",
+                      {'title': 'Authentication Required'})
 
-    return render(request, "notifier/inbox.html", {'notifications': Notification.objects.filter(recipients=user.userprofile)})
+    return render(request,
+                  "notifier/inbox.html",
+                  {'unread_notifications': Notification.objects.filter(recipients=user.userprofile).order_by('-id').filter(~Q(read_by=user.userprofile)),
+                      'read_notifications': Notification.objects.filter(recipients=user.userprofile).order_by('-id').filter(read_by=user.userprofile)})
 
 
 def NotificationView(request, notification_id):
+
     if request.user.is_authenticated:
         user = request.user
     else:
-        return render(request, "dashboard/login.html", {'title': 'Authentication Required'})
+        return render(request,
+                      "dashboard/login.html",
+                      {'title': 'Authentication Required'})
 
     notification = Notification.objects.get(id=notification_id)
     if user.userprofile not in notification.recipients.all():
-        return render(request, "dashboard/login.html", {'title': 'Access Denied'})
-
-    return render(request, "notifier/notification.html", {'notification': notification})
+        return render(request,
+                      "dashboard/login.html", {'title': 'Access Denied'})
+
+    notification.read_by.add(user.userprofile)
+    notification.save()
+    if request.method == 'POST':
+        if 'delete' in request.POST:
+            # handle deleting
+            notification.recipients.remove(user.userprofile)
+            if not notification.recipients.exists():
+                notification.delete()
+            else:
+                notification.save()
+
+        if 'unread' in request.POST:
+            notification.read_by.remove(user.userprofile)
+            notification.save()
+
+    return render(request,
+                  "notifier/notification.html", {'notification': notification})
-- 
cgit 1.2.3-korg