]> gitweb.pimeys.fr Git - bots/hung.git/commitdiff
fichier de conf + reloadable y compris par SIGHUP + utf8_trigger
authorVincent Le Gallic <legallic@crans.org>
Mon, 30 Jul 2012 17:17:21 +0000 (19:17 +0200)
committerVincent Le Gallic <legallic@crans.org>
Mon, 30 Jul 2012 17:17:21 +0000 (19:17 +0200)
config.py [new file with mode: 0644]
hung.py

diff --git a/config.py b/config.py
new file mode 100644 (file)
index 0000000..3d364b9
--- /dev/null
+++ b/config.py
@@ -0,0 +1,47 @@
+#!/usr/bin/python
+# -*- coding:utf8 -*-
+
+debug_stdout=True
+
+irc_password="I'mAHungMan"
+irc_pseudo="hung"
+chanlist=["#bot","#flood"]
+stay_channels=["#bot","#flood"]
+play_channels=["#flood"]
+quiet_channels=[]
+logfile_template="hung.%s.log"
+
+overops=["[20-100]","[20-100]_","Petite-Peste","PEB"]
+ops=[]
+report_bugs_to=["[20-100]"]
+
+# config UTF8-fail
+utf8_fail_answers = [u"Euh, tu fais de la merde avec ton encodage là, j'ai failli crasher…"]
+utf8_trigger = True # râlé-je en cas de non-utf8 ?
+
+dico_mots="mots.txt"
+dico_defs="definitions.txt"
+
+scores_file="scores.pickle"
+
+tag_triggers=[u"t(|a)g",u"ta gueule",u"la ferme",u"ferme( |-)la",u"tais-toi",u"chut"]
+tag_actions=[u"se tait",u"ferme sa gueule",u"se la ferme",u"la ferme"]
+tag_answers=[u"J'me tais si j'veux !",
+u"Je t'entends pas :°",
+u"Héhé, try again",
+u"Non, j'ai pas envie",
+u"Peut-être quand toi tu la fermeras, et encore…"]
+
+quit_messages=[u"_ _ _, _ _ _ _ _   _ _ _ _ _!",
+u"_ _ E, _ _ _ E _   _ _ _ _ _!",
+u"_ _ E, _ _ _ E _   _ O _ _ _!",
+u"_ _ E, _ _ U E _   _ O _ _ _!",
+u"_ _ E, _ R U E _   _ O R _ _!",
+u"_ _ E, _ R U E L   _ O R L _!",
+u"B _ E, _ R U E L   _ O R L _!",
+u"B _ E, C R U E L   _ O R L _!",
+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!",
+]
+leave_messages=["On finira la partie plus tard :p"]
diff --git a/hung.py b/hung.py
index 49371f1caa81b6534fc54cb8a0d0431ee2c7b902..96c8a58a6e4b3dd020f36f5dfdbefb6a18778f45 100755 (executable)
--- a/hung.py
+++ b/hung.py
@@ -3,7 +3,7 @@
 
 # Codé par 20-100 le 23/04/12
 
 
 # Codé par 20-100 le 23/04/12
 
-# Un test de bot irc, parce que c'est cool
+# Un bot IRC qui joue au pendu
 
 import threading
 import random
 
 import threading
 import random
@@ -22,50 +22,13 @@ import irclib
 import ircbot
 
 import sys
 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_chanlist=["#bot","#flood"]
-config_stay_channels=["#bot","#flood"]
-config_play_channels=["#flood"]
-config_quiet_channels=[]
-config_logfile_template="hung.%s.log"
+
+# Fichier de conf
+import config
+
 def get_config_logfile(serveur):
     serveurs={"acoeur.crans.org":"acoeur","irc.crans.org":"crans","localhost":"localhost"}
 def get_config_logfile(serveur):
     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=[]
