diff options
Diffstat (limited to 'service/pixelated/adapter/services')
-rw-r--r-- | service/pixelated/adapter/services/draft_service.py | 8 | ||||
-rw-r--r-- | service/pixelated/adapter/services/mailbox.py | 3 |
2 files changed, 7 insertions, 4 deletions
diff --git a/service/pixelated/adapter/services/draft_service.py b/service/pixelated/adapter/services/draft_service.py index ddb86c5c..df295eec 100644 --- a/service/pixelated/adapter/services/draft_service.py +++ b/service/pixelated/adapter/services/draft_service.py @@ -22,13 +22,13 @@ class DraftService(object): self._mailboxes = mailboxes def create_draft(self, input_mail): - self._drafts().add(input_mail) - return input_mail + pixelated_mail = self._drafts().add(input_mail) + return pixelated_mail def update_draft(self, ident, input_mail): - new_mail = self.create_draft(input_mail) + pixelated_mail = self.create_draft(input_mail) self._drafts().remove(ident) - return new_mail + return pixelated_mail def _drafts(self): return self._mailboxes.drafts() diff --git a/service/pixelated/adapter/services/mailbox.py b/service/pixelated/adapter/services/mailbox.py index fbdbfc30..508dec31 100644 --- a/service/pixelated/adapter/services/mailbox.py +++ b/service/pixelated/adapter/services/mailbox.py @@ -43,6 +43,9 @@ class Mailbox: def remove(self, ident): mail = self.querier.mail(ident) + self.remove_mail(mail) + + def remove_mail(self, mail): mail.remove_all_tags() self.querier.remove_mail(mail) |