# # 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 . from pixelated.adapter.pixelated_mailbox import PixelatedMailbox from pixelated.adapter.soledad_querier import SoledadQuerier class PixelatedMailBoxes(): def __init__(self, account): self.account = account self.querier = SoledadQuerier.get_instance() 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) return PixelatedMailbox.create(mailbox_name) 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') @property def mailboxes(self): return [PixelatedMailbox.create(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