-config_report_bugs_to=["[20-100]"]
-
-config_dico_mots="mots.txt"
-config_dico_defs="definitions.txt"
-
-config_scores_file="scores.pickle"
-
-config_tag_triggers=[u"t(|a)g",u"ta gueule",u"la ferme",u"ferme( |-)la",u"tais-toi",u"chut"]
-config_tag_actions=[u"se tait",u"ferme sa gueule",u"se la ferme",u"la ferme"]
-config_tag_answers=[u"J'me tais si j'veux !",
-u"Je t'entends pas :°",
-u"Héhé, try again",
-u"Non, j'ai pas envie",
-u"Peut-être quand toi tu la fermeras, et encore…"]
-
-config_quit_messages=[u"_ _ _, _ _ _ _ _   _ _ _ _ _!",
-u"_ _ E, _ _ _ E _   _ _ _ _ _!",
-u"_ _ E, _ _ _ E _   _ O _ _ _!",
-u"_ _ E, _ _ U E _   _ O _ _ _!",
-u"_ _ E, _ R U E _   _ O R _ _!",
-u"_ _ E, _ R U E L   _ O R L _!",
-u"B _ E, _ R U E L   _ O R L _!",
-u"B _ E, C R U E L   _ O R L _!",
-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"]
+    return config.logfile_template%(serveurs[serveur])
 
 def log(serveur,channel,auteur=None,message=None):
     f=open(get_config_logfile(serveur),"a")
 
 def log(serveur,channel,auteur=None,message=None):
     f=open(get_config_logfile(serveur),"a")
@@ -75,7 +38,7 @@ def log(serveur,channel,auteur=None,message=None):
     else:
         chain="%s [%s:%s] %s"%(time.strftime("%F %T"),channel,auteur,message)
     f.write(chain+"\n")
     else:
         chain="%s [%s:%s] %s"%(time.strftime("%F %T"),channel,auteur,message)
     f.write(chain+"\n")
-    if config_debug_stdout:
+    if config.debug_stdout:
         print chain
     f.close()
 
         print chain
     f.close()
 
@@ -106,7 +69,7 @@ def is_something(chain,matches,avant=u".*(?:^| )",apres=u"(?:$|\.| |,|;).*",case
     return o
 
 def is_tag(chain):
     return o
 
 def is_tag(chain):
-    return is_something(chain,config_tag_triggers)
+    return is_something(chain,config.tag_triggers)
 
 def is_mot(mot,liste):
     real_word = "".join([lettre[0] for lettre in liste])
 
 def is_mot(mot,liste):
     real_word = "".join([lettre[0] for lettre in liste])
@@ -116,31 +79,31 @@ def is_mot(mot,liste):
 
 class Hung(ircbot.SingleServerIRCBot):
     def __init__(self,serveur,debug=False):
 
 class Hung(ircbot.SingleServerIRCBot):
     def __init__(self,serveur,debug=False):
-        temporary_pseudo=config_irc_pseudo+str(random.randrange(10000,100000))
+        temporary_pseudo=config.irc_pseudo+str(random.randrange(10000,100000))
         ircbot.SingleServerIRCBot.__init__(self, [(serveur, 6667)],
                               temporary_pseudo,"Bot irc pour jouer au pendu", 10)
         self.debug=debug
         self.serveur=serveur
         ircbot.SingleServerIRCBot.__init__(self, [(serveur, 6667)],
                               temporary_pseudo,"Bot irc pour jouer au pendu", 10)
         self.debug=debug
         self.serveur=serveur
-        self.overops=config_overops
-        self.ops=self.overops+config_ops
-        self.report_bugs_to=config_report_bugs_to
-        self.chanlist=config_chanlist
-        self.stay_channels=config_stay_channels
-        self.play_channels=config_play_channels
+        self.overops=config.overops
+        self.ops=self.overops+config.ops
+        self.report_bugs_to=config.report_bugs_to
+        self.chanlist=config.chanlist
+        self.stay_channels=config.stay_channels
+        self.play_channels=config.play_channels
         self.play_status={i:[None,None,None] for i in self.play_channels}
         self.play_status={i:[None,None,None] for i in self.play_channels}
-        self.quiet_channels=config_quiet_channels
+        self.quiet_channels=config.quiet_channels
 
 
     def give_me_my_pseudo(self,serv):
 
 
     def give_me_my_pseudo(self,serv):
-        serv.privmsg("NickServ","RECOVER %s %s"%(config_irc_pseudo,config_irc_password))
-        serv.privmsg("NickServ","RELEASE %s %s"%(config_irc_pseudo,config_irc_password))
+        serv.privmsg("NickServ","RECOVER %s %s"%(config.irc_pseudo,config.irc_password))
+        serv.privmsg("NickServ","RELEASE %s %s"%(config.irc_pseudo,config.irc_password))
         time.sleep(0.3)
         time.sleep(0.3)
-        serv.nick(config_irc_pseudo)
+        serv.nick(config.irc_pseudo)
     
     def on_welcome(self, serv, ev):
         self.serv=serv # ça serv ira :)
         self.give_me_my_pseudo(serv)
     
     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))
+        serv.privmsg("NickServ","IDENTIFY %s"%(config.irc_password))
         log(self.serveur,"Connected")
         if self.debug:
             self.chanlist=["#bot"]
         log(self.serveur,"Connected")
         if self.debug:
             self.chanlist=["#bot"]
@@ -164,13 +127,40 @@ class Hung(ircbot.SingleServerIRCBot):
         try:
             test=bot_unicode(message)
         except UnicodeBotError:
         try:
             test=bot_unicode(message)
         except UnicodeBotError:
-            serv.privmsg(auteur,
-              "Euh, tu fais de la merde avec ton encodage là, j'ai failli crasher…")
+            if config.utf8_trigger:
+                serv.privmsg(auteur, config.utf8_fail_answers)
             return
         message=message.split()
         cmd=message[0].lower()
         notunderstood=False
             return
         message=message.split()
         cmd=message[0].lower()
         notunderstood=False
-        if cmd=="join":
+        if cmd=="help":
+            helpmsg_default="""Liste des commandes :
+ HELP       Affiche ce message d'aide
+ SCORE      Affiche ton score
+ SCORES     Affiche les scores"""
+            helpmsg_ops="""
+ JOIN       Faire rejoindre un channel (sans paramètres, donne la liste des chans actuels)
+ LEAVE      Faire quitter un channel
+ PLAY       Passe un channel en mode "jouer"
+ NOPLAY     Passe un channel en mode "ne pas jouer"
+ QUIET      Se taire sur un channel
+ NOQUIET    Opposé de QUIET
+ RELOAD     Recharge la config"""
+            helpmsg_overops="""
+ SAY        Fais envoyer un message sur un chan ou à une personne
+ DO         Me fait faire une action sur un chan
+ STAY       Ignorera les prochains LEAVE pour un chan
+ NOSTAY     Opposé de STAY
+ STATUS     Montre l'état courant
+ DIE        Mourir"""
+            helpmsg=helpmsg_default
+            if auteur in self.ops:
+                helpmsg+=helpmsg_ops
+            if auteur in self.overops:
+                helpmsg+=helpmsg_overops
+            for ligne in helpmsg.split("\n"):
+                serv.privmsg(auteur,ligne)
+        elif cmd=="join":
             if auteur in self.ops:
                 if len(message)>1:
                     if message[1] in self.chanlist:
             if auteur in self.ops:
                 if len(message)>1:
                     if message[1] in self.chanlist:
@@ -292,6 +282,12 @@ class Hung(ircbot.SingleServerIRCBot):
                         log(self.serveur,"priv",auteur," ".join(message)+"[failed]")
             else:
                 notunderstood=True
                         log(self.serveur,"priv",auteur," ".join(message)+"[failed]")
             else:
                 notunderstood=True
+        elif cmd=="reload":
+            if auteur in self.ops:
+                self.reload(auteur)
+                log(self.serveur,"priv",auteur," ".join(message)+"[successful]")
+            else:
+                notunderstood=True
         elif cmd=="say":
             if auteur in self.overops and len(message)>2:
                 serv.privmsg(message[1]," ".join(message[2:]))
         elif cmd=="say":
             if auteur in self.overops and len(message)>2:
                 serv.privmsg(message[1]," ".join(message[2:]))
@@ -322,8 +318,8 @@ class Hung(ircbot.SingleServerIRCBot):
             serv.privmsg(canal,"%s : %s"%(begin,obfuskated))
 
     def start_partie(self, serv, canal):
             serv.privmsg(canal,"%s : %s"%(begin,obfuskated))
 
     def start_partie(self, serv, canal):
-        mots=[mot.strip() for mot in open(config_dico_mots).readlines()]
-        defs=[defi.strip() for defi in open(config_dico_defs).readlines()]
+        mots=[mot.strip() for mot in open(config.dico_mots).readlines()]
+        defs=[defi.strip() for defi in open(config.dico_defs).readlines()]
         indice = random.randrange(0,len(mots))
         mot,definition=mots[indice],defs[indice]
         # ' et - sont considérés comme déjà devinés
         indice = random.randrange(0,len(mots))
         mot,definition=mots[indice],defs[indice]
         # ' et - sont considérés comme déjà devinés
@@ -338,9 +334,8 @@ class Hung(ircbot.SingleServerIRCBot):
         try:
             test=bot_unicode(message)
         except UnicodeBotError:
         try:
             test=bot_unicode(message)
         except UnicodeBotError:
-            if not canal in self.quiet_channels:
-                serv.privmsg(canal,
-                  "%s: Euh, tu fais de la merde avec ton encodage là, j'ai failli crasher…"%(auteur))
+            if not canal in self.quiet_channels and config.utf8_trigger:
+                serv.privmsg(canal, "%s: %s"%(auteur,config.utf8_fail_answers))
             return
         pour_moi,message=self.pourmoi(serv,message)
         if pour_moi and message.split()!=[]:
             return
         pour_moi,message=self.pourmoi(serv,message)
         if pour_moi and message.split()!=[]:
@@ -356,6 +351,10 @@ class Hung(ircbot.SingleServerIRCBot):
                 else:
                     serv.privmsg(canal,"%s: crève !"%(auteur))
                     log(self.serveur,canal,auteur,message+"[failed]")
                 else:
                     serv.privmsg(canal,"%s: crève !"%(auteur))
                     log(self.serveur,canal,auteur,message+"[failed]")
+            elif cmd == "reload":
+                if auteur in self.ops:
+                    log(self.serveur, canal, auteur, message+"[successful]")
+                    self.reload(canal)
             elif cmd in ["part","leave","dégage"]:
                 if auteur in self.ops and (not (canal in self.stay_channels)
                                            or auteur in self.overops):
             elif cmd in ["part","leave","dégage"]:
                 if auteur in self.ops and (not (canal in self.stay_channels)
                                            or auteur in self.overops):
@@ -401,11 +400,11 @@ class Hung(ircbot.SingleServerIRCBot):
                 serv.privmsg(canal,'%s: Mourir, impératif, 2ème personne du singulier : "meurs" (de rien)'%(auteur))
             if is_tag(message) and not canal in self.quiet_channels:
                 if auteur in self.ops:
                 serv.privmsg(canal,'%s: Mourir, impératif, 2ème personne du singulier : "meurs" (de rien)'%(auteur))
             if is_tag(message) and not canal in self.quiet_channels:
                 if auteur in self.ops:
-                    action=random.choice(config_tag_actions)
+                    action=random.choice(config.tag_actions)
                     serv.action(canal,action.encode("utf8"))
                     self.quiet_channels.append(canal)
                 else:
                     serv.action(canal,action.encode("utf8"))
                     self.quiet_channels.append(canal)
                 else:
