diff --git a/notifications/models.py b/notifications/models.py index 7d39196..6e470dd 100644 --- a/notifications/models.py +++ b/notifications/models.py @@ -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)