From 6df672cc4767aaa6e1510a4d681f4d4f9108fc0c Mon Sep 17 00:00:00 2001 From: Vincent Le Gallic Date: Mon, 30 Jul 2012 19:17:21 +0200 Subject: [PATCH] fichier de conf + reloadable y compris par SIGHUP + utf8_trigger --- config.py | 47 +++++++++++++++ hung.py | 167 ++++++++++++++++++++++++++++-------------------------- 2 files changed, 134 insertions(+), 80 deletions(-) create mode 100644 config.py diff --git a/config.py b/config.py new file mode 100644 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 49371f1..96c8a58 100755 --- a/hung.py +++ b/hung.py @@ -3,7 +3,7 @@ # 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 @@ -22,50 +22,13 @@ 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_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"} - 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") @@ -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") - if config_debug_stdout: + if config.debug_stdout: 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 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]) @@ -116,31 +79,31 @@ def is_mot(mot,liste): 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 - 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.quiet_channels=config_quiet_channels + self.quiet_channels=config.quiet_channels 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) - 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) - 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"] @@ -164,13 +127,40 @@ class Hung(ircbot.SingleServerIRCBot): 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 - 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: @@ -292,6 +282,12 @@ class Hung(ircbot.SingleServerIRCBot): 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:])) @@ -322,8 +318,8 @@ class Hung(ircbot.SingleServerIRCBot): 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 @@ -338,9 +334,8 @@ class Hung(ircbot.SingleServerIRCBot): 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()!=[]: @@ -356,6 +351,10 @@ class Hung(ircbot.SingleServerIRCBot): 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): @@ -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: - action=random.choice(config_tag_actions) + action=random.choice(config.tag_actions) 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 @@ -428,13 +427,7 @@ class Hung(ircbot.SingleServerIRCBot): 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()) @@ -446,7 +439,7 @@ class Hung(ircbot.SingleServerIRCBot): 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: @@ -460,20 +453,20 @@ class Hung(ircbot.SingleServerIRCBot): 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): - 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): - f=open(config_scores_file) + f=open(config.scores_file) 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): @@ -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] - + + 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() @@ -555,6 +558,10 @@ if __name__=="__main__": 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: -- 2.39.2