]> gitweb.pimeys.fr Git - bots/hung.git/blobdiff - hung.py
C'est cool d'être daemonizable
[bots/hung.git] / hung.py
diff --git a/hung.py b/hung.py
index b849f7f9ada7a7c69247d42a72cbd9a1036be303..49371f1caa81b6534fc54cb8a0d0431ee2c7b902 100755 (executable)
--- a/hung.py
+++ b/hung.py
@@ -5,8 +5,6 @@
 
 # Un test de bot irc, parce que c'est cool
 
-import irclib
-import ircbot
 import threading
 import random
 import time
@@ -14,22 +12,29 @@ import socket, ssl, json
 import pickle
 import re
 import os
+import signal
+import sys
 from commands import getstatusoutput as ex
 
+# 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
+
 import sys
 config_debug_stdout=True
 if "--quiet" in sys.argv:
     config_debug_stdout=False
 
 config_irc_password="I'mAHungMan"
-config_irc_pseudo="Hung"
+config_irc_pseudo="hung"
 config_chanlist=["#bot","#flood"]
 config_stay_channels=["#bot","#flood"]
 config_play_channels=["#flood"]
 config_quiet_channels=[]
 config_logfile_template="hung.%s.log"
 def get_config_logfile(serveur):
-    serveurs={"acoeur.crans.org":"acoeur","irc.crans.org":"crans"}
+    serveurs={"acoeur.crans.org":"acoeur","irc.crans.org":"crans","localhost":"localhost"}
     return config_logfile_template%(serveurs[serveur])
 config_overops=["[20-100]","[20-100]_","Petite-Peste","PEB"]
 config_ops=[]
@@ -60,6 +65,7 @@ u"B _ E, C R U E L   _ O R L D!",
 u"B Y E, C R U E L   _ O R L D!",
 u"B Y E, C R U E L   W O R L D!",
 ]
+config_leave_messages=["On finira la partie plus tard :p"]
 
 def log(serveur,channel,auteur=None,message=None):
     f=open(get_config_logfile(serveur),"a")
