deep focus. needed for select in threadbuffer

1 files changed, 12 insertions, 0 deletions

diff --git a/alot/ui.py b/alot/ui.py
index f72779e6..713a621f 100644
--- a/alot/ui.py
+++ b/alot/ui.py
@@ -160,6 +160,18 @@ class UI:
                 self.current_buffer.rebuild()
             self.update()
 
+    def get_deep_focus(self, startfrom=None):
+        """returns focussed leaf in the widget tree"""
+        if not startfrom:
+            startfrom = self.current_buffer
+        if 'get_focus' in dir(startfrom):
+            focus = startfrom.get_focus()
+            if isinstance(focus, tuple):
+                focus = focus[0]
+            if isinstance(focus, urwid.Widget):
+                return self.get_deep_focus(startfrom=focus)
+        return startfrom
+
     def get_buffers_of_type(self, t):
         return filter(lambda x: isinstance(x, t), self.buffers)