Add missing get_bytes() method to mailboxes

mailpile · Oct 23, 2014 · 1ef3c99 · 1ef3c99
1 parent 16d2491
commit 1ef3c99
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/mailpile/mailboxes/__init__.py b/mailpile/mailboxes/__init__.py
@@ -118,4 +118,7 @@ def get_msg_size(self, toc_id):
             fd.seek(0, 2)
             return fd.tell()
 
+        def get_bytes(self, toc_id):
+            return self.get_file(toc_id).read()
+
     return UnorderedPicklableMailbox
diff --git a/mailpile/mailboxes/mbox.py b/mailpile/mailboxes/mbox.py
@@ -164,5 +164,8 @@ def get_file_by_ptr(self, msg_ptr):
         # multiple PartialFile objects in flight at once.
         return mailbox._PartialFile(self._get_fd(), start, start + length)
 
+    def get_bytes(self, toc_id):
+        return self.get_file(toc_id).read()
+
 
 mailpile.mailboxes.register(90, MailpileMailbox)