diff options
Diffstat (limited to 'taskapp/utilities/user.py')
-rw-r--r-- | taskapp/utilities/user.py | 17 |
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 + + |