Make timer tab specific

charles-001 · Jan 23, 2024 · 503ddf6 · 503ddf6
1 parent 24e9f15
commit 503ddf6
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/dolphie/Modules/TabManager.py b/dolphie/Modules/TabManager.py
@@ -6,6 +6,7 @@
 from dolphie.Widgets.quick_switch import QuickSwitchHostModal
 from textual.app import App
 from textual.containers import Container, Horizontal, VerticalScroll
+from textual.timer import Timer
 from textual.widgets import (
     DataTable,
     Label,
@@ -26,6 +27,7 @@ class Tab:
     dolphie: Dolphie
 
     worker: Worker = None
+    worker_timer: Timer = None
     worker_cancel_error: str = None
 
     main_container: VerticalScroll = None

diff --git a/dolphie/app.py b/dolphie/app.py
@@ -598,15 +598,13 @@ def on_worker_state_changed(self, event: Worker.StateChanged):
                 or tab.id != self.tab.id
                 or dolphie.quick_switched_connection
             ):
-                self.worker_timer = self.set_timer(
-                    tab.dolphie.refresh_interval, partial(self.worker_fetch_data, tab.id)
-                )
+                tab.worker_timer = self.set_timer(tab.dolphie.refresh_interval, partial(self.worker_fetch_data, tab.id))
 
                 return
 
             self.refresh_screen(tab)
 
-            self.worker_timer = self.set_timer(tab.dolphie.refresh_interval, partial(self.worker_fetch_data, tab.id))
+            tab.worker_timer = self.set_timer(tab.dolphie.refresh_interval, partial(self.worker_fetch_data, tab.id))
         elif event.state == WorkerState.CANCELLED:
             # Only show the modal if there's a worker cancel error
             if tab.worker_cancel_error:
@@ -854,7 +852,7 @@ async def capture_key(self, key):
             self.tab.quick_switch_connection()
         elif key == "space":
             if self.tab.worker.state != WorkerState.RUNNING:
-                self.worker_timer.stop()
+                self.tab.worker_timer.stop()
                 self.worker_fetch_data(self.tab.id)
         elif key == "plus":