provide retagprompt command in thread mode

retagprompt in search mode allows to retag a whole thread. The new
thread mode version allows to do the same for each selected message.

The common command infers its mode from the UI object that it is passed
when applied. Therefore, the docstring (and everything else) is
mode-agnostic.

When this command command is registered for the wrong mode and applied
it raises a KeyError (on purpose).

1 files changed, 7 insertions, 4 deletions

diff --git a/alot/commands/common.py b/alot/commands/common.py
index 64e68d49..71280d38 100644
--- a/alot/commands/common.py
+++ b/alot/commands/common.py
@@ -9,13 +9,16 @@ from .globals import PromptCommand
 
 class RetagPromptCommand(Command):
 
-    """prompt to retag selected threads\' tags"""
+    """prompt to retag selected thread's or message's tags"""
     def apply(self, ui):
-        thread = ui.current_buffer.get_selected_thread()
-        if not thread:
+        get_selected_item = getattr(ui.current_buffer, {
+                'search': 'get_selected_thread',
+                'thread': 'get_selected_message'}[ui.mode])
+        item = get_selected_item()
+        if not item:
             return
         tags = []
-        for tag in thread.get_tags():
+        for tag in item.get_tags():
             if ' ' in tag:
                 tags.append('"%s"' % tag)
             # skip empty tags