add machanism to replace root until keypress

2 files changed, 22 insertions, 0 deletions

diff --git a/alot/ui.py b/alot/ui.py
index 320f8efe..9fd3b7ab 100644
--- a/alot/ui.py
+++ b/alot/ui.py
@@ -82,6 +82,12 @@ class UI(object):
     def keypress(self, key):
         self.logger.debug('unhandeled input: %s' % key)
 
+    def show_as_root_until_keypress_(self, w, key):
+        def oe():
+            self.mainloop.widget = self.mainframe  # restore main screen
+        self.mainloop.widget = widgets.CatchKeyWidgetWrap(w, key,
+                                                          on_catch=oe)
+
     def prompt(self, prefix='>', text=u'', completer=None, tab=0, history=[]):
         """prompt for text input
 
diff --git a/alot/widgets.py b/alot/widgets.py
index ab19827f..d17bbfc3 100644
--- a/alot/widgets.py
+++ b/alot/widgets.py
@@ -52,6 +52,22 @@ class DialogBox(urwid.WidgetWrap):
         return self.body.keypress(size, key)
 
 
+class CatchKeyWidgetWrap(urwid.WidgetWrap):
+    def __init__(self, widget, key, on_catch):
+        urwid.WidgetWrap.__init__(self, widget)
+        self.key = key
+        self.on_catch = on_catch
+
+    def selectable(self):
+        return True
+
+    def keypress(self, size, key):
+        if command_map[key] == self.key:
+            self.on_catch()
+        elif self._w.selectable():
+            return self._w.keypress(size, key)
+
+
 class ThreadlineWidget(urwid.AttrMap):
     def __init__(self, tid, dbman):
         self.dbman = dbman