Adding update_draft to mailboxes

2 files changed, 10 insertions, 5 deletions

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