def get_user_choice(query): """ Asks the user a boolean question """ print("\n") query = color(query, Fore.GREEN + Style.BRIGHT) + " " while True: user_choice = input(query).strip().lower() if user_choice in ["y", "yes", "1"]: return True if user_choice in ["n", "no", "0"]: return False print(color("Invalid input: '" + user_choice + "'", Fore.RED + Style.BRIGHT))