diff options
| author | Duda Dornelles <ddornell@thoughtworks.com> | 2014-12-23 11:28:48 -0200 |
|---|---|---|
| committer | Duda Dornelles <ddornell@thoughtworks.com> | 2014-12-23 11:28:48 -0200 |
| commit | 0047054a53cb88bbe1b4cf2de0ebdc26daa96ade (patch) | |
| tree | d2f8616eab4fe22c91f5d01e57d4203c6a3bb3f0 /service/pixelated/adapter/services | |
| parent | 51ec3f9edc355f4b824f5eefae19f9824dde0bf8 (diff) | |
Refactoring soledad querier to use mixins
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) |