Front-end notification model

Context processor to count notifications for display in base template
Logging handler to generate notifications for users with permission

notifications/admin.py

@@ -0,0 +1,3 @@
+from django.contrib import admin
+
+# Register your models here.

notifications/context_processors.py

@@ -0,0 +1,4 @@
+from .models import Notification
+
+def user_notification_count(request):
+    return len(Notification.objects.filter(user=request.user).filter(viewed=False))

notifications/handlers.py

@@ -0,0 +1,14 @@
+import logging
+from django.contrib.auth.models import User
+from .models import Notification
+
+def NotificationHandler(logging.Handler):
+    def emit(self, record):
+        for user in User.objects.all():
+            if user.has_perm('auth.logging_notifications'):
+                notif = Notification()
+                notif.user = user
+                notif.title = "%s [%s:%s]" % (record.levelname, record.funcName, record.lineno)
+                notif.level = str([item[0] for item in Notification.LEVEL_CHOICES if item[1] == record.levelname])
+                notif.message = record.getMessage()
+                notif.save()

notifications/models.py

@@ -0,0 +1,28 @@
+from django.db import models
+from django.contrib.auth.models import User
+
+class Notification(models.Model):
+
+    LEVEL_CHOICES = (
+        ('danger', 'CRITICAL'),
+        ('danger', 'ERROR'),
+        ('warning', 'WARN'),
+        ('info', 'INFO'),
+        ('success', 'DEBUG'),
+    )
+
+    user = models.ForeignKey(User, on_delete=models.CASCADE)
+    level = models.CharField(choices=LEVEL_CHOICES, max_length=10)
+    title = models.CharField(max_length=254)
+    message = models.TextField()
+    created = models.DateTimeField(auto_now_add=True)
+    viewed = models.BooleanField()
+
+    def view(self):
+        logger.info("Marking notification as viewed: %s" % self)
+        self.viewed = True
+        self.save()
+
+    def __unicode__(self):
+        output = "%s: %s" % (self.user, self.title)
+        return output.encode('utf-8')

notifications/tests.py

@@ -0,0 +1,3 @@
+from django.test import TestCase
+
+# Create your tests here.

notifications/views.py

@@ -0,0 +1,26 @@
+from django.shortcuts import render, get_object_or_404
+from .models import Notification
+
+@login_required
+def notification_list(request):
+    logger.debug("notification_list called by user %s" % request.user)
+    new_notifs = Notification.objects.filter(user=request.user).filter(read=False)
+    old_notifs = Notification.objects.filter(user=request.user).filter(read=True)
+    logger.debug("User %s has %s unread and %s read notifications" % (request.user, len(new_notifs), len(old_notifs)))
+    context = {
+        'read': old_notifs,
+        'unread': new_notifs,
+    }
+    return render(request, 'registered/notification_list.html', context)
+
+@login_required
+def notification_view(request, notif_id):
+    logger.debug("notification_view called by user %s for notif_id %s" % (request.user, notif_id))
+    notif = get_object_or_404(notification, pk=notif_id)
+    if notif.user == request.user:
+        logger.debug("Providing notification for user %s" % request.user)
+        context = {'notification': notif}
+        notif.view()
+        return render(request, 'registered/notification_view.html', context)
+    else:
+        logger.warn("User %s not authorized to view notif_id %s belonging to user %s" % (request.user, notif_id, notif.user))