#216 when trying to save a draft too fast (before the previous version is saved) simply ignore the attempt

3 files changed, 23 insertions, 14 deletions

diff --git a/service/pixelated/adapter/services/mail_service.py b/service/pixelated/adapter/services/mail_service.py
index f4f37d77..cb92228c 100644
--- a/service/pixelated/adapter/services/mail_service.py
+++ b/service/pixelated/adapter/services/mail_service.py
@@ -41,6 +41,9 @@ class MailService:
     def mail(self, mail_id):
         return self.mailboxes.mail(mail_id)
 
+    def mail_exists(self, mail_id):
+        return not(not(self.querier.get_header_by_chash(mail_id)))
+
     def send(self, last_draft_ident, mail):
         self.mail_sender.sendmail(mail)
         if last_draft_ident:
diff --git a/service/pixelated/adapter/soledad/soledad_facade_mixin.py b/service/pixelated/adapter/soledad/soledad_facade_mixin.py
index 33588d1e..1df038ea 100644
--- a/service/pixelated/adapter/soledad/soledad_facade_mixin.py
+++ b/service/pixelated/adapter/soledad/soledad_facade_mixin.py
@@ -23,17 +23,20 @@ class SoledadDbFacadeMixin(object):
     def get_all_flags_by_mbox(self, mbox):
         return self.soledad.get_from_index('by-type-and-mbox', 'flags', mbox)
 
-    def get_all_headers_by_chash(self, chash):
-        return self.soledad.get_from_index('by-type-and-contenthash', 'head', chash)
-
     def get_content_by_phash(self, phash):
-        return self.soledad.get_from_index('by-type-and-payloadhash', 'cnt', phash)
+        content = self.soledad.get_from_index('by-type-and-payloadhash', 'cnt', phash)
+        if len(content):
+            return content[0]
 
     def get_flags_by_chash(self, chash):
-        return self.soledad.get_from_index('by-type-and-contenthash', 'flags', chash)[0]
+        flags = self.soledad.get_from_index('by-type-and-contenthash', 'flags', chash)
+        if len(flags):
+            return flags[0]
 
     def get_header_by_chash(self, chash):
-        return self.soledad.get_from_index('by-type-and-contenthash', 'head', chash)[0]
+        header = self.soledad.get_from_index('by-type-and-contenthash', 'head', chash)
+        if len(header):
+            return header[0]
 
     def get_recent_by_mbox(self, mbox):
         return self.soledad.get_from_index('by-type-and-mbox', 'rct', mbox)
diff --git a/service/pixelated/adapter/soledad/soledad_reader_mixin.py b/service/pixelated/adapter/soledad/soledad_reader_mixin.py
index 9cb20a9a..24010e2d 100644
--- a/service/pixelated/adapter/soledad/soledad_reader_mixin.py
+++ b/service/pixelated/adapter/soledad/soledad_reader_mixin.py
@@ -34,26 +34,29 @@ class SoledadReaderMixin(SoledadDbFacadeMixin, object):
 
         fdocs_hdocs = []
         for fdoc, chash in fdocs_chash:
-            hdoc = self.get_all_headers_by_chash(chash)
-            if len(hdoc) == 0:
+            hdoc = self.get_header_by_chash(chash)
+            if not hdoc:
                 continue
-            fdocs_hdocs.append((fdoc, hdoc[0]))
+            fdocs_hdocs.append((fdoc, hdoc))
 
         fdocs_hdocs_bodyphash = [(f[0], f[1], f[1].content.get('body')) for f in fdocs_hdocs]
         fdocs_hdocs_bdocs_parts = []
         for fdoc, hdoc, body_phash in fdocs_hdocs_bodyphash:
             bdoc = self.get_content_by_phash(body_phash)
-            if len(bdoc) == 0:
+            if not bdoc:
                 continue
             parts = self._extract_parts(hdoc.content)
-            fdocs_hdocs_bdocs_parts.append((fdoc, hdoc, bdoc[0], parts))
+            fdocs_hdocs_bdocs_parts.append((fdoc, hdoc, bdoc, parts))
 
         return [PixelatedMail.from_soledad(*raw_mail, soledad_querier=self) for raw_mail in fdocs_hdocs_bdocs_parts]
 
+    def mail_exists(self, ident):
+        return self.get_flags_by_chash(ident)
+
     def mail(self, ident):
         fdoc = self.get_flags_by_chash(ident)
         hdoc = self.get_header_by_chash(ident)
-        bdoc = self.get_content_by_phash(hdoc.content['body'])[0]
+        bdoc = self.get_content_by_phash(hdoc.content['body'])
         parts = self._extract_parts(hdoc.content)
 
         return PixelatedMail.from_soledad(fdoc, hdoc, bdoc, parts=parts, soledad_querier=self)
@@ -65,7 +68,7 @@ class SoledadReaderMixin(SoledadDbFacadeMixin, object):
         return self._build_mails_from_fdocs(fdocs_chash)
 
     def attachment(self, attachment_ident, encoding):
-        bdoc = self.get_content_by_phash(attachment_ident)[0]
+        bdoc = self.get_content_by_phash(attachment_ident)
         return {'content': self._try_decode(bdoc.content['raw'], encoding),
                 'content-type': bdoc.content['content-type']}
 
@@ -94,7 +97,7 @@ class SoledadReaderMixin(SoledadDbFacadeMixin, object):
         return parts
 
     def _extract_alternative(self, hdoc, headers_dict):
-        bdoc = self.get_content_by_phash(hdoc['phash'])[0]
+        bdoc = self.get_content_by_phash(hdoc['phash'])
         raw_content = bdoc.content['raw']
         return {'headers': headers_dict, 'content': raw_content}