1
0
Fork 0
nemubot/modules/translate.py

73 lines
2.6 KiB
Python

# coding=utf-8
import http.client
import re
import socket
import json
from urllib.parse import quote
nemubotversion = 3.0
import module_states_file as xmlparser
LANG = ["ar", "zh", "cz", "en", "fr", "gr", "it", "ja", "ko", "pl", "pt", "ro", "es", "tr"]
def parseanswer(msg):
global LANG
if msg.cmd[0] == "translate":
startWord = 1
if msg.cmd[startWord] in LANG:
langTo = msg.cmd[startWord]
startWord += 1
else:
langTo = "fr"
if msg.cmd[startWord] in LANG:
langFrom = langTo
langTo = msg.cmd[startWord]
startWord += 1
else:
if langTo == "en":
langFrom = "fr"
else:
langFrom = "en"
(res, page) = getPage(' '.join(msg.cmd[startWord:]), langFrom, langTo)
if res == http.client.OK:
wres = json.loads(page.decode())
if "Error" in wres:
msg.send_chn(wres["Note"])
else:
start = "Traduction de %s : "%' '.join(msg.cmd[startWord:])
if "Entries" in wres["term0"]:
if "SecondTranslation" in wres["term0"]["Entries"]["0"]:
msg.send_chn(start + wres["term0"]["Entries"]["0"]["FirstTranslation"]["term"] + " ; " + wres["term0"]["Entries"]["0"]["SecondTranslation"]["term"])
else:
msg.send_chn(start + wres["term0"]["Entries"]["0"]["FirstTranslation"]["term"])
elif "PrincipalTranslations" in wres["term0"]:
if "1" in wres["term0"]["PrincipalTranslations"]:
msg.send_chn(start + wres["term0"]["PrincipalTranslations"]["0"]["FirstTranslation"]["term"] + " ; " + wres["term0"]["PrincipalTranslations"]["1"]["FirstTranslation"]["term"])
else:
msg.send_chn(start + wres["term0"]["PrincipalTranslations"]["0"]["FirstTranslation"]["term"])
else:
msg.send_chn("Une erreur s'est produite durant la recherche d'une traduction de %s" % ' '.join(msg.cmd[startWord:]))
return True
return False
def getPage(terms, langfrom="fr", langto="en"):
conn = http.client.HTTPConnection("api.wordreference.com", timeout=5)
try:
conn.request("GET", "/0.8/%s/json/%s%s/%s" % (CONF.getNode("wrapi")["key"], langfrom, langto, quote(terms)))
except socket.gaierror:
print ("impossible de récupérer la page WordReference.")
return (http.client.INTERNAL_SERVER_ERROR, None)
except (TypeError, KeyError):
print ("You need a WordReference API key in order to use this module. Add it to the module configuration file:\n<wrapi key=\"XXXXX\" />\nRegister at http://www.wordreference.com/docs/APIregistration.aspx")
return (http.client.INTERNAL_SERVER_ERROR, None)
res = conn.getresponse()
data = res.read()
conn.close()
return (res.status, data)