Bienvenue à Flo

[bots/salesman.git] / salesman.py

diff --git a/salesman.py b/salesman.py
index 8682b4b350f6bc8f07f64e9462c4a275b4f1f6cc..64dcd002a23926d81e8a9c280c202310109a173c 100755 (executable)
--- a/salesman.py
+++ b/salesman.py
@@ -52,6 +52,17 @@ def log(serveur,channel,auteur=None,message=None):
         print chain
     f.close()
 
+def ignore_event(serv, ev):
+    """Retourne ``True`` si il faut ignorer cet évènement."""
+    for (blackmask, exceptmask) in config.blacklisted_masks:
+        usermask = ev.source()
+        if exceptmask is None:
+            exceptit = False
+        else:
+            exceptit = bool(irclib.mask_matches(usermask, exceptmask))
+        blackit = bool(irclib.mask_matches(usermask, blackmask))
+        if blackit and not exceptit:
+            return True
 
 def reussi(message,answer,auteur):
     if auteur in config.level3:
@@ -167,9 +178,7 @@ class Salesman(ircbot.SingleServerIRCBot):
         mini=min(played.values())
         # on choisit un id dans ceux qui ont ce nombre d'occurences
         id_choisi=random.choice([k for k,v in played.items() if v==mini])
-        capitale,pays=dec[id_choisi]
-        # on peut jouer capitale -> pays ou pays -> capitale
-        enigme,answer=random.choice([[capitale,pays],[pays,capitale]])
+        enigme,answer=dec[id_choisi]
         # on incrémente la choisie
         played[id_choisi]+=1
         # on enregistre le played_file
@@ -187,6 +196,8 @@ class Salesman(ircbot.SingleServerIRCBot):
             return (False,message)
 
     def on_privmsg(self, serv, ev):
+        if ignore_event(serv, ev):
+            return
         message=ev.arguments()[0]
         auteur = irclib.nm_to_n(ev.source())
         try:
@@ -433,6 +444,8 @@ class Salesman(ircbot.SingleServerIRCBot):
             serv.privmsg(auteur,"Je n'ai pas compris. Essaye HELP…")
     
     def on_pubmsg(self, serv, ev):
+        if ignore_event(serv, ev):
+            return
         auteur = irclib.nm_to_n(ev.source())
         canal = ev.target()
         message = ev.arguments()[0]