# 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 irclib
-import ircbot
import threading
import random
import time
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_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"}
- return config_logfile_template%(serveurs[serveur])
-config_overops=["[20-100]","[20-100]_"]
-config_ops=[]
-config_report_bugs_to=["[20-100]"]
-config_dico_mots="mots.txt"
-config_dico_defs="definitions.txt"
+# 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])
def log(serveur,channel,auteur=None,message=None):
f=open(get_config_logfile(serveur),"a")
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()
+
class UnicodeBotError(Exception):
pass
def bot_unicode(chain):
except UnicodeDecodeError as exc:
raise UnicodeBotError
+def remplace_accents(chaine):
+ chaine=chaine.lower()
+ remplacements = {u"á":u"a",u"à":u"a",u"â":u"a",u"ä":u"a",u"é":u"e",u"è":u"e",u"ê":u"e",u"ë":u"e",u"í":u"i",u"ì":u"i",u"î":u"i",u"ï":u"i",u"ó":u"o", u"ò":u"o",u"ô":u"o",u"ö":u"o",u"ú":u"u",u"ù":u"u",u"û":u"u",u"ü":u"u",u"ý":u"y",u"ỳ":u"y",u"ŷ":u"y",u"ÿ":u"y",u"œ":u"oe",u"æ":u"ae"}
+ for avant,apres in remplacements.items():
+ chaine=chaine.replace(avant,apres)
+ return chaine
+
+def is_something(chain,matches,avant=u".*(?:^| )",apres=u"(?:$|\.| |,|;).*",case_sensitive=False,debug=False):
+ if case_sensitive:
+ chain=unicode(chain,"utf8")
+ else:
+ chain=unicode(chain,"utf8").lower()
+ allmatches="("+"|".join(matches)+")"
+ reg=(avant+allmatches+apres).lower()
+ o=re.match(reg,chain)
+ return o
+
+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))
+ temporary_pseudo=config.irc_pseudo+str(random.randrange(10000,100000))
ircbot.SingleServerIRCBot.__init__(self, [(serveur, 6667)],
- temporary_pseudo,"Bot irc pour embêter Shadobot", 10)
+ 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.play_status={}
- self.quiet_channels=config_quiet_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.lives={}
+ self.tried_letters={}
+ 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"]
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()
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 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:
if auteur in self.ops:
if len(message)>1:
if message[1] in self.play_channels:
- serv.privmsg(auteur,"Je play déjà sur %s."%(message[1]))
- log(self.serveur,"priv",auteur," ".join(message)+"[failed]")
+ if len(message) > 2:
+ self.start_partie(serv, message[1], " ".join(message[2:]))
+ else:
+ serv.privmsg(auteur,"Je play déjà sur %s."%(message[1]))
+ log(self.serveur,"priv",auteur," ".join(message)+"[failed]")
else:
self.play_channels.append(message[1])
- self.play_status[message[1]]=[[None,None]]
+ self.play_status[message[1]]=[None,None,None]
serv.privmsg(auteur,"Play channels : "+" ".join(self.play_channels))
log(self.serveur,"priv",auteur," ".join(message)+"[successful]")
else:
elif cmd in ["states","status"]:
if auteur in self.overops:
for k in self.play_status.keys():
- serv.privmsg(auteur,"%s : %s (%s)"%(k,"".join([str(i[0]) for i in self.play_status[k][0]])
- ,self.play_status[k][1]))
+ if self.play_status[k]==[None,None,None]:
+ serv.privmsg(auteur,"None")
+ else:
+ serv.privmsg(auteur,"%s : %s (%s) [%s]"%(k,"".join([str(i[0]) for i in self.play_status[k][0]])
+ ,self.play_status[k][1], self.play_status[k][2]))
elif cmd=="die":
if auteur in self.overops:
log(self.serveur,"priv",auteur," ".join(message)+"[successful]")
- self.die()
+ self.mourir()
else:
notunderstood=True
elif cmd=="quiet":
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:]))
serv.privmsg(auteur,"Syntaxe : DO <channel> <action>")
else:
notunderstood=True
+ elif cmd in ["score","scores"]:
+ self.send_scores(serv,auteur)
else:
notunderstood=True
if notunderstood:
obfuskated=" ".join([lettre[0] if lettre[1] else "_" for lettre in mot])
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()]
- indice = random.randrange(0,len(mots))
- mot,definition=mots[indice],defs[indice]
+ def start_partie(self, serv, canal, mot=None):
+ if mot is None:
+ 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]
+ else:
+ definition = "(custom word)"
# ' et - sont considérés comme déjà devinés
- mot = [(lettre,lettre in "'-") for lettre in list(mot)]
- self.play_status[canal]=[mot,definition]
+ mot = mot.upper()
+ mot = [(lettre,lettre in config.non_guess_chars) for lettre in list(mot)]
+ self.play_status[canal]=[mot,definition,{}]
+ self.tried_letters[canal] = set ()
+ self.lives[canal] = config.lives
self.affiche_mot(serv, canal, begin="Devinez")
def on_pubmsg(self, serv, ev):
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))
- return
+ 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
pour_moi,message=self.pourmoi(serv,message)
if pour_moi and message.split()!=[]:
cmd=message.split()[0].lower()
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 == "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):
- 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)
else:
serv.privmsg(canal,"%s: Non, je reste !"%(auteur))
log(self.serveur,canal,auteur,message+"[failed]")
- elif cmd in ["play","jeu","encore","again","partie","pendu","game","mot"]:
+ elif cmd in ["play","jeu","encore","again","partie","pendu","game","mot","go","allez"]:
if not canal in self.quiet_channels and canal in self.play_channels:
if self.play_status.has_key(canal):
- if self.play_status[canal]==[[None,None]]:
+ if self.play_status[canal]==[None,None,None]:
self.start_partie(serv, canal)
else:
self.affiche_mot(serv, canal, begin="%s: Rappel"%(auteur))
else:
- self.play_status[canal]=[[None,None]]
+ self.play_status[canal]=[None,None,None]
self.start_partie(serv, canal)
elif not canal in self.play_channels:
serv.privmsg(canal,"%s: pas ici…"%(auteur))
elif (cmd in list("azertyuiopqsdfghjklmwxcvbn") and canal in self.play_channels
- and self.play_status.has_key(canal) and self.play_status[canal]!=[[None,None]]):
+ and self.play_status.has_key(canal) and self.play_status[canal]!=[None,None,None]):
giv_let=cmd.upper()
liste=self.play_status[canal][0]
listeapres=[(lettre[0],lettre[1] or lettre[0]==giv_let) for lettre in liste]
if liste!=listeapres:
+ nbtrouvees=(sum([lettre[1] for lettre in listeapres if not lettre[0] in "'-()"])
+ - sum([lettre[1] for lettre in liste if not lettre[0] in "'-()"]))
+ if self.play_status[canal][2].has_key(auteur):
+ self.play_status[canal][2][auteur]+= nbtrouvees
+ else:
+ self.play_status[canal][2][auteur] = nbtrouvees
self.play_status[canal][0]=listeapres
self.affiche_mot(serv, canal, begin="%s placé"%(giv_let))
+ else:
+ if not giv_let in self.tried_letters[canal]:
+ # On perd une chance
+ self.lives[canal] -= 1
+ if self.lives[canal] > 0:
+ serv.privmsg(canal, "Pas de %s. Plus que %s chances…" % (giv_let, self.lives[canal]))
+ if self.lives[canal] == 0:
+ serv.privmsg(canal, "Pas de %s." % (giv_let))
+ self.perd(serv, canal)
+ return
+ self.tried_letters[canal].add(giv_let)
if all([lettre[1] for lettre in listeapres]):
- serv.privmsg(canal,"Bravo ! C'était %s"%("".join([lettre[0] for lettre in listeapres])))
- self.play_status[canal]=[[None,None]]
+ self.gagne(serv, canal)
-
+ elif cmd in ["score","scores","!score","!scores"]:
+ self.send_scores(serv,auteur)
if 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 is_tag(message) and not canal in self.quiet_channels:
+ if auteur in self.ops:
+ action=random.choice(config.tag_actions)
+ serv.action(canal,action.encode("utf8"))
+ self.quiet_channels.append(canal)
+ else:
+ 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
+ mot = cmd
+ if canal in self.play_channels and 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
+
def on_action(self, serv, ev):
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())
+ 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)
+ f.close()
+ return scores
+ def save_scores(self,scores):
+ f=open(config.scores_file,'w')
+ pickle.dump(scores,f)
+ f.close()
+ def add_score(self,dico):
+ scores=self.get_scores()
+ for k,v in dico.items():
+ if scores.has_key(k):
+ scores[k]+=v
+ else:
+ scores[k]=v
+ self.save_scores(scores)
+ def send_scores(self, serv, destinataire):
+ scores=self.get_scores()
+ scores=scores.items()
+ scores.sort(lambda x,y:cmp(x[1],y[1]))
+ scores.reverse()
+ serv.privmsg(destinataire,"Scores by score : "+" ; ".join(["%s %s"%(k,v) for (k,v) in scores]) )
+ scores.sort(lambda x,y:cmp(x[0].lower(),y[0].lower()))
+ serv.privmsg(destinataire,"Scores by pseudo : "+" ; ".join(["%s %s"%(k,v) for (k,v) in scores]) )
+
+ def gagne(self, serv, canal, bonus=None, bonusvalue=2):
+ serv.privmsg(canal,"Bravo !")
+ realword = self.reveal_word(serv, canal)
+ nlettre=float(len(realword.replace("'","").replace("-","")))
+ contribs=["%s:%s%%%s"%(pseudo,str(int(100*contrib/nlettre)),("+bonus(%s)"%(bonusvalue))*(bonus==pseudo)) for pseudo,contrib in self.play_status[canal][2].items()]
+ contribs_score={pseudo:int(10*contrib/nlettre) for pseudo,contrib in self.play_status[canal][2].items()}
+ self.add_score(contribs_score)
+ serv.privmsg(canal,"Contributions : %s"%(" ".join(contribs)) )
+ self.play_status[canal]=[None,None,None]
+
+ def reveal_word(self, serv, canal):
+ realword="".join([lettre[0] for lettre in self.play_status[canal][0]])
+ serv.privmsg(canal, "C'était %s." % (realword))
+ definition = self.play_status[canal][1]
+ serv.privmsg(canal,definition)
+ return realword
+
+ def perd(self, serv, canal):
+ serv.privmsg(canal,"Pendu !")
+ self.reveal_word(serv, canal)
+ 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()
+
+
+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()
+ # 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:
+ 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()
+