New command 'append': Append results of another (notmuch) query to search buffer

Author: MaxGyver83

alot/buffers/search.py

@@ -22,6 +22,7 @@ def __init__(self, ui, initialquery='', sort_order=None):
         self.dbman = ui.dbman
         self.ui = ui
         self.querystring = initialquery
+        self.querystring_list = [initialquery]
         default_order = settings.get('search_threads_sort_order')
         self.sort_order = sort_order or default_order
         self.result_count = 0
@@ -39,7 +40,7 @@ def __str__(self):
 
     def get_info(self):
         info = {}
-        info['querystring'] = self.querystring
+        info['querystring'] = '" + "'.join(self.querystring_list)
         info['result_count'] = self.result_count
         info['result_count_positive'] = 's' if self.result_count > 1 else ''
         return info
@@ -61,20 +62,27 @@ def rebuild(self, reverse=False):
         if exclude_tags:
             exclude_tags = [t for t in exclude_tags.split(';') if t]
 
-        try:
-            self.result_count = self.dbman.count_messages(self.querystring)
-            threads = self.dbman.get_threads(
-                self.querystring, order, exclude_tags)
-        except NotmuchError:
-            self.ui.notify('malformed query string: %s' % self.querystring,
-                           'error')
-            self.listbox = urwid.ListBox([])
-            self.body = self.listbox
-            return
-
-        self.threadlist = IterableWalker(threads, ThreadlineWidget,
-                                         dbman=self.dbman,
-                                         reverse=reverse)
+        self.result_count = 0
+        self.threadlist = None
+        for query in self.querystring_list:
+            try:
+                self.result_count += self.dbman.count_messages(query)
+                threads = self.dbman.get_threads(
+                    query, order, exclude_tags)
+            except NotmuchError:
+                self.ui.notify('malformed query string: %s' % query,
+                               'error')
+                self.listbox = urwid.ListBox([])
+                self.body = self.listbox
+                return
+
+            iterablewalker = IterableWalker(threads, ThreadlineWidget,
+                                            dbman=self.dbman,
+                                            reverse=reverse)
+            if self.threadlist:
+                self.threadlist.append(iterablewalker)
+            else:
+                self.threadlist = iterablewalker
 
         self.listbox = urwid.ListBox(self.threadlist)
         self.body = self.listbox

alot/commands/globals.py

@@ -112,7 +112,7 @@ def apply(self, ui):
             open_searches = ui.get_buffers_of_type(buffers.SearchBuffer)
             to_be_focused = None
             for sb in open_searches:
-                if sb.querystring == self.query:
+                if sb.querystring_list == [self.query]:
                     to_be_focused = sb
             if to_be_focused:
                 if ui.current_buffer != to_be_focused:

alot/commands/search.py

@@ -35,12 +35,13 @@ def apply(self, ui):
         if not self.thread:
             self.thread = ui.current_buffer.get_selected_thread()
         if self.thread:
-            query = ui.current_buffer.querystring
+            query_list = ui.current_buffer.querystring_list
             logging.info('open thread view for %s', self.thread)
 
             tb = buffers.ThreadBuffer(ui, self.thread)
             ui.buffer_open(tb)
-            tb.unfold_matching(query)
+            for query in query_list:
+                tb.unfold_matching(query)
 
 
 @registerCommand(MODE, 'refine', help='refine query', arguments=[
@@ -72,7 +73,7 @@ def apply(self, ui):
             sbuffer = ui.current_buffer
             oldquery = sbuffer.querystring
             if self.querystring not in [None, oldquery]:
-                sbuffer.querystring = self.querystring
+                sbuffer.querystring_list[-1] = self.querystring
                 sbuffer = ui.current_buffer
             if self.sort_order:
                 sbuffer.sort_order = self.sort_order
@@ -90,13 +91,46 @@ class RefinePromptCommand(Command):
 
     async def apply(self, ui):
         sbuffer = ui.current_buffer
-        oldquery = sbuffer.querystring
+        oldquery = sbuffer.querystring_list[-1]
         return await ui.apply_command(PromptCommand('refine ' + oldquery))
 
 
 RetagPromptCommand = registerCommand(MODE, 'retagprompt')(RetagPromptCommand)
 
 
+@registerCommand(MODE, 'append', usage='append query', arguments=[
+    (['--sort'], {'help': 'sort order', 'choices': [
+        'oldest_first', 'newest_first', 'message_id', 'unsorted']}),
+    (['query'], {'nargs': argparse.REMAINDER, 'help': 'search string'})])
+class AppendCommand(Command):
+
+    """append the results of a new query to this buffer. Search obeys the notmuch
+    :ref:`search.exclude_tags <search.exclude_tags>` setting."""
+    repeatable = True
+
+    def __init__(self, query, sort=None, **kwargs):
+        """
+        :param query: notmuch querystring
+        :type query: str
+        :param sort: how to order results. Must be one of
+                     'oldest_first', 'newest_first', 'message_id' or
+                     'unsorted'.
+        :type sort: str
+        """
+        self.query = ' '.join(query)
+        self.order = sort
+        Command.__init__(self, **kwargs)
+
+    def apply(self, ui):
+        if self.query:
+            sbuffer = ui.current_buffer
+            sbuffer.querystring_list.append(self.query)
+            sbuffer.rebuild()
+            ui.update()
+        else:
+            ui.notify('empty query string')
+
+
 @registerCommand(
     MODE, 'tag', forced={'action': 'add'},
     arguments=[

alot/walker.py

@@ -2,6 +2,7 @@
 # Copyright © 2018 Dylan Baker
 # This file is released under the GNU GPL, version 3 or a later revision.
 # For further details see the COPYING file
+import itertools
 import logging
 import urwid
 
@@ -92,3 +93,6 @@ def _get_next_item(self):
 
     def get_lines(self):
         return self.lines
+
+    def append(self, iterableWalker):
+        self.iterable = itertools.chain(self.iterable, iterableWalker.iterable)