diff options
Diffstat (limited to 'service/pixelated/adapter/mailboxes.py')
-rw-r--r-- | service/pixelated/adapter/mailboxes.py | 68 |
1 files changed, 68 insertions, 0 deletions
diff --git a/service/pixelated/adapter/mailboxes.py b/service/pixelated/adapter/mailboxes.py new file mode 100644 index 00000000..241a8050 --- /dev/null +++ b/service/pixelated/adapter/mailboxes.py @@ -0,0 +1,68 @@ +# +# Copyright (c) 2014 ThoughtWorks, Inc. +# +# Pixelated is free software: you can redistribute it and/or modify +# it under the terms of the GNU Affero General Public License as published by +# the Free Software Foundation, either version 3 of the License, or +# (at your option) any later version. +# +# Pixelated is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU Affero General Public License for more details. +# +# You should have received a copy of the GNU Affero General Public License +# along with Pixelated. If not, see <http://www.gnu.org/licenses/>. +from pixelated.adapter.mailbox import Mailbox +from pixelated.adapter.mailbox_indexer_listener import MailboxIndexerListener + + +class Mailboxes(): + + def __init__(self, account, soledad_querier): + self.account = account + self.querier = soledad_querier + for mailbox_name in account.mailboxes: + MailboxIndexerListener.listen(self.account, mailbox_name, soledad_querier) + + def _create_or_get(self, mailbox_name): + mailbox_name = mailbox_name.upper() + if mailbox_name not in self.account.mailboxes: + self.account.addMailbox(mailbox_name) + MailboxIndexerListener.listen(self.account, mailbox_name, self.querier) + return Mailbox.create(mailbox_name, self.querier) + + def inbox(self): + return self._create_or_get('INBOX') + + def drafts(self): + return self._create_or_get('DRAFTS') + + def trash(self): + return self._create_or_get('TRASH') + + def sent(self): + return self._create_or_get('SENT') + + def mailboxes(self): + return [self._create_or_get(leap_mailbox_name) for leap_mailbox_name in self.account.mailboxes] + + def mails_by_tag(self, query_tags): + mails = [] + for mailbox in self.mailboxes(): + mails.extend(mailbox.mails_by_tags(query_tags)) + + return mails + + def move_to_trash(self, mail_id): + mail = self.querier.mail(mail_id) + mail.remove_all_tags() + mail.set_mailbox(self.trash().mailbox_name) + mail.save() + return mail + + def mail(self, mail_id): + for mailbox in self.mailboxes(): + mail = mailbox.mail(mail_id) + if mail: + return mail |