X-Git-Url: http://gitweb.pimeys.fr/?p=bots%2Fhistorien.git;a=blobdiff_plain;f=historien.py;h=66a4ff6a9b604b2d7d79793c1da72445a6e0d32d;hp=57e2e0cae175a06540c42eb965bf136b55831ff7;hb=HEAD;hpb=68b90c3209dd9aedb444e8e246572b982ff8f972 diff --git a/historien.py b/historien.py index 57e2e0c..66a4ff6 100755 --- a/historien.py +++ b/historien.py @@ -5,61 +5,31 @@ # 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 +import os from cast_as_date import * -config_password="EtTaMère,ElleEstNéeQuand?" -config_pseudo="historien" -config_chanlist=["#bot","#flood"] -config_play_channels=["#flood"] -config_stay_channels=["#flood","#bot"] -config_overops=["[20-100]","[20-100]_"] -config_ops=["PEB","Petite-Peste"] +# 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 + +# Fichier de conf +import config -config_source_file="dates.txt" -config_played_file_template="played.%s.txt" #il faut rajouter le nom du serveur def get_config_played_file(serveur): serveurs={"acoeur.crans.org":"acoeur","irc.crans.org":"crans"} - return config_played_file_template%(serveurs[serveur]) -ttrig=120 #time trigger (normalement 120, mais diminué pour les tests) -Ttrig=600 #between two enigms -config_time_incompressible=15 #on peut pas retrigger en dessous de ce temps (60) -config_time_incompressible_clue=60 #on peut pas forcer la demande d'indice en dessous + return config.played_file_template%(serveurs[serveur]) -config_score_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_level2=[] -config_level3=[] - -config_debug_stdout = True -config_logfile_template="historien.%s.log" def get_config_logfile(serveur): serveurs={"acoeur.crans.org":"acoeur","irc.crans.org":"crans"} - return config_logfile_template%(serveurs[serveur]) - -config_quit_messages=[u"%s : %s quitte le serveur IRC"] - -config_leave_messages=[u"%s : %s quitte le channel"] - -# Quand personne ne cause, on finit par se taire -# 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","___"] + return config.logfile_template%(serveurs[serveur]) class UnicodeBotError(Exception): pass @@ -77,24 +47,32 @@ 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() - -config_score_annee=1 -config_score_mois=3 -config_score_jour=4 - -config_noscore=["[20-100]","[20-100]_"] # parce que 20-100 est nul en histoire +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)) + if blackit and not exceptit: + return True class GoodCentury(Exception): pass +class GoodDeceny(Exception): + pass + def reussi(message,answer,auteur): - if auteur in config_level3: + if auteur in config.level3: return answer in message - if auteur in config_level2: + if auteur in config.level2: return answer in message else: try: @@ -105,11 +83,13 @@ def reussi(message,answer,auteur): realdate.reverse() score=0 if date[0]==realdate[0]: - score=config_score_annee + score=config.score_annee if date[1]==realdate[1]: - score+=config_score_mois + score+=config.score_mois if date[2]==realdate[2]: - score+=config_score_jour + score+=config.score_jour + elif date[0]/10 == realdate[0]/10: + raise GoodDeceny elif date[0]/100 == realdate[0]/100: raise GoodCentury return score @@ -125,37 +105,37 @@ 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) class RefuseError(Exception): pass class Historien(ircbot.SingleServerIRCBot): def __init__(self,serveur,debug=False): - temporary_pseudo=config_pseudo+str(random.randrange(10000,100000)) + temporary_pseudo=config.pseudo+str(random.randrange(10000,100000)) ircbot.SingleServerIRCBot.__init__(self, [(serveur, 6667)], temporary_pseudo,"Un bot irc qui a au moins l'agreg d'histoire", 10) self.debug=debug self.serveur=serveur - self.overops=config_overops - self.ops=self.overops+config_ops - 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.chanlist=config.chanlist + self.stay_channels=config.stay_channels + self.play_channels=config.play_channels self.play_status={i:[0] for i in self.play_channels} self.last_activity={} self.quiet_channels=[] def give_me_my_pseudo(self,serv): - serv.privmsg("NickServ","RECOVER %s %s"%(config_pseudo,config_password)) - serv.privmsg("NickServ","RELEASE %s %s"%(config_pseudo,config_password)) + serv.privmsg("NickServ","RECOVER %s %s"%(config.pseudo,config.password)) + serv.privmsg("NickServ","RELEASE %s %s"%(config.pseudo,config.password)) time.sleep(0.3) - serv.nick(config_pseudo) + serv.nick(config.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_password)) + serv.privmsg("NickServ","identify %s"%(config.password)) log(self.serveur,"Connected") if self.debug: self.chanlist=["#bot"] @@ -163,23 +143,24 @@ class Historien(ircbot.SingleServerIRCBot): for c in self.chanlist: log(self.serveur,"JOIN %s"%(c)) serv.join(c) - self.update_activity(c,"") # la chaîne vide ne sera jamais un nom de bot et donc marchera toujours for c in self.play_channels: token=time.time()-3600 self.play_status[c]=[0,token] - serv.execute_delayed(random.randrange(ttrig),self.start_enigme,(serv,c,token)) + if config.auto_trigger: + serv.execute_delayed(random.randrange(config.ttrig),self.start_enigme,(serv,c,token)) def start_enigme(self,serv,channel,token=None): # On reste silencieux si lechan n'est pas actif if not self.is_active(channel): - serv.execute_delayed(ttrig*5,self.start_enigme,(serv,channel,token)) + if config.auto_trigger: + serv.execute_delayed(config.ttrig*5,self.start_enigme,(serv,channel,token)) return if self.play_status[channel][0]==0 and channel in self.play_channels: ok="skip" if token==self.play_status[channel][-1]: ok="do_it" if token==None: - if time.time() > self.play_status[channel][-1]+config_time_incompressible: + if time.time() > self.play_status[channel][-1]+config.time_incompressible: ok="do_it" else: ok="refuse" @@ -188,34 +169,35 @@ 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)) + serv.execute_delayed(random.randrange(config.ttrig*3,config.ttrig*5),self.give_indice,(serv,channel,token)) elif ok=="refuse": raise RefuseError def give_indice(self,serv,channel,token): if self.play_status[channel][0]==1: if token==None: # c'est donc que l'indice a été demandé - if self.play_status[channel][-1]+config_time_incompressible_clueconfig_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: + if auteur in config.noscore: score_obtenu=0 self.add_score(auteur,score_obtenu) token=time.time() self.play_status[canal]=[0,token] - serv.execute_delayed(random.randrange(Ttrig*5,Ttrig*10),self.start_enigme,(serv,canal,token)) + if config.auto_trigger: + serv.execute_delayed(random.randrange(config.Ttrig*5,config.Ttrig*10),self.start_enigme,(serv,canal,token)) def on_kick(self,serv,ev): auteur = irclib.nm_to_n(ev.source()) @@ -608,9 +612,9 @@ 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,force=True) # 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: @@ -620,7 +624,7 @@ class Historien(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) try: leave_message=leave_message%(time.strftime("le %d/%m/%Y à %T").decode("utf8"),self.nick) except: @@ -628,7 +632,7 @@ class Historien(ircbot.SingleServerIRCBot): 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) try: quit_message=quit_message%(time.strftime("le %d/%m/%Y à %T").decode("utf8"),self.nick) except: @@ -636,7 +640,7 @@ class Historien(ircbot.SingleServerIRCBot): self.die(msg=quit_message.encode("utf8")) def get_scores(self): - f=open(config_score_file) + f=open(config.score_file) scores=pickle.load(f) f.close() return scores @@ -650,7 +654,7 @@ class Historien(ircbot.SingleServerIRCBot): self.save_scores(scores) def save_scores(self,scores): - f=open(config_score_file,"w") + f=open(config.score_file,"w") pickle.dump(scores,f) f.close() @@ -658,22 +662,64 @@ class Historien(ircbot.SingleServerIRCBot): return self.serv.get_nickname() nick = property(_getnick) - def update_activity(self,canal,pseudo): - if not pseudo in config_idle_bots: + def update_activity(self,canal="",pseudo="",message="",force=False): + if force or (not pseudo in config.idle_bots and all([not re.match(ignore, message) for ignore in config.idle_messages])): self.last_activity[canal]=time.time() def is_active(self,canal): - return time.time()-self.last_activity[canal] [--debug]" + print "Usage : historien.py [--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 +728,25 @@ if __name__=="__main__": print "Server Unknown : %s"%(serveur) exit(404) historien=Historien(serveur,debug) - historien.start() + # Si on reçoit un SIGHUP, on reload la config + def sighup_handler(signum, frame): + historien.reload("SIGHUP") + signal.signal(signal.SIGHUP, sighup_handler) + 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() +