summaryrefslogtreecommitdiff
path: root/taskapp/utilities/user.py
diff options
context:
space:
mode:
Diffstat (limited to 'taskapp/utilities/user.py')
-rw-r--r--taskapp/utilities/user.py17
1 files changed, 17 insertions, 0 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
+
+