X-Git-Url: http://gitweb.pimeys.fr/?p=bots%2Fhung.git;a=blobdiff_plain;f=hung.py;h=2c3e7086dbb9955b8b5e970cf8e30a1a5b736911;hp=0691b364cb245dbff411ada7304443497dc53c88;hb=HEAD;hpb=85cad7e8e4154a14729a3e2594b4a23a27793994 diff --git a/hung.py b/hung.py index 0691b36..2c3e708 100755 --- a/hung.py +++ b/hung.py @@ -3,10 +3,8 @@ # 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 @@ -14,42 +12,48 @@ import socket, ssl, json 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") - if auteur==message==None: + if auteur == message == None: # alors c'est que c'est pas un channel mais juste une ligne de log chain="%s %s"%(time.strftime("%F %T"),channel) 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() +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 UnicodeBotError(Exception): pass def bot_unicode(chain): @@ -58,32 +62,61 @@ 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.startswith(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"] @@ -102,18 +135,47 @@ class Hung(ircbot.SingleServerIRCBot): return (False,message) def on_privmsg(self, serv, ev): + if ignore_event(serv, ev): + return message=ev.arguments()[0] auteur = irclib.nm_to_n(ev.source()) 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: @@ -131,7 +193,7 @@ class Hung(ircbot.SingleServerIRCBot): 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: @@ -145,11 +207,14 @@ class Hung(ircbot.SingleServerIRCBot): 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: @@ -198,12 +263,15 @@ class Hung(ircbot.SingleServerIRCBot): 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": @@ -232,6 +300,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:])) @@ -248,6 +322,8 @@ class Hung(ircbot.SingleServerIRCBot): serv.privmsg(auteur,"Syntaxe : DO ") else: notunderstood=True + elif cmd in ["score","scores"]: + self.send_scores(serv,auteur) else: notunderstood=True if notunderstood: @@ -259,27 +335,34 @@ class Hung(ircbot.SingleServerIRCBot): 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): + if ignore_event(serv, ev): + return auteur = irclib.nm_to_n(ev.source()) canal = ev.target() message = ev.arguments()[0] 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() @@ -290,14 +373,18 @@ class Hung(ircbot.SingleServerIRCBot): 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) @@ -307,64 +394,239 @@ class Hung(ircbot.SingleServerIRCBot): 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 ["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 = message + 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) + return 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 chance%s…" % (giv_let, self.lives[canal], "s" * (self.lives[canal] > 1))) + 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]): - realword="".join([lettre[0] for lettre in self.play_status[canal][0]]) - definition = self.play_status[canal][1] - serv.privmsg(canal,"Bravo ! C'était %s"%(realword)) - serv.privmsg(canal,definition) - self.play_status[canal]=[[None,None]] + self.gagne(serv, canal) - - 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)) else: pass + def on_action(self, serv, ev): + if ignore_event(serv, ev): + return 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([l for l in realword if not l in config.non_guess_chars])) + 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.0*contrib/nlettre)+(bonusvalue*(bonus==pseudo)) 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 [--debug]" + print "Usage : hung.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/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() +