Syno module: add english language support; closing #64

This commit is contained in:
nemunaire 2014-10-04 07:33:34 +02:00
parent ce9dec7ed4
commit 5d5030efe1

View File

@ -1,11 +1,11 @@
# coding=utf-8
"""Find french synonyms"""
"""Find synonyms"""
import json
import re
import traceback
import sys
from urllib.parse import quote
from urllib.request import urlopen
from hooks import hook
from tools import web
@ -15,56 +15,28 @@ nemubotversion = 3.4
from more import Response
def help_full():
return "!syno <word>: give a list of synonyms for <word>."
return "!syno [LANG] <word>: give a list of synonyms for <word>."
@hook("cmd_hook", "synonymes")
def cmd_syno(msg):
return go("synonymes", msg)
@hook("cmd_hook", "antonymes")
def cmd_anto(msg):
return go("antonymes", msg)
def go(what, msg):
if len(msg.cmds) < 2:
raise IRCException("de quel mot veux-tu connaître la liste des synonymes ?")
word = ' '.join(msg.cmds[1:])
try:
best, synos, anton = get_synos(word)
except:
best, synos, anton = (list(), list(), list())
if what == "synonymes":
if len(synos) > 0:
res = Response(best, channel=msg.channel,
title="Synonymes de %s" % word)
res.append_message(synos)
return res
else:
raise IRCException("Aucun synonyme de %s n'a été trouvé" % word)
elif what == "antonymes":
if len(anton) > 0:
res = Response(anton, channel=msg.channel,
title="Antonymes de %s" % word)
return res
else:
raise IRCException("Aucun antonyme de %s n'a été trouvé" % word)
def load(context):
global lang_binding
if not CONF or not CONF.hasNode("bighugelabs") or not CONF.getNode("bighugelabs").hasAttribute("key"):
print ("You need a NigHugeLabs API key in order to have english "
"theasorus. Add it to the module configuration file:\n<bighugelabs"
" key=\"XXXXXXXXXXXXXXXX\" />\nRegister at "
"https://words.bighugelabs.com/getkey.php")
else:
raise IRCException("WHAT?!")
lang_binding["en"] = lambda word: get_english_synos(CONF.getNode("bighugelabs")["key"], word)
def get_synos(word):
def get_french_synos(word):
url = "http://www.crisco.unicaen.fr/des/synonymes/" + quote(word.encode("ISO-8859-1"))
print_debug (url)
print_debug(url)
page = web.getURLContent(url)
best = list(); synos = list(); anton = list()
if page is not None:
best = list()
synos = list()
anton = list()
for line in page.split("\n"):
if line.find("!-- Fin liste des antonymes --") > 0:
@ -79,7 +51,64 @@ def get_synos(word):
for elt in re.finditer(">&[^;]+;([^&]*)&[^;]+;<", line):
best.append(elt.group(1))
return (best, synos, anton)
return (best, synos, anton)
def get_english_synos(key, word):
raw = urlopen("http://words.bighugelabs.com/api/2/%s/%s/json" % (quote(key), quote(word.encode("ISO-8859-1"))))
cnt = json.loads(raw.read().decode())
best = list(); synos = list(); anton = list()
if cnt is not None:
for k, c in cnt.items():
if "syn" in c: best += c["syn"]
if "rel" in c: synos += c["rel"]
if "ant" in c: anton += c["ant"]
return (best, synos, anton)
lang_binding = { 'fr': get_french_synos }
@hook("cmd_hook", "synonymes", data="synonymes")
@hook("cmd_hook", "antonymes", data="antonymes")
def go(msg, what):
if len(msg.cmds) < 2:
raise IRCException("de quel mot veux-tu connaître la liste des synonymes ?")
# Detect lang
if msg.cmds[1] in lang_binding:
func = lang_binding[msg.cmds[1]]
word = ' '.join(msg.cmds[2:])
else:
func = lang_binding["fr"]
word = ' '.join(msg.cmds[1:])
# TODO: depreciate usage without lang
#raise IRCException("language %s is not handled yet." % msg.cmds[1])
try:
best, synos, anton = func(word)
except:
best, synos, anton = (list(), list(), list())
if what == "synonymes":
if len(synos) > 0 or len(best) > 0:
res = Response(channel=msg.channel, title="Synonymes de %s" % word)
if len(best) > 0: res.append_message(best)
if len(synos) > 0: res.append_message(synos)
return res
else:
raise IRCException("Aucun synonyme de %s n'a été trouvé" % word)
elif what == "antonymes":
if len(anton) > 0:
res = Response(anton, channel=msg.channel,
title="Antonymes de %s" % word)
return res
else:
raise IRCException("Aucun antonyme de %s n'a été trouvé" % word)
else:
return (list(), list(), list())
raise IRCException("WHAT?!")