taskapp/utilities/notification.py
changeset 123 a6b4234388c8
parent 78 c5bcafccc135
child 129 e747da8bc110
--- a/taskapp/utilities/notification.py	Fri Feb 26 17:49:26 2010 +0530
+++ b/taskapp/utilities/notification.py	Fri Feb 26 18:34:30 2010 +0530
@@ -1,5 +1,6 @@
+from datetime import datetime
+from django.contrib.auth.models import User
 from pytask.taskapp.models import Notification
-from datetime import datetime
 
 def create_notification(to,subject,message):
     """
@@ -10,7 +11,7 @@
     """
     notification = Notification(sent_date = datetime.now())
     notification.save()
-    notification.to = to
+    notification.sent_to = to
     notification.sub = subject
     notification.message = message
     notification.save()
@@ -39,6 +40,24 @@
         notification = Notification.objects.get(id = notification_id)
     except Notification.DoesNotExist:
         return False
-    notification.deleted = True
+    notification.is_deleted = True
     notification.save()
     return True
+
+def get_notification(nid, user):
+    """ if notification exists, and belongs to the current user, return it.
+    else return None.
+    """
+
+    try:
+        notify_obj = Notification.objects.get(id=nid)
+    except Notification.DoesNotExist:
+        return None
+
+    try:
+        notify_obj.sent_to.get(id=user.id)
+    except User.DoesNotExist:
+        return None
+
+    if not notify_obj.is_deleted:
+        return notify_obj