Hopiu/django-notifications
1
0
Fork 0
mirror of https://github.com/Hopiu/django-notifications.git synced 2026-04-05 14:30:57 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

update queryset methods mark_as_sent() and mark_as_unsent()

Browse source
This commit is contained in:
Aaron Zhao 2017-08-01 14:48:08 +08:00
parent aeb89dc273
commit cf0a11d775
1 changed files with 2 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
notifications/models.py
View file

@ -125,11 +125,13 @@ class NotificationQuerySet(models.query.QuerySet):
return qs.update(deleted=False)
def mark_as_unsent(self, recipient=None):
qs = self.sent()
if recipient:
qs = self.filter(recipient=recipient)
return qs.update(emailed=False)
def mark_as_sent(self, recipient=None):
qs = self.unsent()
if recipient:
qs = self.filter(recipient=recipient)
return qs.update(emailed=True)