Added soledad querier

1 files changed, 23 insertions, 0 deletions

diff --git a/service/pixelated/adapter/soledad_querier.py b/service/pixelated/adapter/soledad_querier.py
new file mode 100644
index 00000000..8d0a7072
--- /dev/null
+++ b/service/pixelated/adapter/soledad_querier.py
@@ -0,0 +1,23 @@
+from pixelated.adapter.pixelated_mail import PixelatedMail
+
+
+class SoledadQuerier:
+
+    def __init__(self, soledad):
+        self.soledad = soledad
+
+    def all_mails(self):
+        fdocs_chash = [(fdoc, fdoc.content['chash']) for fdoc in self.soledad.get_from_index('by-type', 'flags')]
+        return self._build_mails_from_fdocs(fdocs_chash)
+
+    def all_mails_by_mailbox(self, mailbox_name):
+        fdocs_chash = [(fdoc, fdoc.content['chash']) for fdoc in self.soledad.get_from_index('by-type-and-mbox', 'flags', mailbox_name)]
+        return self._build_mails_from_fdocs(fdocs_chash)
+
+    def _build_mails_from_fdocs(self, fdocs_chash):
+        fdocs_hdocs = [(f[0], self.soledad.get_from_index('by-type-and-contenthash', 'head', f[1])[0]) for f in fdocs_chash]
+        fdocs_hdocs_phash = [(f[0], f[1], f[1].content.get('body')) for f in fdocs_hdocs]
+        fdocs_hdocs_bdocs = [(f[0], f[1], self.soledad.get_from_index('by-type-and-payloadhash', 'cnt', f[2])[0]) for f in fdocs_hdocs_phash]
+        return [PixelatedMail.from_soledad(*raw_mail) for raw_mail in fdocs_hdocs_bdocs]
+
+