]> gitweb.pimeys.fr Git - bots/historien.git/blobdiff - historien.py
C'est cool d'être daemonizable
[bots/historien.git] / historien.py
index 57e2e0cae175a06540c42eb965bf136b55831ff7..8836af7caac02158ac554c6089eb366188f2e174 100755 (executable)
@@ -5,15 +5,20 @@
 
 # Un bot IRC qui pose des questions d'histoire
 
-import irclib
-import ircbot
 import threading
 import random
 import time
 import pickle
 import re
+import signal
+import sys
 from cast_as_date import *
 
+# Oui, j'ai recodé ma version d'irclib pour pouvoir rattrapper les SIGHUP
+sys.path.insert(0, "/home/vincent/scripts/python-myirclib")
+import irclib
+import ircbot
+
 config_password="EtTaMère,ElleEstNéeQuand?"
 config_pseudo="historien"
 config_chanlist=["#bot","#flood"]
@@ -59,7 +64,7 @@ config_leave_messages=[u"%s : %s quitte le channel"]
 # temps au bout duquel, si personne n'a parlé, on se tait
 config_idle_time=20*60
 # liste des bots, qui ne sont pas considérés comme de l'activité
-config_idle_bots=["deconnaisseur","Basile","historien","hung","salesman","Shadobot","Wen","___"]
+config_idle_bots=["deconnaisseur","Basile","historien","hung","salesman","Shadobot","Wen","___","Sanctuary","Saturnin"]
 
 class UnicodeBotError(Exception):
     pass
@@ -91,6 +96,9 @@ config_noscore=["[20-100]","[20-100]_"] # parce que 20-100 est nul en histoire
 class GoodCentury(Exception):
     pass
 
+class GoodDeceny(Exception):
+    pass
+
 def reussi(message,answer,auteur):
     if auteur in config_level3:
         return answer in message
@@ -110,6 +118,8 @@ def reussi(message,answer,auteur):
                 score+=config_score_mois
                 if date[2]==realdate[2]:
                     score+=config_score_jour
+        elif date[0]/10 == realdate[0]/10:
+            raise GoodDeceny
         elif date[0]/100 == realdate[0]/100:
             raise GoodCentury
         return score
@@ -188,7 +198,7 @@ class Historien(ircbot.SingleServerIRCBot):
                 log(self.serveur,channel,u"$Date$".encode("utf8"),("%s : %s"%(date, evenement)).encode("utf8"))
                 serv.privmsg(channel,evenement.encode("utf8"))
                 token=time.time()
-                # le 0 est le flag "bon siècle" n' pas encore été dit
+                # le 0 est le flag "bon siècle" n'a pas encore été dit
                 self.play_status[channel]=[1,date,evenement,0,token]
                 serv.execute_delayed(random.randrange(ttrig*3,ttrig*5),self.give_indice,(serv,channel,token))
             elif ok=="refuse":
@@ -584,12 +594,19 @@ class Historien(ircbot.SingleServerIRCBot):
                         serv.privmsg(canal,"%s: C'est le bon siècle, mais pas la bonne année, cherche encore ;)"%(auteur))
                         self.play_status[canal][3]=1
                     return
+                except GoodDeceny:
+                    if flag_century in [0,1]:
+                        serv.privmsg(canal,"%s: C'est la bonne décennie, mais pas la bonne année, encore un effort ;)"%(auteur))
+                        self.play_status[canal][3]=2
+                    return
                 if score_obtenu:
                     if self.play_status[canal][0]==1:
-                        bonusmsg=u" [+bonus_mois"*(score_obtenu>config_score_annee)+u"+bonus_jour"*(score_obtenu>config_score_annee+config_score_mois)+u"]"
+                        bonusmsg=u" [+bonus_mois"*(score_obtenu>config_score_annee)+u"+bonus_jour"*(score_obtenu>config_score_annee+config_score_mois)
                     else:
                         bonusmsg=""
                         score_obtenu=1
+                    if bonusmsg:
+                        bonusmsg+=u"]"
                     serv.privmsg(canal,(u"%s: bravo ! (C'était le %s)%s"%(auteur,answer,bonusmsg)).encode("utf8"))
                     log(self.serveur,canal,auteur+"$win",message)
                     if auteur in config_noscore:
@@ -608,7 +625,7 @@ class Historien(ircbot.SingleServerIRCBot):
             log(self.serveur,"%s kické de %s par %s (raison : %s)" %(victime,channel,auteur,raison))
             time.sleep(5)
             serv.join(channel)
-            self.update_activity(message[1],"")
+            self.update_activity(channel,"")
             # on ne dit rien au rejoin
             #l1,l2=config_kick_answers,config_kick_actions
             #n1,n2=len(l1),len(l2)
@@ -663,17 +680,48 @@ class Historien(ircbot.SingleServerIRCBot):
             self.last_activity[canal]=time.time()
     def is_active(self,canal):
         return time.time()-self.last_activity[canal]<config_idle_time
+    
+    def start_as_daemon(self, outfile):
+        sys.stderr = Logger(outfile)
+        self.start()
+
+
+class Logger(object):
+    """Pour écrire ailleurs que sur stdout"""
+    def __init__(self, filename="historien.full.log"):
+        self.filename = filename
+
+    def write(self, message):
+        f = open(self.filename, "a")
+        f.write(message)
+        f.close()
+
 
 if __name__=="__main__":
     import sys
     if len(sys.argv)==1:
-        print "Usage : historien.py <serveur> [--debug]"
+        print "Usage : historien.py <serveur> [--debug] [--no-output] [--daemon [--pidfile]] [--outfile]"
+        print "        --outfile sans --no-output ni --daemon n'a aucun effet"
         exit(1)
     serveur=sys.argv[1]
+    if "--daemon" in sys.argv:
+        thisfile = os.path.realpath(__file__)
+        thisdirectory = thisfile.rsplit("/", 1)[0]
+        os.chdir(thisdirectory)
+        daemon = True
+    else:
+        daemon = False
     if "debug" in sys.argv or "--debug" in sys.argv:
         debug=True
     else:
         debug=False
+    if "--no-output" in sys.argv or "--daemon" in sys.argv:
+        outfile = "/var/log/bots/historien.full.log"
+        for arg in sys.argv:
+            arg = arg.split("=")
+            if arg[0].strip('-') in ["out", "outfile", "logfile"]:
+                outfile = arg[1]
+        sys.stdout = Logger(outfile)
     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"}
     try:
@@ -682,4 +730,21 @@ if __name__=="__main__":
         print "Server Unknown : %s"%(serveur)
         exit(404)
     historien=Historien(serveur,debug)
-    historien.start()
+    if daemon:
+        child_pid = os.fork()
+        if child_pid == 0:
+            os.setsid()
+            historien.start_as_daemon(outfile)
+        else:
+            # on enregistre le pid de historien
+            pidfile = "/var/run/bots/historien.pid"
+            for arg in sys.argv:
+                arg = arg.split("=")
+                if arg[0].strip('-') in ["pidfile"]:
+                    pidfile = arg[1]
+            f = open(pidfile, "w")
+            f.write("%s\n" % child_pid)
+            f.close()
+    else:
+        historien.start()
+