Au revoir jben

[bots/saturnin.git] / saturnin.py

diff --git a/saturnin.py b/saturnin.py
index 87e5bafbfb486f0f2b90a1a78edb35b3f5d93528..75165dfbe8615dfb1f891b93a853226db9a23e05 100755 (executable)
--- a/saturnin.py
+++ b/saturnin.py
@@ -57,6 +57,17 @@ regexp_pan = re.compile(u".*(" + "|".join(config.killwords) + u").*")
 def is_pan(chain):
     return regexp_pan.match(unicode(chain,"utf8").lower())
 
+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))
+        return blackit and not exceptit
+
 class UnicodeBotError(Exception):
     pass
 def bot_unicode(chain):
@@ -116,6 +127,8 @@ class Saturnin(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:
@@ -306,6 +319,11 @@ class Saturnin(ircbot.SingleServerIRCBot):
                         log(self.serveur,"priv",auteur," ".join(message)+"[failed]")
             else:
                 notunderstood=True
+        elif cmd=="status":
+            if auteur in self.ops:
+                serv.privmsg(auteur,"Status : %s" % (self.status, ))
+            else:
+                notunderstood=True
         elif cmd=="spawn":
             if auteur in self.ops:
                 if len(message)>1:
@@ -355,8 +373,8 @@ class Saturnin(ircbot.SingleServerIRCBot):
                 notunderstood=True
         elif cmd=="reload":
             if auteur in self.ops:
-                reload(config)
-                serv.privmsg(auteur,"done")
+                self.reload(auteur)
+                log(self.serveur,"priv",auteur," ".join(message)+"[successful]")
             else:
                 notunderstood=True
         elif cmd=="score":
@@ -444,6 +462,8 @@ class Saturnin(ircbot.SingleServerIRCBot):
         self.serv.privmsg(to, "Scores by pseudo : "+" ; ".join(["%s %s"%(i[0],i[1]) for i in scores]))
     
     def on_pubmsg(self, serv, ev):
+        if ignore_event(serv, ev):
+            return
         auteur = irclib.nm_to_n(ev.source())
         channel = ev.target()
         message = ev.arguments()[0]
@@ -467,6 +487,10 @@ class Saturnin(ircbot.SingleServerIRCBot):
                 else:
                     serv.privmsg(channel,("%s: %s"%(auteur,random.choice(config.quit_fail_messages))).encode("utf8"))
                     log(self.serveur,channel,auteur,message+"[failed]")
+            elif cmd == "reload":
+                if auteur in self.ops:
+                    log(self.serveur, channel, auteur, message+"[successful]")
+                    self.reload(channel)
             elif cmd in ["part","leave","dégage","va-t-en","tut'tiresailleurs,c'estmesgalets"]:
                 if auteur in self.ops and (not (channel in self.stay_channels)
                                            or auteur in self.overops):
@@ -486,6 +510,8 @@ class Saturnin(ircbot.SingleServerIRCBot):
                 self.shot(channel, auteur)
     
     def on_action(self, serv, ev):
+        if ignore_event(serv, ev):
+            return
         action = ev.arguments()[0]
         auteur = irclib.nm_to_n(ev.source())
         channel = ev.target()
@@ -583,6 +609,16 @@ class Saturnin(ircbot.SingleServerIRCBot):
         return self.serv.get_nickname()
     nick=property(_getnick)
     
+    def reload(self, auteur=None):
+        reload(config)
+        if auteur in [None, "SIGHUP"]:
+            towrite = "Config reloaded" + " (SIGHUP received)"*(auteur == "SIGHUP")
+            for to in config.report_bugs_to:
+                self.serv.privmsg(to, towrite)
+            log(self.serveur, towrite)
+        else:
+            self.serv.privmsg(auteur,"Config reloaded")
+    
     def start_as_daemon(self, outfile):
         sys.stderr = Logger(outfile)
         self.start()
@@ -627,13 +663,18 @@ if __name__=="__main__":
     if "--quiet" in sys.argv:
         config.debug_stdout=False
     serveurs={"a♡":"acoeur.crans.org","acoeur":"acoeur.crans.org","acoeur.crans.org":"acoeur.crans.org",
-              "irc":"irc.crans.org","crans":"irc.crans.org","irc.crans.org":"irc.crans.org"}
+              "irc":"irc.crans.org","crans":"irc.crans.org","irc.crans.org":"irc.crans.org",
+              "local":"localhost"}
     try:
         serveur=serveurs[serveur]
     except KeyError:
         print "Server Unknown : %s"%(serveur)
         exit(404)
     saturnin = Saturnin(serveur,debug)
+    # Si on reçoit un SIGHUP, on reload la config
+    def sighup_handler(signum, frame):
+        saturnin.reload("SIGHUP")
+    signal.signal(signal.SIGHUP, sighup_handler)
     if daemon:
         child_pid = os.fork()
         if child_pid == 0: