115 lines
3.2 KiB
Python
115 lines
3.2 KiB
Python
# coding=utf-8
|
|
|
|
import re
|
|
import traceback
|
|
import sys
|
|
from urllib.parse import quote
|
|
|
|
from tools import web
|
|
from tools.web import striphtml
|
|
from collections import defaultdict
|
|
|
|
nemubotversion = 3.3
|
|
|
|
dic = {'pr' : '0',
|
|
'ps' : '12',
|
|
'pa' : '112',
|
|
'pc' : '100',
|
|
'f' : '18',
|
|
'fa' : '118',
|
|
'spr' : '24',
|
|
'spa' : '124',
|
|
'ii' : '6',
|
|
'pqp' : '106'}
|
|
|
|
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)
|
|
|
|
def help_tiny ():
|
|
return "Find french conjugaison"
|
|
|
|
def help_full ():
|
|
return "!conjugaison <tens> <verb>: give the conjugaison for <verb> in <tens>."
|
|
|
|
def load(context):
|
|
from hooks import Hook
|
|
add_hook("cmd_hook", Hook(cmd_conjug, "conjugaison"))
|
|
|
|
|
|
def cmd_conjug(msg):
|
|
if len(msg.cmds) < 3:
|
|
return Response(msg.sender,
|
|
"Demande incorrecte.\n %s" % help_full(),
|
|
msg.channel)
|
|
tens = msg.cmds[1]
|
|
verb = msg.cmds[2]
|
|
try:
|
|
conjug = get_conjug(verb, tens)
|
|
except:
|
|
conjug = None
|
|
exc_type, exc_value, exc_traceback = sys.exc_info()
|
|
traceback.print_exception(exc_type, exc_value,
|
|
exc_traceback)
|
|
|
|
if conjug is None:
|
|
return Response(msg.sender,
|
|
"Une erreur s'est produite durant la recherche"
|
|
" du verbe %s" % verb, msg.channel)
|
|
elif len(conjug) > 0:
|
|
return Response(msg.sender, conjug, msg.channel,
|
|
title="Conjugaison de %s" % verb)
|
|
else:
|
|
return Response(msg.sender,
|
|
"Aucune conjugaison de %s n'a été trouvé" % verb,
|
|
msg.channel)
|
|
return False
|
|
|
|
|
|
def get_conjug(verb, stringTens):
|
|
url = "http://leconjugueur.lefigaro.fr/conjugaison/verbe/" + quote(verb.encode("ISO-8859-1")) + ".html"
|
|
print_debug (url)
|
|
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)
|
|
else:
|
|
return None
|
|
|
|
def compute_line(line, stringTens):
|
|
res = list()
|
|
try:
|
|
idTemps = d[stringTens]
|
|
except:
|
|
res.append("Le temps demandé n'existe pas")
|
|
return res
|
|
|
|
if len(idTemps) == 0:
|
|
res.append("Le temps demandé n'existe pas")
|
|
return res
|
|
|
|
index = line.index('<div id="temps' + idTemps[0] + '\"')
|
|
endIndex = line[index:].index('<div class=\"conjugBloc\"')
|
|
|
|
endIndex += index
|
|
newLine = line[index:endIndex]
|
|
|
|
for elt in re.finditer("[p|/]>([^/]*/b>)", newLine):
|
|
res.append(striphtml(elt.group(1)))
|
|
|
|
return res
|
|
|