From 348f2bc4f727b27abd9878334e886527479d260c Mon Sep 17 00:00:00 2001
From: adityacp
Date: Fri, 17 Apr 2020 16:54:35 +0530
Subject: Merge monitor, grade user and regrade intosingle dashboard

---
 yaksh/middleware/get_notifications.py | 21 +++++++++++++++++++++
 1 file changed, 21 insertions(+)
 create mode 100644 yaksh/middleware/get_notifications.py

(limited to 'yaksh/middleware/get_notifications.py')

diff --git a/yaksh/middleware/get_notifications.py b/yaksh/middleware/get_notifications.py
new file mode 100644
index 0000000..d211ad3
--- /dev/null
+++ b/yaksh/middleware/get_notifications.py
@@ -0,0 +1,21 @@
+from notifications_plugin.models import Notification
+
+class NotificationMiddleware(object):
+    """ Middleware to get user's notifications """
+    def __init__(self, get_response):
+        self.get_response = get_response
+
+    def __call__(self, request):
+        # Code to be executed for each request before
+        # the view (and later middleware) are called.
+        user = request.user
+        if user.is_authenticated:
+            notifications = Notification.objects.get_unread_receiver_notifications(
+                user.id
+            ).count()
+            request.custom_notifications = notifications
+        response = self.get_response(request)
+
+        # Code to be executed for each request/response after
+        # the view is called.
+        return response
-- 
cgit