Merge pull request #475 from serokell/krendelhoff/#472-ledger-setup-i…

…nterruption-fail [#472] Handle ledger setup interruption
serokell · May 25, 2022 · 9ed204b · 9ed204b
2 parents e7a2a45 + eb42a52
commit 9ed204b
Showing 1 changed file with 11 additions and 5 deletions.
diff --git a/docker/package/tezos_setup_wizard.py b/docker/package/tezos_setup_wizard.py
@@ -382,12 +382,18 @@ def import_baker_key(self):
 
             key_mode_query = get_key_mode_query(key_import_modes)
 
-            self.import_key(key_mode_query, "Baking")
+            ledger_set_up = False
+            while not ledger_set_up:
+                self.import_key(key_mode_query, "Baking")
 
-            proc_call(
-                f"sudo -u tezos {suppress_warning_text} tezos-client {tezos_client_options} "
-                f"setup ledger to bake for {baker_alias} --main-hwm {self.get_current_head_level()}"
-            )
+                try:
+                    proc_call(
+                        f"sudo -u tezos {suppress_warning_text} tezos-client {tezos_client_options} "
+                        f"setup ledger to bake for {baker_alias} --main-hwm {self.get_current_head_level()}"
+                    )
+                    ledger_set_up = True
+                except PermissionError:
+                    print("Going back to the import mode selection.")
 
     def register_baker(self):
         print()