# 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"]
+ return config.logfile_template%(serveurs[serveur])
class UnicodeBotError(Exception):
pass
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
-
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:
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
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"]
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))
+ 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(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"
log(self.serveur,channel,u"$Date$".encode("utf8"),("%s : %s"%(date, evenement)).encode("utf8"))
serv.privmsg(channel,evenement.encode("utf8"))
token=time.time()
- self.play_status[channel]=[1,date,evenement,token]
- serv.execute_delayed(random.randrange(ttrig*3,ttrig*5),self.give_indice,(serv,channel,token))
+ # 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(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_clue<time.time():
+ if self.play_status[channel][-1]+config.time_incompressible_clue<time.time():
token=self.play_status[channel][-1]
if self.play_status[channel][-1]==token:
date=self.play_status[channel][1]
indice=date[:5]
serv.privmsg(channel,"indice : %s"%(indice).encode("utf8"))
self.play_status[channel][0]=2
- serv.execute_delayed(random.randrange(ttrig*1,ttrig*3),self.give_answer,(serv,channel,token))
+ serv.execute_delayed(random.randrange(config.ttrig*1,config.ttrig*3),self.give_answer,(serv,channel,token))
def give_answer(self,serv,channel,token):
if self.play_status[channel][0]==2 and self.play_status[channel][-1]==token:
date=self.play_status[channel][1]
serv.privmsg(channel,"C'était le %s"%(date).encode("utf8"))
token=time.time()
self.play_status[channel]=[0,token]
- serv.execute_delayed(random.randrange(Ttrig*5,Ttrig*10),self.start_enigme,(serv,channel,token))
+ serv.execute_delayed(random.randrange(config.Ttrig*5,config.Ttrig*10),self.start_enigme,(serv,channel,token))
def get_enigme(self):
# on récupère les dates
- f=open(config_source_file)
+ f=open(config.source_file)
l=f.readlines()
f.close()
l=[i.split(" : ",2) for i in l]
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, random.choice(config.utf8_fail_answers).encode("utf8"))
return
message=message.split()
cmd=message[0].lower()
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"""
+ NOQUIET Opposé de QUIET
+ RELOAD Me fait recharger la conf"""
helpmsg_overops="""
SCORES {DEL|ADD|SUB} Tu veux un dessin ?
SAY Fais envoyer un message sur un chan ou à une personne
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 in ["states","status"]:
if auteur in self.overops:
for k in self.play_status.keys():
auteur = irclib.nm_to_n(ev.source())
canal = ev.target()
message = ev.arguments()[0]
+ self.update_activity(canal,auteur)
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 config.utf8_trigger and not canal in self.quiet_channels:
+ serv.privmsg(canal, (u"%s: %s"%(auteur, random.choice(config.utf8_fail_answers))).encode("utf8"))
return
tryother=False
pour_moi,message=self.pourmoi(serv,message)
else:
serv.privmsg(canal,"%s: crève !"%(auteur))
log(self.serveur,canal,auteur,message+"[failed]")
- if cmd in ["meur", "meurt","meurre","meurres"] and not canal in self.quiet_channels:
+ elif cmd in ["meur", "meurt","meurre","meurres"] and not canal in self.quiet_channels:
serv.privmsg(canal,'%s: Mourir, impératif, 2ème personne du singulier : "meurs" (de rien)'%(auteur))
- if cmd in ["part","leave","dégage"]:
+ 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):
self.quitter(canal)
serv.privmsg(canal,"%s: Non, je reste !"%(auteur))
log(self.serveur,canal,auteur,message+"[failed]")
- if cmd in ["deviens","pseudo"]:
+ elif cmd in ["deviens","pseudo"]:
if auteur in self.ops:
become=args
serv.nick(become)
log(self.serveur,canal,auteur,message+"[successful]")
- if cmd in ["coucou"] and not canal in self.quiet_channels:
+ elif cmd in ["coucou"] and not canal in self.quiet_channels:
serv.privmsg(canal,"%s: coucou"%(auteur))
- if cmd in ["ping"] and not canal in self.quiet_channels:
+ elif cmd in ["ping"] and not canal in self.quiet_channels:
serv.privmsg(canal,"%s: pong"%(auteur))
- if cmd in ["date","dates","histoire","énigme","enigme","encore"]:
+ elif cmd in ["date","dates","histoire","énigme","enigme","encore"]:
if canal in self.play_channels:
if self.play_status.get(canal,[-1])[0]==0:
try:
serv.privmsg(canal,("%s: Rappel : %s"%(auteur,self.play_status[canal][2])).encode("utf8") )
else:
serv.privmsg(canal,"%s: pas ici…"%(auteur))
- if cmd in ["score","!score"]:
+ elif cmd in ["score","!score"]:
serv.privmsg(auteur,"Votre score : %s"%(self.get_scores().get(auteur,0)) )
- if cmd in ["scores","!scores"]:
+ elif cmd in ["scores","!scores"]:
scores=self.get_scores().items()
# trie par score
scores.sort(lambda x,y:cmp(x[1],y[1]))
# trie par pseudo
scores.sort(lambda x,y:cmp(x[0].lower(),y[0].lower()))
serv.privmsg(auteur,"Scores by pseudo : "+" ; ".join(["%s %s"%(i[0],i[1]) for i in scores]))
- if cmd=="indice" and canal in self.play_channels:
+ elif cmd == "indice" and canal in self.play_channels:
self.give_indice(serv,canal,None)
- if is_tag(message) and not canal in self.quiet_channels:
+ elif 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")))
else:
if tryother:
if self.play_status.get(canal,[-1])[0] in [1,2]:
answer=self.play_status[canal][1]
+ flag_century=self.play_status[canal][3]
try:
score_obtenu=reussi(message.decode("utf8"),answer,auteur)
except GoodCentury:
- serv.privmsg(canal,"%s: C'est le bon siècle, mais pas la bonne année, cherche encore ;)"%(auteur))
+ if not flag_century:
+ 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:
+ 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))
+ 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())
victime = ev.arguments()[0]
raison = ev.arguments()[1]
if victime==self.nick:
- log(self.serveur,"%s kické par %s (raison : %s)" %(victime,auteur,raison))
+ log(self.serveur,"%s kické de %s par %s (raison : %s)" %(victime,channel,auteur,raison))
time.sleep(5)
serv.join(channel)
+ self.update_activity(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:
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:
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:
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
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()
def _getnick(self):
return self.serv.get_nickname()
nick = property(_getnick)
+
+ def update_activity(self,canal,pseudo):
+ if not pseudo in config.idle_bots:
+ self.last_activity[canal]=time.time()
+ def is_active(self,canal):
+ return time.time()-self.last_activity[canal]<config.idle_time
+
+ 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()
+
+
+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:
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()
+