reinit apps if new key was loaded

src/keystore/flash.py

@@ -238,7 +238,7 @@ async def init(self, show_fn, show_loader):
         await super().init(show_fn, show_loader)
 
     async def storage_menu(self):
-        """Manage storage"""
+        """Manage storage, return True if new key was loaded"""
         buttons = [
             # id, text
             (None, "Key management"),
@@ -254,7 +254,7 @@ async def storage_menu(self):
             # process the menu button:
             # back button
             if menuitem == 255:
-                return
+                return False
             elif menuitem == 0:
                 await self.save_mnemonic()
                 await self.show(
@@ -267,6 +267,7 @@ async def storage_menu(self):
                 await self.show(
                     Alert("Success!", "Your key is loaded.", button_text="OK")
                 )
+                return True
             elif menuitem == 2:
                 await self.delete_mnemonic()
                 await self.show(

src/keystore/memorycard.py

@@ -332,7 +332,7 @@ def hexid(self):
         return hexlify(tagged_hash("smartcard/pubkey", self.applet.card_pubkey)[:4]).decode()
 
     async def storage_menu(self):
-        """Manage storage"""
+        """Manage storage, return True if new key was loaded"""
         enabled = self.connection.isCardInserted()
         buttons = [
             # id, text, enabled, color
@@ -356,7 +356,7 @@ async def storage_menu(self):
             # process the menu button:
             # back button
             if menuitem == 255:
-                return
+                return False
             elif menuitem == 0:
                 await self.save_mnemonic()
                 await self.show(
@@ -371,6 +371,7 @@ async def storage_menu(self):
                 await self.show(
                     Alert("Success!", "Your key is loaded.", button_text="OK")
                 )
+                return True
             elif menuitem == 2:
                 if await self.show(Prompt(
                     "Are you sure?",

src/keystore/sdcard.py

@@ -237,7 +237,7 @@ async def export_mnemonic(self):
 
 
     async def storage_menu(self):
-        """Manage storage"""
+        """Manage storage, return True if new key was loaded"""
         buttons = [
             # id, text
             (None, "Manage keys on SD card and internal flash"),
@@ -256,7 +256,7 @@ async def storage_menu(self):
             # process the menu button:
             # back button
             if menuitem == 255:
-                return
+                return False
             elif menuitem == 0:
                 filename = await self.save_mnemonic()
                 if filename:
@@ -268,6 +268,7 @@ async def storage_menu(self):
                     await self.show(
                         Alert("Success!", "Your key is loaded.", button_text="OK")
                     )
+                return True
             elif menuitem == 2:
                 if await self.delete_mnemonic():
                     await self.show(

src/specter.py

@@ -339,7 +339,10 @@ async def settingsmenu(self):
         if menuitem == 255:
             return self.mainmenu
         elif menuitem == 1:
-            await self.keystore.storage_menu()
+            res = await self.keystore.storage_menu()
+            # storage_menu returns True if app reinit is required
+            if res:
+                self.init_apps()
         elif menuitem == 2:
             pwd = await self.gui.get_input()
             if pwd is None: