refresh trees when database update

dooit-org · Sep 23, 2024 · 83062d6 · 83062d6
1 parent b22dfaa
commit 83062d6
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/dooit/ui/tui.py b/dooit/ui/tui.py
@@ -9,6 +9,7 @@
 from dooit.ui.widgets.bars import StatusBar
 from dooit.ui.widgets.trees import WorkspacesTree
 from dooit.ui.screens import MainScreen, HelpScreen
+from dooit.ui.widgets.trees.model_tree import ModelTree
 from dooit.utils import CssManager
 from .api import DooitAPI
 from ..api import manager
@@ -67,7 +68,14 @@ def current_theme(self) -> DooitThemeBase:
         return self.api.css_manager.theme
 
     async def poll(self):
-        return
+        def refresh_all_trees():
+            trees = self.query(ModelTree)
+            self.notify(f"{len(trees)}")
+            for tree in trees:
+                tree.force_refresh()
+
+        if manager.has_changed():
+            refresh_all_trees()
 
     @on(DooitEvent)
     def global_message(self, event: DooitEvent):