From 896ab385ffce10082c0787c205c77c11be1d618e Mon Sep 17 00:00:00 2001 From: CordlessCoder <42666308+CordlessCoder@users.noreply.github.com> Date: Wed, 22 Jun 2022 17:07:01 +0300 Subject: [PATCH] Removed unnecessary chosen flag --- video_creation/voices.py | 27 +++++++++------------------ 1 file changed, 9 insertions(+), 18 deletions(-) diff --git a/video_creation/voices.py b/video_creation/voices.py index bbc755e..288db4d 100644 --- a/video_creation/voices.py +++ b/video_creation/voices.py @@ -1,4 +1,5 @@ -#!/usr/bin/env python3 +#!/usr/bin/env python + import os from rich.console import Console @@ -31,33 +32,23 @@ def save_text_to_mp3(reddit_obj): """ env = os.getenv("TTSCHOICE", "") if env.casefold in map(lambda _: _.casefold(), TTSProviders): - text_to_mp3 = TTSEngine( - get_case_insensitive_key_value(TTSProviders, env), reddit_obj - ) + text_to_mp3 = TTSEngine(get_case_insensitive_key_value(TTSProviders, env), reddit_obj) else: - chosen = False choice = "" - while not chosen: + while True: print_step("Please choose one of the following TTS providers: ") print_table(TTSProviders) choice = input("\n") - if choice.casefold() not in map(lambda _: _.casefold(), TTSProviders): - print("Unknown Choice") - else: - chosen = True - text_to_mp3 = TTSEngine( - get_case_insensitive_key_value(TTSProviders, choice), reddit_obj - ) + if choice.casefold() in map(lambda _: _.casefold(), TTSProviders): + break + print("Unknown Choice") + text_to_mp3 = TTSEngine(get_case_insensitive_key_value(TTSProviders, choice), reddit_obj) return text_to_mp3.run() def get_case_insensitive_key_value(input_dict, key): return next( - ( - value - for dict_key, value in input_dict.items() - if dict_key.lower() == key.lower() - ), + (value for dict_key, value in input_dict.items() if dict_key.lower() == key.lower()), None, )