diff --git a/STT.py b/STT.py index f83cbd9..8524402 100644 --- a/STT.py +++ b/STT.py @@ -19,7 +19,7 @@ def read(self, file_name, language=__NOT_DEFINED): if not file_name: raise ValueError('Expected file name') - if not (".flac" in file_name): + if ".flac" not in file_name: raise ValueError('Expected .flac file') if language == self.__NOT_DEFINED: diff --git a/TTS.py b/TTS.py index ac3d700..7668561 100644 --- a/TTS.py +++ b/TTS.py @@ -25,10 +25,14 @@ def __try_detect_language(self, target_language): return self.__DEFAULT_VOICE all_voices = self.list_voices() - for voice in all_voices['voices']: - if target_language in voice['language']: - return voice['name'] - return self.__DEFAULT_VOICE + return next( + ( + voice['name'] + for voice in all_voices['voices'] + if target_language in voice['language'] + ), + self.__DEFAULT_VOICE, + ) def speech(self, text, file_name=__NOT_DEFINED, voice=__NOT_DEFINED, language=__NOT_DEFINED, play_after=True): diff --git a/Translate.py b/Translate.py index dc651c6..a89f625 100644 --- a/Translate.py +++ b/Translate.py @@ -21,10 +21,14 @@ def list_languages(self): def search_language_code(self, language_name): all_languages = self.list_languages() - for language in all_languages['languages']: - if language['language_name'].lower() == language_name.lower(): - return language['language'] - return None + return next( + ( + language['language'] + for language in all_languages['languages'] + if language['language_name'].lower() == language_name.lower() + ), + None, + ) def translate(self, text, source, target): self.__validate()