summaryrefslogtreecommitdiff
path: root/py-fake-service
diff options
context:
space:
mode:
authorVictor Shyba <victor.shyba@gmail.com>2014-08-14 18:28:03 -0300
committerVictor Shyba <victor.shyba@gmail.com>2014-08-14 18:28:03 -0300
commit3e3ab15d98bf7ca866e0ff40e6adf43902a5eac3 (patch)
treed4591fc2606c13a8d4f0f9e0d1dc63ec15c7f63a /py-fake-service
parentea88d617a0d6a942850922664590f80d32de7d87 (diff)
fix py-fake-service by reverting js changes and sendind lowercase content type
Diffstat (limited to 'py-fake-service')
-rw-r--r--py-fake-service/app/adapter/mail.py17
-rw-r--r--py-fake-service/app/adapter/mail_service.py14
2 files changed, 16 insertions, 15 deletions
diff --git a/py-fake-service/app/adapter/mail.py b/py-fake-service/app/adapter/mail.py
index 1cbe065f..a6dee92e 100644
--- a/py-fake-service/app/adapter/mail.py
+++ b/py-fake-service/app/adapter/mail.py
@@ -18,7 +18,7 @@ class Mail:
mail.draft_reply_for = mail_json.get('draft_reply_for', 0)
return mail
-
+
def __init__(self, mbox_mail=None, ident=None):
if mbox_mail:
self.header = self._get_headers(mbox_mail)
@@ -31,11 +31,14 @@ class Mail:
def _get_body(self, message):
if message.is_multipart():
- boundary = message.get_boundary()
- start_boundary = '--'+boundary
- join_boundary = start_boundary+'\n'
+ boundary = '--{boundary}'.format(boundary= message.get_boundary().strip())
+ body_parts = [x.as_string() for x in message.get_payload()]
+
+ body = boundary + '\n'
+ body += '{boundary}\n'.format(boundary=boundary).join(body_parts)
+ body += '{boundary}--\n'.format(boundary=boundary)
- return join_boundary + join_boundary.join([x.as_string() for x in message.get_payload()]) + start_boundary+ '--'
+ return body
else:
return message.get_payload()
@@ -53,14 +56,14 @@ class Mail:
headers['to'] = [mbox_mail.get('To')]
headers['subject'] = mbox_mail.get('Subject')
headers['date'] = datetime.fromtimestamp(random.randrange(1222222222, self.NOW)).isoformat()
- headers['Content-Type'] = mbox_mail.get('Content-Type')
+ headers['content_type'] = mbox_mail.get('Content-Type')
return headers
def _get_tags(self, mbox_mail):
return mbox_mail.get('X-TW-Pixelated-Tags').split(', ')
-
+
@property
def subject(self):
return self.header['subject']
diff --git a/py-fake-service/app/adapter/mail_service.py b/py-fake-service/app/adapter/mail_service.py
index 7c2570f9..41195c0a 100644
--- a/py-fake-service/app/adapter/mail_service.py
+++ b/py-fake-service/app/adapter/mail_service.py
@@ -16,12 +16,10 @@ class MailService:
def load_mailset(self):
mbox_filenames = [filename for filename in os.listdir(self.MAILSET_PATH) if filename.startswith('mbox')]
- boxes = (mailbox.mbox(os.path.join(self.MAILSET_PATH, mbox)) for mbox in mbox_filenames)
+ boxes = (mailbox.mbox(os.path.join(self.MAILSET_PATH, mbox)) for mbox in mbox_filenames)
for box in boxes:
message = box.popitem()[1]
- if message.is_multipart():
- continue
self.mailset.add(message)
self.tagsset.add(message)
self.contacts.add(message)
@@ -49,13 +47,13 @@ class MailService:
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
+ 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)
@@ -71,8 +69,8 @@ class MailService:
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)