searching/filtering for logs

TomasTomecek · Jun 29, 2016 · 3aaf08c · 3aaf08c
1 parent 887e438
commit 3aaf08c
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/sen/tui/widgets/list/base.py b/sen/tui/widgets/list/base.py
@@ -60,7 +60,11 @@ def _search(self, reverse_search=False):
             if hasattr(obj, "matches_search"):
                 condition = obj.matches_search(self.search_string)
             else:
-                condition = self.search_string in obj.original_widget.text
+                if hasattr(obj, "original_widget"):
+                    text = obj.original_widget.text
+                else:
+                    text = obj.text
+                condition = self.search_string in text
             if condition:
                 self.set_focus(pos)
                 self.reload_widget()
@@ -81,7 +85,11 @@ def filter(self, s, widgets_to_filter=None):
             if hasattr(obj, "matches_search"):
                 condition = obj.matches_search(s)
             else:
-                condition = s in obj.original_widget.text
+                if hasattr(obj, "original_widget"):
+                    text = obj.original_widget.text
+                else:
+                    text = obj.text
+                condition = s in text
 
             if condition:
                 widgets.append(obj)