feat: press escape to reset modified state

dooit-org · Sep 19, 2024 · a0361e9 · a0361e9
1 parent ddf6d8a
commit a0361e9
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/dooit/ui/widgets/trees/model_tree.py b/dooit/ui/widgets/trees/model_tree.py
@@ -121,6 +121,12 @@ def stop_edit(self):
         self.current.stop_edit()
         self.app.post_message(ModeChanged("NORMAL"))
 
+    def reset_state(self):
+        """
+        Reset tree of any modified status for e.g. search
+        """
+        self.set_filter("")
+
     async def handle_key(self, event: events.Key) -> bool:
         key = event.key
         if self.is_editing:
@@ -133,6 +139,9 @@ async def handle_key(self, event: events.Key) -> bool:
             self.refresh_options()
             return True
         else:
+            if key == "escape":
+                self.reset_state()
+
             return await super().handle_key(event)
 
     def refresh_options(self) -> None: