nemubot/modules/syno.py

84 lines
2.3 KiB
Python

# coding=utf-8
"""Find french synonyms"""
import re
import traceback
import sys
from urllib.parse import quote
from hooks import hook
from tools import web
nemubotversion = 3.4
def help_full():
return "!syno <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(msg.sender, 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(msg.sender, anton, channel=msg.channel,
title="Antonymes de %s" % word)
return res
else:
raise IRCException("Aucun antonyme de %s n'a été trouvé" % word)
else:
raise IRCException("WHAT?!")
def get_synos(word):
url = "http://www.crisco.unicaen.fr/des/synonymes/" + quote(word.encode("ISO-8859-1"))
print_debug (url)
page = web.getURLContent(url)
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:
for elt in re.finditer(">([^<>]+)</a>", line):
anton.append(elt.group(1))
elif line.find("!--Fin liste des synonymes--") > 0:
for elt in re.finditer(">([^<>]+)</a>", line):
synos.append(elt.group(1))
elif re.match("[ \t]*<tr[^>]*>.*</tr>[ \t]*</table>.*", line) is not None:
for elt in re.finditer(">&[^;]+;([^&]*)&[^;]+;<", line):
best.append(elt.group(1))
return (best, synos, anton)
else:
return (list(), list(), list())