forked from bot/app
48 lines
1.0 KiB
Python
48 lines
1.0 KiB
Python
|
# -*- coding: utf-8 -*-
|
||
|
|
"""
|
||
|
|
Copyright (C) 2020-2024 LiteyukiStudio. All Rights Reserved
|
||
|
|
|
||
|
|
@Time : 2024/8/29 下午12:02
|
||
|
|
@Author : snowykami
|
||
|
|
@Email : snowykami@outlook.com
|
||
|
|
@File : translator.py
|
||
|
|
@Software: PyCharm
|
||
|
|
"""
|
||
|
|
from typing import Optional
|
||
|
|
|
||
|
|
from translate import Translator # type: ignore
|
||
|
|
|
||
|
|
# 特殊映射语言
|
||
|
|
i18n_lang2googletrans_lang = {
|
||
|
|
"zh-Hans": "zh-cn",
|
||
|
|
"zh-Hant": "zh-tw",
|
||
|
|
"en" : "en",
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
def get_google_lang(lang: str) -> str:
|
||
|
|
"""
|
||
|
|
Get google translate language
|
||
|
|
Args:
|
||
|
|
lang: language
|
||
|
|
Returns:
|
||
|
|
google translate language
|
||
|
|
"""
|
||
|
|
return i18n_lang2googletrans_lang.get(lang, lang)
|
||
|
|
|
||
|
|
|
||
|
|
def translate(text: str, lang: str, source_lang: str) -> str:
|
||
|
|
"""
|
||
|
|
Translate text to target language
|
||
|
|
Args:
|
||
|
|
source_lang:
|
||
|
|
text: text
|
||
|
|
lang: target language
|
||
|
|
Returns:
|
||
|
|
translated text
|
||
|
|
"""
|
||
|
|
if lang == source_lang:
|
||
|
|
return text
|
||
|
|
google_lang = get_google_lang(lang)
|
||
|
|
return Translator(to_lang=google_lang, from_lang=source_lang).translate(text)
|