Code reformatted with black

pull/1271/head
John Toniutti 3 years ago
parent 21e71d1228
commit e746efe5fe
No known key found for this signature in database
GPG Key ID: 3C5434E3920EE1E1

@ -10,7 +10,7 @@ from utils import settings
__all__ = ["TikTok", "TikTokTTSException"] __all__ = ["TikTok", "TikTokTTSException"]
disney_voices: Final = ( disney_voices: Final[tuple] = (
"en_us_ghostface", # Ghost Face "en_us_ghostface", # Ghost Face
"en_us_chewbacca", # Chewbacca "en_us_chewbacca", # Chewbacca
"en_us_c3po", # C3PO "en_us_c3po", # C3PO
@ -22,7 +22,7 @@ disney_voices: Final = (
"en_male_pirate", # pirate "en_male_pirate", # pirate
) )
eng_voices: Final = ( eng_voices: Final[tuple] = (
"en_au_001", # English AU - Female "en_au_001", # English AU - Female
"en_au_002", # English AU - Male "en_au_002", # English AU - Male
"en_uk_001", # English UK - Male 1 "en_uk_001", # English UK - Male 1
@ -39,7 +39,7 @@ eng_voices: Final = (
"en_male_cody", # Serious "en_male_cody", # Serious
) )
non_eng_voices: Final = ( non_eng_voices: Final[tuple] = (
# Western European voices # Western European voices
"fr_001", # French - Male 1 "fr_001", # French - Male 1
"fr_002", # French - Male 2 "fr_002", # French - Male 2
@ -64,7 +64,7 @@ non_eng_voices: Final = (
"kr_004", # Korean - Male 2 "kr_004", # Korean - Male 2
) )
vocals = ( vocals: Final[tuple] = (
"en_female_f08_salut_damour", # Alto "en_female_f08_salut_damour", # Alto
"en_male_m03_lobby", # Tenor "en_male_m03_lobby", # Tenor
"en_male_m03_sunshine_soon", # Sunshine Soon "en_male_m03_sunshine_soon", # Sunshine Soon
@ -78,13 +78,16 @@ vocals = (
class TikTok: class TikTok:
"""TikTok Text-to-Speech Wrapper""" """TikTok Text-to-Speech Wrapper"""
BASE_URL: Final = "https://api16-normal-c-useast1a.tiktokv.com/media/api/text/speech/invoke/"
BASE_URL: Final[
str
] = "https://api16-normal-c-useast1a.tiktokv.com/media/api/text/speech/invoke/"
def __init__(self): def __init__(self):
headers = { headers = {
"User-Agent": "com.zhiliaoapp.musically/2022600030 (Linux; U; Android 7.1.2; es_ES; SM-G988N; " "User-Agent": "com.zhiliaoapp.musically/2022600030 (Linux; U; Android 7.1.2; es_ES; SM-G988N; "
"Build/NRD90M;tt-ok/3.12.13.1)", "Build/NRD90M;tt-ok/3.12.13.1)",
"Cookie": f"sessionid={settings.config['settings']['tts']['tiktok_sessionid']}" "Cookie": f"sessionid={settings.config['settings']['tts']['tiktok_sessionid']}",
} }
self._session = requests.Session() self._session = requests.Session()
@ -120,11 +123,7 @@ class TikTok:
text = text.replace("+", "plus").replace("&", "and").replace("r/", "") text = text.replace("+", "plus").replace("&", "and").replace("r/", "")
# prepare url request # prepare url request
params = { params = {"req_text": text, "speaker_map_type": 0, "aid": 1233}
"req_text": text,
"speaker_map_type": 0,
"aid": 1233
}
if voice is not None: if voice is not None:
params["text_speaker"] = voice params["text_speaker"] = voice
@ -135,7 +134,7 @@ class TikTok:
except ConnectionError: except ConnectionError:
time.sleep(random.randrange(1, 7)) time.sleep(random.randrange(1, 7))
response = self._session.post(self.BASE_URL, params=params) response = self._session.post(self.BASE_URL, params=params)
return response.json() return response.json()
@staticmethod @staticmethod
@ -148,12 +147,14 @@ class TikTokTTSException(Exception):
self._code = code self._code = code
self._message = message self._message = message
def __str__(self): def __str__(self) -> str:
if self._code == 1: if self._code == 1:
return f"Code: {self._message}, reason: probably the aid value isn't correct, message: {self._message}" return f"Code: {self._message}, reason: probably the aid value isn't correct, message: {self._message}"
elif self._code == 2:
if self._code == 2:
return f"Code: {self._message}, reason: the text is too long, message: {self._message}" return f"Code: {self._message}, reason: the text is too long, message: {self._message}"
elif self._code == 4:
if self._code == 4:
return f"Code: {self._message}, reason: the speaker doesn't exist, message: {self._message}" return f"Code: {self._message}, reason: the speaker doesn't exist, message: {self._message}"
else:
return f"Code: {self._message}, reason: unknown, message: {self._message}" return f"Code: {self._message}, reason: unknown, message: {self._message}"

Loading…
Cancel
Save