Add query uid__range

martinrusev · Sep 8, 2018 · 5a52df8 · 5a52df8
1 parent d0b3fa4
commit 5a52df8
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/imbox/query.py b/imbox/query.py
@@ -23,6 +23,7 @@ def build_search_query(**kwargs):
     date__lt = kwargs.get('date__lt', False)
     date__on = kwargs.get('date__on', False)
     subject = kwargs.get('subject')
+    uid__range = kwargs.get('uid__range')
 
     query = []
 
@@ -53,6 +54,9 @@ def build_search_query(**kwargs):
     if subject is not None:
         query.append('(SUBJECT "%s")' % subject)
 
+    if uid__range:
+        query.append('(UID %s)' % uid__range)
+
     if query:
         logger.debug("IMAP query: {}".format(" ".join(query)))
         return " ".join(query)

diff --git a/tests/query_tests.py b/tests/query_tests.py
@@ -48,3 +48,7 @@ def test_date__lt(self):
     def test_date__on(self):
         res = build_search_query(date__on=date(2014, 1, 1))
         self.assertEqual(res, '(ON "01-Jan-2014")')
+
+    def test_uid__range(self):
+        res = build_search_query(uid__range='1000:*')
+        self.assertEqual(res, '(UID 1000:*)')