eff99aff05
Uses the new google v2 translate service which is more accurate. Also reduces code ugliness by quite a bit.
50 lines
1.9 KiB
Python
50 lines
1.9 KiB
Python
#--depends-on commands
|
|
#--require-config google-api-key
|
|
|
|
import json, re
|
|
from src import ModuleManager, utils
|
|
|
|
URL_TRANSLATE = "https://translation.googleapis.com/language/translate/v2"
|
|
URL_LANGUAGES = "https://cloud.google.com/translate/docs/languages"
|
|
REGEX_LANGUAGES = re.compile("(\w+)?:(\w+)? ")
|
|
|
|
|
|
class Module(ModuleManager.BaseModule):
|
|
|
|
@utils.hook("received.command.tr", alias_of="translate")
|
|
@utils.hook("received.command.translate")
|
|
@utils.spec("?<from:to>lstring !<phrase>lstring")
|
|
def translate(self, event):
|
|
"""
|
|
:help: Translate the provided phrase or the last line in thie current
|
|
channel
|
|
:usage: [phrase]
|
|
"""
|
|
phrase = event["spec"][0]
|
|
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]
|
|
|
|
page = utils.http.request(URL_TRANSLATE,
|
|
method="POST",
|
|
post_data={
|
|
"q": phrase,
|
|
"format": "text",
|
|
"source": source_language,
|
|
"target": target_language,
|
|
"key": self.bot.config["google-api-key"]
|
|
}).json()
|
|
|
|
if "data" in page:
|
|
translation = page["data"]["translations"][0]["translatedText"]
|
|
event["stdout"].write("(%s -> %s) %s" % (source_language, target_language, translation))
|
|
else:
|
|
event["stderr"].write("Failed to translate, try checking "
|
|
"source/target languages (" + URL_LANGUAGES + ")")
|