From d70f0daa22e209b908964a192400afd1d5533590 Mon Sep 17 00:00:00 2001 From: derrod Date: Thu, 30 Dec 2021 19:05:23 +0100 Subject: [PATCH] [utils] Fix get_integer_choice behaviour In my defense, the original function was 100% generated by GitHub Copilot. --- legendary/utils/cli.py | 16 ++++++++-------- 1 file changed, 8 insertions(+), 8 deletions(-) diff --git a/legendary/utils/cli.py b/legendary/utils/cli.py index db46bfc..a52b305 100644 --- a/legendary/utils/cli.py +++ b/legendary/utils/cli.py @@ -21,15 +21,15 @@ def get_int_choice(prompt, default=None, min_choice=None, max_choice=None, retur while True: try: - choice = int(input(prompt)) - except ValueError: - if default is not None: + inp = input(prompt) + if not inp: return default - else: - if return_on_invalid: - return None - return_on_invalid = True - continue + choice = int(inp) + except ValueError: + if return_on_invalid: + return None + return_on_invalid = True + continue else: if min_choice is not None and choice < min_choice: print(f'Number must be greater than {min_choice}')