From e3871a45b584e8dcb633ba314e1ab1b8593c636b Mon Sep 17 00:00:00 2001 From: Duda Dornelles Date: Thu, 18 Sep 2014 18:30:28 -0300 Subject: Adding update_draft to mailboxes --- service/pixelated/adapter/pixelated_mail.py | 7 +++---- service/pixelated/adapter/pixelated_mailboxes.py | 8 +++++++- 2 files changed, 10 insertions(+), 5 deletions(-) (limited to 'service/pixelated') diff --git a/service/pixelated/adapter/pixelated_mail.py b/service/pixelated/adapter/pixelated_mail.py index af82eb7f..5c3025b6 100644 --- a/service/pixelated/adapter/pixelated_mail.py +++ b/service/pixelated/adapter/pixelated_mail.py @@ -26,11 +26,13 @@ from email.MIMEText import MIMEText class PixelatedMail: def __init__(self): - self.body = '' + self.body = None self.headers = {} self.status = [] self.security_casing = {} self.tags = [] + self.mailbox_name = None + self.uid = None @staticmethod def from_leap_mail(leap_mail, tag_service=TagService.get_instance()): @@ -149,9 +151,6 @@ class PixelatedMail: mime_multipart['From'] = PixelatedMail.from_email_address return mime_multipart.as_string() - def set_ident(self, mailbox_name, uid): - self.ident = gen_pixelated_uid(mailbox_name, uid) - @staticmethod def from_dict(mail_dict): return from_dict(mail_dict) diff --git a/service/pixelated/adapter/pixelated_mailboxes.py b/service/pixelated/adapter/pixelated_mailboxes.py index e8b302a7..e8d73b24 100644 --- a/service/pixelated/adapter/pixelated_mailboxes.py +++ b/service/pixelated/adapter/pixelated_mailboxes.py @@ -33,9 +33,15 @@ class PixelatedMailBoxes(): def add_draft(self, mail): drafts = self.drafts() draft_id = drafts.add(mail) - mail.set_ident(drafts.mailbox_name, draft_id) + mail.mailbox_name = drafts.mailbox_name + mail.uid = draft_id return mail + def update_draft(self, mail): + new_mail = self.add_draft(mail) + self.drafts().remove(mail) + return new_mail + def move_to_trash(self, mail): mail.remove_all_tags() origin_mailbox = mail.mailbox_name -- cgit v1.2.3