Compare commits
No commits in common. "9b78a2265303c5c3099a38aeea8d55405abc0f25" and "a3cc18da053c784d1e2deee42736b6bf43d4e7c6" have entirely different histories.
9b78a22653
...
a3cc18da05
75
translate.py
75
translate.py
|
|
@ -1,75 +0,0 @@
|
||||||
# --depends-on commands
|
|
||||||
|
|
||||||
import json
|
|
||||||
import re
|
|
||||||
from src import ModuleManager, utils
|
|
||||||
|
|
||||||
URL_TRANSLATE = "http://translate.googleapis.com/translate_a/single"
|
|
||||||
URL_LANGUAGES = "https://cloud.google.com/translate/docs/languages"
|
|
||||||
REGEX_LANGUAGES = re.compile(r"^([\w-]+)?\:([\w-]+)?\s+", re.I)
|
|
||||||
REGEX_TARGET = re.compile(r"^([\w-]{2,8})\s+", re.I)
|
|
||||||
|
|
||||||
|
|
||||||
class Module(ModuleManager.BaseModule):
|
|
||||||
@utils.hook("received.command.tr", alias_of="translate")
|
|
||||||
@utils.hook("received.command.translate")
|
|
||||||
@utils.spec("!<phrase>lstring")
|
|
||||||
def translate(self, event):
|
|
||||||
"""
|
|
||||||
:help: Translate the given text – supports either **src:dst phrase**
|
|
||||||
(e.g. `fr:en bonjour`) **dst phrase** (e.g. `es hola`) or just
|
|
||||||
`phrase` (auto-detect ➜ English).
|
|
||||||
:usage: [.tr|.translate] [src:][dst] <phrase>
|
|
||||||
"""
|
|
||||||
phrase = event["spec"][0].strip()
|
|
||||||
|
|
||||||
# Defaults preserve existing behaviour
|
|
||||||
source_language = "auto"
|
|
||||||
target_language = "en"
|
|
||||||
|
|
||||||
# Pattern 1 – full "src:dst " prefix
|
|
||||||
match = REGEX_LANGUAGES.match(phrase)
|
|
||||||
if match:
|
|
||||||
if match.group(1):
|
|
||||||
source_language = match.group(1)
|
|
||||||
if match.group(2):
|
|
||||||
target_language = match.group(2)
|
|
||||||
phrase = phrase[len(match.group(0)):]
|
|
||||||
else:
|
|
||||||
# Pattern 2 – single "dst " prefix
|
|
||||||
match = REGEX_TARGET.match(phrase)
|
|
||||||
if match:
|
|
||||||
target_language = match.group(1)
|
|
||||||
phrase = phrase[len(match.group(0)):]
|
|
||||||
|
|
||||||
if not phrase:
|
|
||||||
event["stderr"].write("Nothing to translate.")
|
|
||||||
return
|
|
||||||
|
|
||||||
page = utils.http.request(
|
|
||||||
URL_TRANSLATE,
|
|
||||||
get_params={
|
|
||||||
"client": "gtx",
|
|
||||||
"dt": "t",
|
|
||||||
"q": phrase,
|
|
||||||
"sl": source_language,
|
|
||||||
"tl": target_language,
|
|
||||||
},
|
|
||||||
)
|
|
||||||
|
|
||||||
if page and not page.data.startswith(b"[null,null,"):
|
|
||||||
data = page.decode("utf8")
|
|
||||||
# Google Translate occasionally returns `,,` → patch to valid JSON
|
|
||||||
while ",," in data:
|
|
||||||
data = data.replace(",,", ",null,").replace("[,", "[null,")
|
|
||||||
data_json = json.loads(data)
|
|
||||||
detected_source = data_json[2]
|
|
||||||
translated = data_json[0][0][0]
|
|
||||||
|
|
||||||
event["stdout"].write(
|
|
||||||
f"({detected_source} → {target_language.lower()}) {translated}"
|
|
||||||
)
|
|
||||||
else:
|
|
||||||
event["stderr"].write(
|
|
||||||
"Failed to translate; check language codes: " + URL_LANGUAGES
|
|
||||||
)
|
|
||||||
Loading…
Reference in New Issue