Some menu items reverted to return status None

selfcustody · Jul 16, 2024 · 03e2807 · 03e2807
1 parent 20de2f0
commit 03e2807
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 13 deletions.
diff --git a/src/krux/pages/__init__.py b/src/krux/pages/__init__.py
@@ -109,7 +109,7 @@ def load_method(self):
                     t("Load from SD card"),
                     None if not self.has_sd_card() else lambda: None,
                 ),
-                ("< " + t("Back"), lambda: MENU_EXIT),
+                ("< " + t("Back"), lambda: None),
             ],
         )
         index, _ = load_menu.run_loop()
@@ -902,15 +902,11 @@ def choose_len_mnemonic(ctx):
     submenu = Menu(
         ctx,
         [
-            (t("12 words"), lambda: MENU_EXIT),
-            (t("24 words"), lambda: MENU_EXIT),
-            ("< " + t("Back"), lambda: MENU_EXIT),
+            (t("12 words"), lambda: 12),
+            (t("24 words"), lambda: 24),
+            ("< " + t("Back"), lambda: None),
         ],
     )
-    index, _ = submenu.run_loop()
+    _, num_words = submenu.run_loop()
     ctx.display.clear()
-    if index == 0:
-        return 12
-    if index == 1:
-        return 24
-    return None
+    return num_words
diff --git a/src/krux/pages/home_pages/home.py b/src/krux/pages/home_pages/home.py
@@ -228,7 +228,7 @@ def _sign_menu(self):
                     t("Sign to SD card"),
                     None if not self.has_sd_card() else lambda: None,
                 ),
-                ("< " + t("Back"), lambda: MENU_EXIT),
+                ("< " + t("Back"), lambda: None),
             ],
         )
         index, _ = sign_menu.run_loop()

diff --git a/src/krux/pages/home_pages/sign_message_ui.py b/src/krux/pages/home_pages/sign_message_ui.py
@@ -24,7 +24,7 @@
 from embit import bip32, compact
 import hashlib
 import binascii
-from .. import MENU_CONTINUE, MENU_EXIT, LOAD_FROM_CAMERA, LOAD_FROM_SD, Menu
+from .. import MENU_CONTINUE, LOAD_FROM_CAMERA, LOAD_FROM_SD, Menu
 from ...themes import theme
 from ...display import (
     DEFAULT_PADDING,
@@ -199,7 +199,7 @@ def sign_message(self):
                     t("Sign to SD card"),
                     None if not self.has_sd_card() else lambda: None,
                 ),
-                ("< " + t("Back"), lambda: MENU_EXIT),
+                ("< " + t("Back"), lambda: None),
             ],
         )
         index, _ = sign_menu.run_loop()