]> gitweb.pimeys.fr Git - bots/basile.git/blobdiff - basile.py
Modularité : sortie des is_
[bots/basile.git] / basile.py
index b467ac85b33c72387723d45b1f3daa2fb813017b..68c2bb35f3b47824de9806b570f8b813afc148f2 100755 (executable)
--- a/basile.py
+++ b/basile.py
@@ -25,6 +25,8 @@ from commands import getstatusoutput as ex
 import config
 #: Module responsable du dialogue avec la NoteKfet2015
 import nk
+#: Module de réponse aux questions de base
+import isit
 
 # la partie qui réfère au fichier lui-même est mieux ici
 # sinon on réfère la config et pas le fichier lui-même
@@ -52,122 +54,21 @@ def log(serveur, channel, auteur=None, message=None):
     f.write((chain + u"\n").encode("utf-8"))
     f.close()
     if config.debug_stdout:
-        print chain
+        print chain.encode("utf-8")
 
-def regex_join(liste, avant=u".*(?:^| )", apres=u"(?:$|\.| |,|;).*"):
-    """Fabrique une regexp à partir d'une liste d'éléments à matcher."""
-    return avant + u"(" + u"|".join(liste) + u")" + apres
-
-def is_something(chain, regexp=None, matches=[], avant=u".*(?:^| )", apres=u"(?:$|\.| |,|;).*",
-                 case_sensitive=False):
-    """Vérifie si chain contient un des éléments de ``matches``.
-       Si ``regexp`` est fournie, c'est simplement elle qui est testée"""
-    if case_sensitive:
-        chain = chain.lower()
-    if regexp == None:
-        regexp = regex_join(matches, avant, apres)
-        regexp = re.compile(regexp)
-    o = regexp.match(chain)
-    return o
+def ignore_event(serv, ev):
+    """Retourne ``True`` si il faut ignorer cet évènement."""
+    for (blackmask, exceptlist) in config.blacklisted_masks:
+        usermask = ev.source()
+        blackit = bool(irclib.mask_matches(usermask, blackmask))
+        exceptit = any([bool(irclib.mask_matches(usermask, exceptmask)) for exceptmask in exceptlist])
+        if exceptit: # Il est exempté
+            return False
+        else:
+            if blackit: # Il n'est pas exempté et matche la blacklist
+                return True
 
-def regexp_compile():
-    """Compilation des regexp à partir de la conf.
-       Place les résultats dans le namespace de ``config``"""
-    config.insult_regexp = regex_join(config.insultes, avant=u".*(?:^| |')")
-    config.insult_regexp_compiled = re.compile(config.insult_regexp)
-    
-    config.not_insult_regexp = u".*pas %s%s" % (config.amplifier_regexp, config.insult_regexp)
-    config.not_insult_regexp_compiled = re.compile(config.not_insult_regexp)
-    
-    config.compliment_regexp = regex_join(config.compliment_triggers, avant=u".*(?:^| |')")
-    config.compliment_regexp_compiled = re.compile(config.compliment_regexp)
-    
-    config.perdu_regexp = regex_join(config.perdu)
-    config.perdu_regexp_compiled = re.compile(config.perdu_regexp)
-    
-    config.tag_regexp = regex_join(config.tag_triggers)
-    config.tag_regexp_compiled = re.compile(config.tag_regexp)
-    
-    config.gros_regexp = regex_join(config.gros)
-    config.gros_regexp_compiled = re.compile(config.gros_regexp)
-    
-    config.tesla_regexp = regex_join(config.tesla_triggers, avant=u"^", apres="$")
-    config.tesla_regexp_compiled = re.compile(config.tesla_regexp)
-    
-    config.merci_regexp = regex_join(config.merci_triggers)
-    config.merci_regexp_compiled = re.compile(config.merci_regexp)
-    
-    config.tamere_regexp = regex_join(config.tamere_triggers)
-    config.tamere_regexp_compiled = re.compile(config.tamere_regexp)
-    
-    config.bonjour_regexp = regex_join(config.bonjour_triggers, avant=u"^")
-    config.bonjour_regexp_compiled = re.compile(config.bonjour_regexp)
-    
-    config.bonne_nuit_regexp = regex_join(config.bonne_nuit_triggers, avant=u"^")
-    config.bonne_nuit_regexp_compiled = re.compile(config.bonne_nuit_regexp)
-    
-    config.pan_regexp = regex_join(config.pan_triggers, avant=".*", apres=".*")
-    config.pan_regexp_compiled = re.compile(config.pan_regexp)
-    
-    
-regexp_compile()
-def is_insult(chain, debug=True):
-    """Vérifie si ``chain`` contient une insulte."""
-    return is_something(chain, config.insult_regexp_compiled)
-def is_not_insult(chain):
-    """Vérifie si ``chain`` contient une insulte à la forme négative."""
-    return is_something(chain, config.not_insult_regexp_compiled)
-def is_compliment(chain, debug=True):
-    """Vérifie si ``chain`` contient un compliment."""
-    return is_something(chain, config.compliment_regexp_compiled)
-def is_perdu(chain):
-    """Vérifie si ``chain`` contient une raison de perdre."""
-    return is_something(chain, config.perdu_regexp_compiled)
-def is_tag(chain):
-    """Vérifie si ``chain`` demande de fermer sa gueule."""
-    return is_something(chain, config.tag_regexp_compiled)
-def is_gros(chain):
-    """Vérifie si ``chain`` traite de gros."""
-    return is_something(chain, config.gros_regexp_compiled)
-def is_tesla(chain):
-    """Vérifie si ``chain`` est un ping."""
-    return is_something(chain, config.tesla_regexp_compiled)
-def is_merci(chain):
-    """Vérifie si ``chain`` contient un remerciement."""
-    return is_something(chain, config.merci_regexp_compiled)
-def is_tamere(chain):
-    """Vérifie si ``chain`` traite ma mère."""
-    return is_something(chain, config.tamere_regexp_compiled)
-def is_bad_action_trigger(chain,pseudo):
-    """Vérifie si ``chain`` est une action méchante.
-       On a besoin d'une regexp dynamique à cause du pseudo qui peut changer."""
-    return is_something(chain, matches=config.bad_action_triggers, avant=u"^",
-                            apres="(?: [a-z]*ment)? %s($|\.| |,|;).*" % (pseudo))
-def is_good_action_trigger(chain,pseudo):
-    """Vérifie si ``chain`` est une action gentille.
-       On a besoin d'une regexp dynamique à cause du pseudo qui peut changer."""
-    return is_something(chain, matches=config.good_action_triggers, avant=u"^",
-                            apres="(?: [a-z]*ment)? %s($|\.| |,|;).*" % (pseudo))
-def is_bonjour(chain):
-    """Vérifie si ``chain`` contient un bonjour."""
-    return is_something(chain, config.bonjour_regexp_compiled)
-def is_bonne_nuit(chain):
-    """Vérifie si ``chain`` contient un bonne nuit."""
-    return is_something(chain, config.bonne_nuit_regexp_compiled)
-def is_pan(chain):
-    """Vérifie si ``chain`` contient un pan."""
-    return is_something(chain, config.pan_regexp_compiled)
 
-def is_time(conf):
-    """Vérifie si l'heure actuelle est entre les deux heures ``conf[0]`` et ``conf[1]``"""
-    _, _, _, h, m, s, _, _, _ = time.localtime()
-    return (conf[0], 0, 0) < (h, m, s) < (conf[1], 0, 0)
-def is_day():
-    """Vérifie si on est le jour."""
-    return is_time(config.daytime)
-def is_night():
-    """Vérifie si on est la nuit."""
-    return is_time(config.nighttime)
 
       
 class UnicodeBotError(Exception):
@@ -286,7 +187,7 @@ class Basile(ircbot.SingleServerIRCBot):
     def execute_reload(self, auteur=None):
         """Recharge la config."""
         reload(config)
-        regexp_compile()
+        isit.regexp_compile()
         if auteur in [None, "SIGHUP"]:
             towrite = "Config reloaded" + " (SIGHUP received)" * (auteur == "SIGHUP")
             for to in config.report_bugs_to:
@@ -299,7 +200,7 @@ class Basile(ircbot.SingleServerIRCBot):
     def crash(self, chan="nowhere", who="nobody"):
         """Fait crasher le bot."""
         where = "en privé" if chan == "priv" else "sur le chan %s" % chan
-        raise CrashError("Crash demandé par %s %s" % (who, where))
+        raise CrashError((u"Crash demandé par %s %s" % (who, where)).encode("utf-8"))
     
     ACTIONS = {
         "reload" : execute_reload,
@@ -336,6 +237,8 @@ class Basile(ircbot.SingleServerIRCBot):
 
     def on_privmsg(self, serv, ev):
         """À la réception d'un message en privé."""
+        if ignore_event(serv, ev):
+            return
         message = ev.arguments()[0]
         auteur = irclib.nm_to_n(ev.source())
         try:
@@ -425,7 +328,7 @@ class Basile(ircbot.SingleServerIRCBot):
             if auteur in self.ops and len(message) > 1:
                 if message[1] in self.chanlist:
                     if not (message[1] in self.stay_channels) or auteur in self.overops:
-                        self.quitter(message[1], " ".join(message[2:]))
+                        self.quitter(message[1].encode("utf-8"), " ".join(message[2:]))
                         self.chanlist.remove(message[1])
                         log(self.serveur, "priv", auteur, " ".join(message) + "[successful]")
                     else:
@@ -524,7 +427,7 @@ class Basile(ircbot.SingleServerIRCBot):
                 notunderstood = True
         elif cmd == u"say":
             if auteur in self.overops and len(message) > 2:
-                serv.privmsg(message[1], " ".join(message[2:]))
+                serv.privmsg(message[1].encode("utf-8"), (u" ".join(message[2:])).encode("utf-8"))
                 log(self.serveur, "priv", auteur, " ".join(message))
             elif len(message) <= 2:
                 serv.privmsg(auteur, "Syntaxe : SAY <channel> <message>")
@@ -540,7 +443,7 @@ class Basile(ircbot.SingleServerIRCBot):
                 notunderstood = True
         elif cmd == u"kick":
             if auteur in self.overops and len(message) > 2:
-                serv.kick(message[1], message[2], " ".join(message[3:]))
+                serv.kick(message[1].encode("utf-8"), message[2].encode("utf-8"), " ".join(message[3:]).encode("utf-8"))
                 log(self.serveur, "priv", auteur, " ".join(message))
             elif len(message) <= 2:
                 serv.privmsg(auteur, "Syntaxe : KICK <channel> <pseudo> [<raison>]")
@@ -559,7 +462,7 @@ class Basile(ircbot.SingleServerIRCBot):
                 if self.identities.has_key(auteur):
                     success, solde, pseudo = nk.get_solde(self.nk, self.identities[auteur]["idbde"], serv, auteur)
                     if success:
-                        serv.privmsg(auteur, "%s (%s)" % (float(solde)/100, pseudo.encode("utf8")))
+                        serv.privmsg(auteur, "%.2f (%s)" % (solde/100.0, pseudo.encode("utf8")))
                     log(self.serveur, "priv", auteur, " ".join(message) + ("[successful]" if success else "[failed]"))
                 else:
                     serv.privmsg(canal, "Je ne connais pas ton pseudo note.")
@@ -580,6 +483,8 @@ class Basile(ircbot.SingleServerIRCBot):
     
     def on_pubmsg(self, serv, ev):
         """À la réception d'un message sur un channel."""
+        if ignore_event(serv, ev):
+            return
         auteur = irclib.nm_to_n(ev.source())
         canal = ev.target()
         message = ev.arguments()[0]
@@ -651,31 +556,31 @@ class Basile(ircbot.SingleServerIRCBot):
             elif cmd in [u"ping"] and not canal in self.quiet_channels:
                 serv.privmsg(canal, "%s: pong" % (auteur))
 
-            elif cmd in [u"solde", u"!solde", u"!coca"]:
+            elif cmd in [u"solde", u"!solde", u"!coca"] or cmd.startswith("!"):
                 if self.identities.has_key(auteur):
                     idbde = self.identities[auteur]["idbde"]
                     if cmd in [u"solde", u"!solde"]:
                         success, solde, pseudo = nk.get_solde(self.nk, self.identities[auteur]["idbde"], serv, canal)
                         if success:
                             serv.privmsg(canal, "%s: %s (%s)" % (auteur, float(solde)/100, pseudo.encode("utf8")))
-                    elif cmd in [u"!coca"]:
-                        success = nk.consomme(self.nk, self.identities[auteur]["idbde"], u"Coca", serv, canal)
+                    elif cmd in [u"!coca"] or cmd.startswith("!"):
+                        success = nk.consomme(self.nk, self.identities[auteur]["idbde"], message[1:], serv, canal)
                     log(self.serveur, canal, auteur, message + ("[successful]" if success else "[failed]"))
                 else:
                     serv.privmsg(canal, "%s: Je ne connais pas votre pseudo note." % (auteur))
                     log(self.serveur, canal, auteur, message + "[unknown]")
-            elif (re.match("!?(pain au chocolat|chocolatine)", message.lower())
+            elif (re.match("(pain au chocolat|chocolatine)", message.lower())
                  and not canal in self.quiet_channels):
                 serv.action(canal, "sert un pain au chocolat à %s" % (auteur))
-            elif re.match("!?manzana",message.lower()) and not canal in self.quiet_channels:
+            elif re.match("manzana",message.lower()) and not canal in self.quiet_channels:
                 if auteur in config.manzana:
                     serv.action(canal, "sert une bouteille de manzana à %s" % (auteur))
                 elif auteur in config.manzana_bis:
                     serv.action(canal, "sert un grand verre de jus de pomme à %s : tout le monde sait qu'il ne boit pas." % (auteur))
                 else:
                     serv.action(canal, "sert un verre de manzana à %s" % (auteur))
-            if is_insult(message) and not canal in self.quiet_channels:
-                if is_not_insult(message):
+            if isit.is_insult(message) and not canal in self.quiet_channels:
+                if isit.is_not_insult(message):
                     answer = random.choice(config.compliment_answers)
                     for ligne in answer.split("\n"):
                         serv.privmsg(canal, "%s: %s" % (auteur, ligne.encode("utf8")))
@@ -683,16 +588,16 @@ class Basile(ircbot.SingleServerIRCBot):
                     answer = random.choice(config.insultes_answers)
                     for ligne in answer.split("\n"):
                         serv.privmsg(canal, "%s: %s" % (auteur, ligne.encode("utf8")))
-            elif is_compliment(message) and not canal in self.quiet_channels:
+            elif isit.is_compliment(message) and not canal in self.quiet_channels:
                 answer = random.choice(config.compliment_answers)
                 for ligne in answer.split("\n"):
                     serv.privmsg(canal, "%s: %s" % (auteur,ligne.encode("utf8")))
-            gros_match = is_gros(message)
+            gros_match = isit.is_gros(message)
             if gros_match and not canal in self.quiet_channels:
                 taille = get_filesize()
                 answer = u"Mais non, je ne suis pas %s, %sKo tout au plus…" % (gros_match.groups()[0], taille)
                 serv.privmsg(canal, "%s: %s"%(auteur, answer.encode("utf8")))
-            if is_tesla(message) and not canal in self.quiet_channels:
+            if isit.is_tesla(message) and not canal in self.quiet_channels:
                 l1, l2 = config.tesla_answers, config.tesla_actions
                 n1, n2 = len(l1), len(l2)
                 i = random.randrange(n1 + n2)
@@ -700,11 +605,11 @@ class Basile(ircbot.SingleServerIRCBot):
                     serv.action(canal, l2[i - n1].encode("utf8"))
                 else:
                     serv.privmsg(canal, "%s: %s" % (auteur, l1[i].encode("utf8")))
-            if is_tamere(message) and not canal in self.quiet_channels:
+            if isit.is_tamere(message) and not canal in self.quiet_channels:
                 answer = random.choice(config.tamere_answers)
                 for ligne in answer.split("\n"):
                     serv.privmsg(canal, "%s: %s"%(auteur, ligne.encode("utf8")))
-            if is_tag(message) and not canal in self.quiet_channels:
+            if isit.is_tag(message) and not canal in self.quiet_channels:
                 if auteur in self.ops:
                     action = random.choice(config.tag_actions)
                     serv.action(canal, action.encode("utf8"))
@@ -713,7 +618,7 @@ class Basile(ircbot.SingleServerIRCBot):
                     answer = random.choice(config.tag_answers)
                     for ligne in answer.split("\n"):
                         serv.privmsg(canal, "%s: %s" % (auteur, ligne.encode("utf8")))
-            if is_merci(message):
+            if isit.is_merci(message):
                 answer = random.choice(config.merci_answers)
                 for ligne in answer.split("\n"):
                     serv.privmsg(canal, "%s: %s"%(auteur, ligne.encode("utf8")))
@@ -747,18 +652,18 @@ class Basile(ircbot.SingleServerIRCBot):
                                        for j in mess])
                     out = int(translate(out))
                     serv.privmsg(canal,"%s: %s !" % (auteur, translate(str(out + 1)).encode("utf8")))
-            if is_bonjour(message) and not canal in self.quiet_channels:
-                if is_night():
+            if isit.is_bonjour(message) and not canal in self.quiet_channels:
+                if isit.is_night():
                     answer = random.choice(config.night_answers)
-                elif is_day():
+                elif isit.is_day():
                     answer = random.choice(config.bonjour_answers)
                 else:
                     answer = random.choice(config.bonsoir_answers)
                 serv.privmsg(canal, answer.format(auteur).encode("utf8"))
-            if is_bonne_nuit(message) and not canal in self.quiet_channels:
+            if isit.is_bonne_nuit(message) and not canal in self.quiet_channels:
                 answer = random.choice(config.bonne_nuit_answers)
                 serv.privmsg(canal, answer.format(auteur).encode("utf8"))
-            if is_pan(message) and not canal in self.quiet_channels:
+            if isit.is_pan(message) and not canal in self.quiet_channels:
                 serv.privmsg(canal, "%s: ce n'est pas sur moi qu'il faut tirer, même si je sais que j'attire l'œil !" % (auteur))
         else:
             if message in [u"!pain au chocolat", u"!chocolatine"] and not canal in self.quiet_channels:
@@ -781,7 +686,7 @@ class Basile(ircbot.SingleServerIRCBot):
                              ).format(mypseudo).lower(), message.strip().lower()):
                     answer = random.choice(config.bonjour_answers)
                     serv.privmsg(canal, answer.format(auteur).encode("utf8"))
-        if (is_perdu(message) and not canal in self.quiet_channels):
+        if (isit.is_perdu(message) and not canal in self.quiet_channels):
             # proba de perdre sur trigger :
             #  avant 30min (enfin, config) : 0
             #  ensuite, +25%/30min, linéairement
@@ -793,6 +698,8 @@ class Basile(ircbot.SingleServerIRCBot):
 
     def on_action(self, serv, ev):
         """À la réception d'une action."""
+        if ignore_event(serv, ev):
+            return
         action = ev.arguments()[0]
         auteur = irclib.nm_to_n(ev.source())
         channel = ev.target()
@@ -804,7 +711,7 @@ class Basile(ircbot.SingleServerIRCBot):
             return
         mypseudo = self.nick
         
-        if is_bad_action_trigger(action, mypseudo) and not channel in self.quiet_channels:
+        if isit.is_bad_action_trigger(action, mypseudo) and not channel in self.quiet_channels:
             l1, l2 = config.bad_action_answers, config.bad_action_actions
             n1, n2 = len(l1), len(l2)
             i = random.randrange(n1 + n2)
@@ -812,7 +719,7 @@ class Basile(ircbot.SingleServerIRCBot):
                 serv.action(channel, l2[i - n1].format(auteur).encode("utf8"))
             else:
                 serv.privmsg(channel, l1[i].format(auteur).encode("utf8"))
-        if is_good_action_trigger(action, mypseudo) and not channel in self.quiet_channels:
+        if isit.is_good_action_trigger(action, mypseudo) and not channel in self.quiet_channels:
             l1, l2 = config.good_action_answers, config.good_action_actions
             n1, n2 = len(l1), len(l2)
             i = random.randrange(n1 + n2)
@@ -828,7 +735,7 @@ class Basile(ircbot.SingleServerIRCBot):
         victime = ev.arguments()[0]
         raison = ev.arguments()[1]
         if victime == self.nick:
-            log(self.serveur, "%s kické de %s par %s (raison : %s)" % (victime, channel, auteur, raison))
+            log(self.serveur, u"%s kické de %s par %s (raison : %s)" % (victime, channel.decode("utf-8"), auteur, raison))
             time.sleep(2)
             serv.join(channel)
             l1, l2 = config.kick_answers, config.kick_actions