95 lines
2.9 KiB
Python
95 lines
2.9 KiB
Python
"""Find french conjugaison"""
|
|
|
|
# PYTHON STUFFS #######################################################
|
|
|
|
from collections import defaultdict
|
|
import re
|
|
from urllib.parse import quote
|
|
|
|
from nemubot.exception import IMException
|
|
from nemubot.hooks import hook
|
|
from nemubot.tools import web
|
|
from nemubot.tools.web import striphtml
|
|
|
|
from more import Response
|
|
|
|
|
|
# GLOBALS #############################################################
|
|
|
|
s = [('present', '0'), ('présent', '0'), ('pr', '0'),
|
|
('passé simple', '12'), ('passe simple', '12'), ('ps', '12'),
|
|
('passé antérieur', '112'), ('passe anterieur', '112'), ('pa', '112'),
|
|
('passé composé', '100'), ('passe compose', '100'), ('pc', '100'),
|
|
('futur', '18'), ('f', '18'),
|
|
('futur antérieur', '118'), ('futur anterieur', '118'), ('fa', '118'),
|
|
('subjonctif présent', '24'), ('subjonctif present', '24'), ('spr', '24'),
|
|
('subjonctif passé', '124'), ('subjonctif passe', '124'), ('spa', '124'),
|
|
('plus que parfait', '106'), ('pqp', '106'),
|
|
('imparfait', '6'), ('ii', '6')]
|
|
|
|
d = defaultdict(list)
|
|
|
|
for k, v in s:
|
|
d[k].append(v)
|
|
|
|
|
|
# MODULE CORE #########################################################
|
|
|
|
def get_conjug(verb, stringTens):
|
|
url = ("http://leconjugueur.lefigaro.fr/conjugaison/verbe/%s.html" %
|
|
quote(verb.encode("ISO-8859-1")))
|
|
page = web.getURLContent(url)
|
|
|
|
if page is not None:
|
|
for line in page.split("\n"):
|
|
if re.search('<div class="modeBloc">', line) is not None:
|
|
return compute_line(line, stringTens)
|
|
return list()
|
|
|
|
|
|
def compute_line(line, stringTens):
|
|
try:
|
|
idTemps = d[stringTens]
|
|
except:
|
|
raise IMException("le temps demandé n'existe pas")
|
|
|
|
if len(idTemps) == 0:
|
|
raise IMException("le temps demandé n'existe pas")
|
|
|
|
index = line.index('<div id="temps' + idTemps[0] + '\"')
|
|
endIndex = line[index:].index('<div class=\"conjugBloc\"')
|
|
|
|
endIndex += index
|
|
newLine = line[index:endIndex]
|
|
|
|
res = list()
|
|
for elt in re.finditer("[p|/]>([^/]*/b>)", newLine):
|
|
res.append(striphtml(elt.group(1)
|
|
.replace("<b>", "\x02")
|
|
.replace("</b>", "\x0F")))
|
|
return res
|
|
|
|
|
|
# MODULE INTERFACE ####################################################
|
|
|
|
@hook.command("conjugaison",
|
|
help_usage={
|
|
"TENS VERB": "give the conjugaison for VERB in TENS."
|
|
})
|
|
def cmd_conjug(msg):
|
|
if len(msg.args) < 2:
|
|
raise IMException("donne moi un temps et un verbe, et je te donnerai "
|
|
"sa conjugaison!")
|
|
|
|
tens = ' '.join(msg.args[:-1])
|
|
|
|
verb = msg.args[-1]
|
|
|
|
conjug = get_conjug(verb, tens)
|
|
|
|
if len(conjug) > 0:
|
|
return Response(conjug, channel=msg.channel,
|
|
title="Conjugaison de %s" % verb)
|
|
else:
|
|
raise IMException("aucune conjugaison de '%s' n'a été trouvé" % verb)
|