summaryrefslogtreecommitdiff
path: root/py-fake-service/app/adapter/mail_service.py
diff options
context:
space:
mode:
authorBruno Wagner <bwagner@thoughtworks.com>2014-09-15 16:28:31 -0300
committerBruno Wagner <bwagner@thoughtworks.com>2014-09-15 16:28:31 -0300
commit5dc16a1e654e78d9b600578a0e2276cba8d94158 (patch)
tree0b12d78f54a53112e110ecf0e1bcce197f13e3ca /py-fake-service/app/adapter/mail_service.py
parentf2bb13595d67775e8ea89ea595cdbe8b7db96dd8 (diff)
Moved py-fake-service to fake-service, because we only have one now
Diffstat (limited to 'py-fake-service/app/adapter/mail_service.py')
-rw-r--r--py-fake-service/app/adapter/mail_service.py128
1 files changed, 0 insertions, 128 deletions
diff --git a/py-fake-service/app/adapter/mail_service.py b/py-fake-service/app/adapter/mail_service.py
deleted file mode 100644
index 2825af9d..00000000
--- a/py-fake-service/app/adapter/mail_service.py
+++ /dev/null
@@ -1,128 +0,0 @@
-#
-# 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/>.
-import os
-import re
-import mailbox
-
-from tagsset import TagsSet
-from mailset import MailSet
-from contacts import Contacts
-from mail import Mail
-
-
-class MailService:
- MAILSET_PATH = os.path.join(os.environ['HOME'], 'mailsets', 'mediumtagged')
-
- def __init__(self):
- self.mailset = MailSet()
- self.tagsset = TagsSet()
- self.contacts = Contacts()
-
- def reset(self):
- self.mailset = MailSet()
- self.tagsset = TagsSet()
- self.contacts = Contacts()
-
- def _read_file(self, filename):
- with open(filename, 'r') as fd:
- return fd.read()
-
- def _create_message_from_file(self, filename):
- data = self._read_file(filename)
- return self.create_message_from_string(data, filename)
-
- def create_message_from_string(self, data, filename=None):
- if data.startswith('From '):
- msg = mailbox.mboxMessage(data)
- from_addr = re.sub(r"^From ", "", msg.get_unixfrom())
- msg.from_addr = from_addr
- msg.set_from(from_addr)
- else:
- msg = mailbox.Message(data)
- msg.from_addr = msg.get('From')
- return msg
-
- def _create_message_from_string(self, data):
- return mailbox.Message(data)
-
- def load_mailset(self):
- mbox_filenames = [
- filename
- for filename in os.listdir
- (self.MAILSET_PATH) if filename.startswith('mbox')]
- messages = (self._create_message_from_file(os.path.join(self.MAILSET_PATH, mbox))
- for mbox in mbox_filenames)
-
- self.index_messages(messages)
-
- def index_messages(self, messages):
- for message in messages:
- self.mailset.add(message)
- self.tagsset.add(message)
- self.contacts.add(message)
-
- def mails(self, query, page, window_size):
- mails = self.mailset.values()
- mails = [mail for mail in mails if query.test(mail)]
- return sorted(mails, key=lambda mail: mail.date, reverse=True)
-
- def mail(self, mail_id):
- return self.mailset.get(mail_id)
-
- def search_contacts(self, query):
- return self.contacts.search(query)
-
- def mark_as_read(self, mail_id):
- self.mailset.mark_as_read(mail_id)
- self.tagsset.mark_as_read(self.mail(mail_id).tags)
-
- def delete_mail(self, mail_id):
- purged = self.mailset.delete(mail_id)
- if not purged:
- self.tagsset.increment_tag_total_count('trash')
-
- def update_tags_for(self, mail_id, new_tags):
- mail = self.mail(mail_id)
-
- new_tags_set = set(new_tags)
- old_tags_set = set(mail.tags)
-
- increment_set = new_tags_set - old_tags_set
- decrement_set = old_tags_set - new_tags_set
-
- map(lambda x: self.tagsset.increment_tag_total_count(x), increment_set)
- map(lambda x: self.tagsset.decrement_tag_total_count(x), decrement_set)
-
- mail.tags = new_tags
-
- def send(self, mail):
- mail = Mail.from_json(mail)
- self.mailset.update(mail)
- self.tagsset.increment_tag_total_count('sent')
- self.tagsset.decrement_tag_total_count('drafts')
- return mail.ident
-
- def save_draft(self, mail):
- mail = self.mailset.add_draft(Mail.from_json(mail))
- return mail.ident
-
- def update_draft(self, mail):
- mail = Mail.from_json(mail)
- self.mailset.update(mail)
- return mail.ident
-
- def draft_reply_for(self, mail_id):
- return self.mailset.find(draft_reply_for=mail_id)