now requests in sidebar shows the no.of unread notifications.

2 files changed, 21 insertions, 3 deletions

diff --git a/taskapp/utilities/user.py b/taskapp/utilities/user.py
new file mode 100644
index 0000000..bafac0b
--- /dev/null
+++ b/taskapp/utilities/user.py
@@ -0,0 +1,17 @@
+"""
+A collection of utility functions for user.
+"""
+
+def get_user(user):
+    """ get the no of unread requests and notifications and add them as properties for user.
+    """
+
+    unread_notifications = user.notification_sent_to.filter(is_read=False,is_deleted=False).count
+    unread_requests = user.request_sent_to.filter(is_valid=True,is_replied=False).count
+
+    user.unread_notifications = unread_notifications
+    user.unread_requests = unread_requests
+
+    return user
+
+
diff --git a/taskapp/views/user.py b/taskapp/views/user.py
index bc578a7..95448fa 100644
--- a/taskapp/views/user.py
+++ b/taskapp/views/user.py
@@ -14,6 +14,7 @@ from pytask.taskapp.forms.user import UserProfileEditForm
 
 from pytask.taskapp.utilities.request import get_request
 from pytask.taskapp.utilities.notification import get_notification
+from pytask.taskapp.utilities.user import get_user
 
 about = {
     "addmentors":"about/addmentors.html",
@@ -27,8 +28,8 @@ def show_msg(user, message, redirect_url=None, url_desc=None):
 
 def homepage(request):
     """ check for authentication and display accordingly. """
-    
-    user = request.user
+   
+    user = get_user(request.user)
     is_guest = False
     is_mentor = False
     can_create_task = False
@@ -43,7 +44,7 @@ def homepage(request):
         else:
             task_list = Task.objects.order_by('id').reverse()[:10]
             
-        return render_to_response('index.html', {'is_guest':is_guest, 'task_list':task_list})
+        return render_to_response('index.html', {'user':user, 'is_guest':is_guest, 'task_list':task_list})
         
     else:
         user_profile = user.get_profile()