diff --git a/sen/tui/widgets/list/base.py b/sen/tui/widgets/list/base.py index 88b1cf3..3e242f1 100644 --- 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)