diff --git a/modules/conjugaison.py b/modules/conjugaison.py index 6b5d31e..9c724f2 100644 --- a/modules/conjugaison.py +++ b/modules/conjugaison.py @@ -7,9 +7,37 @@ 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" @@ -63,36 +91,24 @@ def get_conjug(verb, stringTens): def compute_line(line, stringTens): res = list() - idTemps = get_conjug_for_tens(stringTens) + try: + idTemps = d[stringTens] + except: + res.append("Le temps demandé n'existe pas") + return res - if idTemps is None: - return Response(msg.sender, - "Le temps que vous avez spécifiez n'existe pas", msg.channel) + if len(idTemps) == 0: + res.append("Le temps demandé n'existe pas") + return res - index = line.index('
([^/]*/b>)", newLine): -# res.append(strip_tags(elt.group(1))) res.append(striphtml(elt.group(1))) return res - - -def get_conjug_for_tens(stringTens): - dic = {'pr' : '0', - 'ps' : '12', - 'pa' : '112', - 'pc' : '100', - 'f' : '18', - 'fa' : '118', - 'spr' : '24', - 'spa' : '124', - 'ii' : '6', - 'pqp' : '106'} - - return dic[stringTens]