Hopiu/django-notifications
1
0
Fork 0
mirror of https://github.com/Hopiu/django-notifications.git synced 2026-03-16 21:30:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

sent/unsent convenience functions

Browse source
This commit is contained in:
Roman Vasilyev 2017-02-23 09:27:00 -08:00
parent 0039065461
commit f7f90b83c8
1 changed files with 12 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
notifications/models.py
View file

@ -40,6 +40,12 @@ def assert_soft_delete():
class NotificationQuerySet(models.query.QuerySet):
def unsent(self):
return self.filter(emailed=False)
def sent(self):
return self.filter(emailed=True)
def unread(self, include_deleted=False):
"""Return only unread items in the current queryset"""
if is_soft_delete() and not include_deleted:
@ -117,6 +123,12 @@ class NotificationQuerySet(models.query.QuerySet):
return qs.update(deleted=False)
def mark_as_unsent(self):
return self.update(emailed=False)
def mark_as_sent(self):
return self.update(emailed=True)
class Notification(models.Model):
"""