fix: breakage by textual 0.81

kraanzu · Oct 1, 2024 · d607b1a · d607b1a
1 parent 1308189
commit d607b1a
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/smassh/ui/tui.py b/smassh/ui/tui.py
@@ -70,19 +70,21 @@ def show_results(self, event: ShowResults) -> None:
 
         data_parser.add_stats(event.stats, event.failed)
 
-    async def handle_key(self, event: events.Key) -> None:
+    async def handle_key(self, event: events.Key) -> bool:
         visible = self.query_one(ContentSwitcher).visible_content
         if visible:
             await visible.handle_key(event)
 
+        return True
+
 
 class Smassh(App):
     CSS_PATH = str(TARGET_FILE)
     SCREENS = {
-        "main": MainScreen(),
-        "theme": ThemePaletteScreen(),
-        "language": LanguagePaletteScreen(),
-        "confirm": ConfirmScreen(),
+        "main": MainScreen,
+        "theme": ThemePaletteScreen,
+        "language": LanguagePaletteScreen,
+        "confirm": ConfirmScreen,
     }
 
     def __init__(self, *args, **kwargs) -> None: