From e7db94a986f411e1a86b9a6cc2af21a7e74e1fcf Mon Sep 17 00:00:00 2001 From: Duda Dornelles Date: Tue, 12 Aug 2014 13:30:24 -0300 Subject: adding mails query, tags and contacts to py-fake-service --- py-fake-service/app/adapter/mail.py | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) (limited to 'py-fake-service/app/adapter/mail.py') diff --git a/py-fake-service/app/adapter/mail.py b/py-fake-service/app/adapter/mail.py index 1140f062..859eb50c 100644 --- a/py-fake-service/app/adapter/mail.py +++ b/py-fake-service/app/adapter/mail.py @@ -1,4 +1,5 @@ from datetime import datetime +import random class Mail: @@ -13,14 +14,15 @@ class Mail: def _get_headers(self, mbox_mail): headers = {} headers['from'] = mbox_mail.get_from() - headers['to'] = [mbox_mail.items()['To']] - headers['subject'] = mbox_mail.items()['Subject'] + headers['to'] = [mbox_mail.get('To')] + headers['subject'] = mbox_mail.get('Subject') headers['date'] = datetime.fromtimestamp(random.randrange(1222222222, 1444444444)).isoformat() return headers def _get_tags(self, mbox_mail): - return mbox_mail.items()['X-TW-Pixelated-Tags'].split(', ') - - + return mbox_mail.get('X-TW-Pixelated-Tags').split(', ') + @property + def subject(self): + return self.headers['subject'] -- cgit v1.2.3