@@ -102,6 +108,12 @@ def is_something(chain,matches,avant=u".*(?:^| )",apres=u"(?:$|\.| |,|;).*",case
 def is_tag(chain):
     return is_something(chain,config_tag_triggers)
 
+def is_mot(mot,liste):
+    real_word = "".join([lettre[0] for lettre in liste])
+    real_word = real_word.decode("utf8").lower()
+    mot=remplace_accents(mot.decode("utf8"))
+    return mot==real_word
+
 class Hung(ircbot.SingleServerIRCBot):
     def __init__(self,serveur,debug=False):
         temporary_pseudo=config_irc_pseudo+str(random.randrange(10000,100000))
@@ -115,7 +127,7 @@ class Hung(ircbot.SingleServerIRCBot):
         self.chanlist=config_chanlist
         self.stay_channels=config_stay_channels
         self.play_channels=config_play_channels
-        self.play_status={}
+        self.play_status={i:[None,None,None] for i in self.play_channels}
         self.quiet_channels=config_quiet_channels
 
 
@@ -126,6 +138,7 @@ class Hung(ircbot.SingleServerIRCBot):
         serv.nick(config_irc_pseudo)
     
     def on_welcome(self, serv, ev):
+        self.serv=serv # ça serv ira :)
         self.give_me_my_pseudo(serv)
         serv.privmsg("NickServ","IDENTIFY %s"%(config_irc_password))
         log(self.serveur,"Connected")
@@ -175,7 +188,7 @@ class Hung(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:
-                        serv.part(message[1])
+                        self.quitter(message[1]," ".join(message[2:]))
                         self.chanlist.remove(message[1])
                         log(self.serveur,"priv",auteur," ".join(message)+"[successful]")
                     else:
@@ -250,8 +263,7 @@ class Hung(ircbot.SingleServerIRCBot):
         elif cmd=="die":
             if auteur in self.overops:
                 log(self.serveur,"priv",auteur," ".join(message)+"[successful]")
-                quit_message=random.choice(config_quit_messages)
-                self.die(message=config_quit_message)
+                self.mourir()
             else:
                 notunderstood=True
         elif cmd=="quiet":
@@ -340,14 +352,14 @@ class Hung(ircbot.SingleServerIRCBot):
             if cmd in ["meurs","die","crève"]:
                 if auteur in self.overops:
                     log(self.serveur,canal,auteur,message+"[successful]")
-                    self.die()
+                    self.mourir()
                 else:
                     serv.privmsg(canal,"%s: crève !"%(auteur))
                     log(self.serveur,canal,auteur,message+"[failed]")
             elif cmd in ["part","leave","dégage"]:
                 if auteur in self.ops and (not (canal in self.stay_channels)
                                            or auteur in self.overops):
-                    serv.part(canal,message="Éjecté par %s"%(auteur))
+                    self.quitter(canal)
                     log(self.serveur,canal,auteur,message+"[successful]")
                     if canal in self.chanlist:
                         self.chanlist.remove(canal)
@@ -398,21 +410,16 @@ class Hung(ircbot.SingleServerIRCBot):
                         serv.privmsg(canal,"%s: %s"%(auteur,ligne.encode("utf8")))
             # on essaye de voir si le mot fourni matche la partie en cours
             mot = cmd
-            # bon, ce teste merde et j'arrive pas à trouver pourquoi, alors j'ai craqué, je l'ai mis dans un try
-            # bouh ! beurk ! pas bien ! promis, j'irai me flageller…
-            try:
-                if remplace_accents(mot)==("".join([i[0] for i in self.play_status[canal][0]])).lower():
-                    # on a trouvé le mot
-                    # on regarde combien de lettre il manquait
-                    manquait = sum([not lettre[1] for lettre in self.play_status[canal][0]])
-                    self.add_score({auteur: manquait})
-                    if self.play_status[canal][2].has_key(auteur):
-                        self.play_status[canal][2][auteur]+=manquait
-                    else:
-                        self.play_status[canal][2][auteur]=manquait
-                    self.gagne(serv, canal, bonus=auteur, bonusvalue=manquait)
-            except:
-                pass
+            if self.play_status[canal][0]!=None and is_mot(mot, self.play_status[canal][0]):
+                # on a trouvé le mot
+                # on regarde combien de lettre il manquait
+                manquait = sum([not lettre[1] for lettre in self.play_status[canal][0]])
+                self.add_score({auteur: manquait})
+                if self.play_status[canal][2].has_key(auteur):
+                    self.play_status[canal][2][auteur]+=manquait
+                else:
+                    self.play_status[canal][2][auteur]=manquait
+                self.gagne(serv, canal, bonus=auteur, bonusvalue=manquait)
         else:
             pass
 
@@ -428,7 +435,38 @@ class Hung(ircbot.SingleServerIRCBot):
               "%s : Euh, tu fais de la merde avec ton encodage là, j'ai failli crasher…"%(auteur))
             return
         mypseudo=serv.get_nickname()
-        
+
+    def on_kick(self,serv,ev):
+        auteur = irclib.nm_to_n(ev.source())
+        channel = ev.target()
+        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))
+            time.sleep(5)
+            serv.join(channel)
+            # on ne dit rien au rejoin
+            #l1,l2=config_kick_answers,config_kick_actions
+            #n1,n2=len(l1),len(l2)
+            #i=random.randrange(n1+n2)
+            #if i>=n1:
+            #    serv.action(channel,l2[i-n1].format(auteur).encode("utf8"))
+            #else:
+            #    serv.privmsg(channel,l1[i].format(auteur).encode("utf8"))
+    
+    def _getnick(self):
+        return self.serv.get_nickname()
+    nick = property(_getnick)
+    
+    def quitter(self,chan,leave_message=None):
+        if leave_message==None:
+            leave_message=random.choice(config_leave_messages)
+        self.serv.part(chan,message=leave_message.encode("utf8"))
+    
+    def mourir(self):
+        quit_message=random.choice(config_quit_messages)
+        self.die(msg=quit_message.encode("utf8"))
+    
     def get_scores(self):
         f=open(config_scores_file)
         scores=pickle.load(f)
@@ -467,23 +505,71 @@ class Hung(ircbot.SingleServerIRCBot):
         serv.privmsg(canal,"Contributions : %s"%("  ".join(contribs)) )
         self.play_status[canal]=[None,None,None]
 
+    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="hung.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 : hung.py <serveur> [--debug]"
+        print "Usage : hung.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/hung.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"}
+              "irc":"irc.crans.org","crans":"irc.crans.org","irc.crans.org":"irc.crans.org",
+              "localhost":"localhost"}
     try:
         serveur=serveurs[serveur]
     except KeyError:
         print "Server Unknown : %s"%(serveur)
         exit(404)
     hung=Hung(serveur,debug)
-    hung.start()
+    if daemon:
+        child_pid = os.fork()
+        if child_pid == 0:
+            os.setsid()
+            hung.start_as_daemon(outfile)
+        else:
+            # on enregistre le pid de hung
+            pidfile = "/var/run/bots/hung.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:
+        hung.start()
+