Reorganize loop in level_prompt

jeremyn · Aug 16, 2021 · 0ba98fb · 0ba98fb
1 parent e1fc65a
commit 0ba98fb
Showing 1 changed file with 10 additions and 8 deletions.
diff --git a/src/levelloader.py b/src/levelloader.py
@@ -225,14 +225,16 @@ def level_prompt(self, scrn: curses.window) -> str:
             curses.noecho()
             if raw_choice == QUIT:
                 raise KeyboardInterrupt
-            else:
-                try:
-                    choice = int(raw_choice)
-                    if choice in range(1, len(level_names) + 1):
-                        break
-                except ValueError:
-                    pass
-                prompt = invalid_input_prompt
+
+            try:
+                choice = int(raw_choice)
+                if choice in range(1, len(level_names) + 1):
+                    break
+            except ValueError:
+                pass
+
+            prompt = invalid_input_prompt
+
         choice -= 1
         chosen_level_name = level_names[choice]
         return chosen_level_name.rstrip()