Implement mark_as_read for emails

1 files changed, 9 insertions, 5 deletions

diff --git a/service/pixelated/adapter/pixelated_mail.py b/service/pixelated/adapter/pixelated_mail.py
index cf0e12a5..8ed27262 100644
--- a/service/pixelated/adapter/pixelated_mail.py
+++ b/service/pixelated/adapter/pixelated_mail.py
@@ -33,7 +33,7 @@ class PixelatedMail:
         mail.headers = mail._extract_headers()
         mail.date = PixelatedMail._get_date(mail.headers)
         mail.ident = leap_mail.getUID()
-        mail.status = mail._extract_status()
+        mail.status = set(mail._extract_status())
         mail.security_casing = {}
         mail.tags = mail._extract_tags()
         return mail
@@ -75,6 +75,9 @@ class PixelatedMail:
         self._persist_mail_tags(tags)
         return self.tags
 
+    def mark_as_read(self):
+        self.status.add("read")
+
     def _persist_mail_tags(self, current_tags):
         tags_headers = [tag.name for tag in current_tags]
         hdoc = self.leap_mail.hdoc
@@ -126,8 +129,9 @@ class PixelatedMail:
 
 def from_dict(mail_dict):
     mail = PixelatedMail()
-    mail.headers = mail_dict['header']
-    mail.body = mail_dict['body']
-    mail.ident = mail_dict['ident']
-    mail.tags = mail_dict['tags']
+    mail.headers = mail_dict.get('header', {})
+    mail.body = mail_dict.get('body', '')
+    mail.ident = mail_dict.get('ident', None)
+    mail.tags = mail_dict.get('tags', [])
+    mail.status = set(mail_dict.get('status', []))
     return mail