bitbot-3.11-fork/modules/translate.py

52 lines
1.9 KiB
Python
Raw Normal View History

2018-07-02 11:51:36 +00:00
import json, re
from src import ModuleManager, utils
2016-03-29 11:56:58 +00:00
2018-07-02 11:51:36 +00:00
URL_TRANSLATE = "http://translate.googleapis.com/translate_a/single"
URL_LANGUAGES = "https://cloud.google.com/translate/docs/languages"
REGEX_LANGUAGES = re.compile("(\w+)?:(\w+)? ")
2016-03-29 11:56:58 +00:00
class Module(ModuleManager.BaseModule):
@utils.hook("received.command.tr", alias_of="translate")
@utils.hook("received.command.translate")
2016-03-29 11:56:58 +00:00
def translate(self, event):
"""
:help: Translate the provided phrase or the last line in thie current
channel
:usage: [phrase]
"""
2016-03-29 11:56:58 +00:00
phrase = event["args"]
if not phrase:
phrase = event["target"].buffer.get()
2016-03-29 11:56:58 +00:00
if phrase:
phrase = utils.irc.strip_font(phrase.message)
2016-03-29 11:56:58 +00:00
if not phrase:
raise utils.EventError("No phrase provided.")
2016-03-29 11:56:58 +00:00
source_language = "auto"
target_language = "en"
language_match = re.match(REGEX_LANGUAGES, phrase)
if language_match:
if language_match.group(1):
source_language = language_match.group(1)
if language_match.group(2):
target_language = language_match.group(2)
phrase = phrase.split(" ", 1)[1]
data = utils.http.get_url(URL_TRANSLATE, get_params={
2018-07-02 11:51:36 +00:00
"client": "gtx", "sl": source_language,
"tl": target_language, "dt": "t", "q": phrase})
if data and not data == "[null,null,\"\"]":
while ",," in data:
data = data.replace(",,", ",null,")
data = data.replace("[,", "[null,")
data_json = json.loads(data)
detected_source = data_json[2]
event["stdout"].write("(%s -> %s) %s" % (
detected_source, target_language.lower(),
data_json[0][0][0]))
else:
event["stderr"].write("Failed to translate, try checking "
"source/target languages (" + URL_LANGUAGES + ")")
2016-03-29 11:56:58 +00:00