feat: add callback for confirm screen

kraanzu · Feb 11, 2024 · e1afdc1 · e1afdc1
1 parent ad3d838
commit e1afdc1
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/smassh/ui/widgets/settings/option.py b/smassh/ui/widgets/settings/option.py
@@ -82,8 +82,12 @@ def load_current_setting(self) -> None:
     def select_prev_option(self) -> None:
         pass
 
+    def perform(self, ok: bool) -> None:
+        if not ok:
+            return
+
     def select_next_option(self) -> None:
-        self.app.push_screen("confirm")
+        self.app.push_screen("confirm", self.perform)
 
     def render(self) -> RenderableType:
         return self.message