-                    answer=random.choice(config_tag_answers)
+                    answer=random.choice(config.tag_answers)
                     for ligne in answer.split("\n"):
                         serv.privmsg(canal,"%s: %s"%(auteur,ligne.encode("utf8")))
             # on essaye de voir si le mot fourni matche la partie en cours
                     for ligne in answer.split("\n"):
                         serv.privmsg(canal,"%s: %s"%(auteur,ligne.encode("utf8")))
             # on essaye de voir si le mot fourni matche la partie en cours
@@ -428,13 +427,7 @@ class Hung(ircbot.SingleServerIRCBot):
         action = ev.arguments()[0]
         auteur = irclib.nm_to_n(ev.source())
         channel = ev.target()
         action = ev.arguments()[0]
         auteur = irclib.nm_to_n(ev.source())
         channel = ev.target()
-        try:
-            test=bot_unicode(action)
-        except UnicodeBotError:
-            serv.privmsg(channel,
-              "%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())
 
     def on_kick(self,serv,ev):
         auteur = irclib.nm_to_n(ev.source())
@@ -446,7 +439,7 @@ class Hung(ircbot.SingleServerIRCBot):
             time.sleep(5)
             serv.join(channel)
             # on ne dit rien au rejoin
             time.sleep(5)
             serv.join(channel)
             # on ne dit rien au rejoin
-            #l1,l2=config_kick_answers,config_kick_actions
+            #l1,l2=config.kick_answers,config.kick_actions
             #n1,n2=len(l1),len(l2)
             #i=random.randrange(n1+n2)
             #if i>=n1:
             #n1,n2=len(l1),len(l2)
             #i=random.randrange(n1+n2)
             #if i>=n1:
@@ -460,20 +453,20 @@ class Hung(ircbot.SingleServerIRCBot):
     
     def quitter(self,chan,leave_message=None):
         if leave_message==None:
     
     def quitter(self,chan,leave_message=None):
         if leave_message==None:
-            leave_message=random.choice(config_leave_messages)
+            leave_message=random.choice(config.leave_messages)
         self.serv.part(chan,message=leave_message.encode("utf8"))
     
     def mourir(self):
         self.serv.part(chan,message=leave_message.encode("utf8"))
     
     def mourir(self):
-        quit_message=random.choice(config_quit_messages)
+        quit_message=random.choice(config.quit_messages)
         self.die(msg=quit_message.encode("utf8"))
     
     def get_scores(self):
         self.die(msg=quit_message.encode("utf8"))
     
     def get_scores(self):
-        f=open(config_scores_file)
+        f=open(config.scores_file)
         scores=pickle.load(f)
         f.close()
         return scores
     def save_scores(self,scores):
         scores=pickle.load(f)
         f.close()
         return scores
     def save_scores(self,scores):
-        f=open(config_scores_file,'w')
+        f=open(config.scores_file,'w')
         pickle.dump(scores,f)
         f.close()
     def add_score(self,dico):
         pickle.dump(scores,f)
         f.close()
     def add_score(self,dico):
@@ -504,7 +497,17 @@ class Hung(ircbot.SingleServerIRCBot):
         self.add_score(contribs_score)
         serv.privmsg(canal,"Contributions : %s"%("  ".join(contribs)) )
         self.play_status[canal]=[None,None,None]
         self.add_score(contribs_score)
         serv.privmsg(canal,"Contributions : %s"%("  ".join(contribs)) )
         self.play_status[canal]=[None,None,None]
-
+    
+    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()
     def start_as_daemon(self, outfile):
         sys.stderr = Logger(outfile)
         self.start()
@@ -555,6 +558,10 @@ if __name__=="__main__":
         print "Server Unknown : %s"%(serveur)
         exit(404)
     hung=Hung(serveur,debug)
         print "Server Unknown : %s"%(serveur)
         exit(404)
     hung=Hung(serveur,debug)
+    # Si on reçoit un SIGHUP, on reload la config
+    def sighup_handler(signum, frame):
+        hung.reload("SIGHUP")
+    signal.signal(signal.SIGHUP, sighup_handler)
     if daemon:
         child_pid = os.fork()
         if child_pid == 0:
     if daemon:
         child_pid = os.fork()
         if child_pid == 0: