X-Git-Url: http://gitweb.pimeys.fr/?p=bots%2Fsaturnin.git;a=blobdiff_plain;f=saturnin.py;h=848019c2ea51a4001b4295f0862eb468ddb1fae8;hp=931ecd7e5a77929deaef335b76aa122619cd886b;hb=HEAD;hpb=e5a491afbd015fa9a95a2fef3199fb8852ee93e2 diff --git a/saturnin.py b/saturnin.py index 931ecd7..848019c 100755 --- a/saturnin.py +++ b/saturnin.py @@ -6,8 +6,6 @@ # Un bot IRC pour remplacer le canard. # parce que le canard, c'est le bien et que braice ne pong pas -import irclib -import ircbot import threading import random import time @@ -15,8 +13,15 @@ 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 + # on récupère la config import config @@ -52,6 +57,18 @@ regexp_pan = re.compile(u".*(" + "|".join(config.killwords) + u").*") def is_pan(chain): return regexp_pan.match(unicode(chain,"utf8").lower()) +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): @@ -72,8 +89,12 @@ class Saturnin(ircbot.SingleServerIRCBot): self.chanlist=config.chanlist self.stay_channels=config.stay_channels self.quiet_channels=config.quiet_channels - self.spawn_channels=config.spawn_channels - self.status = { chan : False for chan in self.spawn_channels } + self.play_channels=config.play_channels + self.status = { chan : [0, None] for chan in self.play_channels } + # 0 : pas de spawn prévu + # 1 : un spawn prévu, + # avec en deuxième paramètre le timestamp du moment où il a été déclenché (pas du moment où il se fera) + # 2 : alive self.last_perdu=0 def give_me_my_pseudo(self,serv): @@ -88,11 +109,14 @@ class Saturnin(ircbot.SingleServerIRCBot): serv.privmsg("NickServ","IDENTIFY %s"%(config.irc_password)) log(self.serveur,"Connected") if self.debug: - self.chanlist = self.spawn_channels = ["#bot"] - self.status = { chan : False for chan in self.spawn_channels } + self.chanlist = self.play_channels = ["#bot"] + self.status = { chan : [0, 0] for chan in self.play_channels } for c in self.chanlist: log(self.serveur,"JOIN %s"%(c)) serv.join(c) + if c in self.play_channels: + spawn_delay = random.randrange(*config.spawn_delays) + self.spawn(c, time.time(), spawn_delay) def pourmoi(self, serv, message): """renvoie (False,lemessage) ou (True, le message amputé de "pseudo: ")""" @@ -104,6 +128,8 @@ class Saturnin(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: @@ -130,6 +156,14 @@ class Saturnin(ircbot.SingleServerIRCBot): Me rend silencieux sur le channel.""",None], "noquiet": [None,"""NOQUIET Me rend la parole sur le channel.""",None], + "play": [None, """PLAY + Passe un channel en mode "jouer" """,None], + "noplay": [None, """NOPLAY + Passe un channel en mode "ne pas jouer" """,None], + "SPAWN": [None, """SPAWN + Me fait spawner sur le channel.""",None], + "reload": [None,"""RELOAD + Recharge la configuration.""",None], "say": [None,None,"""SAY Me fait parler sur le channel."""], "do": [None,None,"""DO @@ -148,7 +182,7 @@ class Saturnin(ircbot.SingleServerIRCBot): Me déconnecte du serveur IRC."""] } helpmsg_default="Liste des commandes disponibles :\nHELP SCORE SCORES" - helpmsg_ops=" JOIN LEAVE QUIET NOQUIET LOST" + helpmsg_ops=" JOIN LEAVE QUIET NOQUIET PLAY NOPLAY SPAWN" helpmsg_overops=" SAY DO STAY NOSTAY OPS OVEROPS KICK DIE" op,overop=auteur in self.ops, auteur in self.overops if len(message)==1: @@ -170,6 +204,8 @@ class Saturnin(ircbot.SingleServerIRCBot): helpmsg+="\n"+helpmsgs[2] else: helpmsg=helpmsgs[2] + if not helpmsg: # Un non-op a demandé de l'aide sur une commande dont il n'est pas censé connaître l'existence + helpmsg = "Commande inacessible." for ligne in helpmsg.split("\n"): serv.privmsg(auteur,ligne) elif cmd=="join": @@ -259,6 +295,51 @@ class Saturnin(ircbot.SingleServerIRCBot): log(self.serveur,"priv",auteur," ".join(message)+"[failed]") else: notunderstood=True + elif cmd=="play": + 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]") + else: + self.play_channels.append(message[1]) + self.spawn(message[1], 1) + serv.privmsg(auteur,"Play channels : "+" ".join(self.play_channels)) + log(self.serveur,"priv",auteur," ".join(message)+"[successful]") + else: + serv.privmsg(auteur,"Play channels : "+" ".join(self.play_channels)) + else: + notunderstood=True + elif cmd=="noplay": + if auteur in self.ops: + if len(message)>1: + if message[1] in self.play_channels: + self.play_channels.remove(message[1]) + serv.privmsg(auteur,"Play channels : "+" ".join(self.play_channels)) + log(self.serveur,"priv",auteur," ".join(message)+"[successful]") + else: + serv.privmsg(auteur,"Je ne play pas sur %s."%(message[1])) + log(self.serveur,"priv",auteur," ".join(message)+"[failed]") + else: + notunderstood=True + elif cmd=="status": + if auteur in self.ops: + serv.privmsg(auteur,"Status : %s" % (self.status, )) + else: + notunderstood=True + elif cmd=="spawn": + if auteur in self.ops: + if len(message)>1: + if message[1] in self.play_channels: + # Le plus pratique pour pas s'embêter c'est de mettre un + # delay d'une seconde, comme ça .spawn() fait le boulot + self.spawn(message[1], time.time(), 1) + else: + serv.privmsg(auteur, "Je ne joue pas sur %s" % message[1]) + else: + serv.privmsg(auteur, "Syntaxe : SPAWN ") + else: + notunderstood=True elif cmd=="say": if auteur in self.overops and len(message)>2: serv.privmsg(message[1]," ".join(message[2:])) @@ -293,6 +374,12 @@ class Saturnin(ircbot.SingleServerIRCBot): serv.privmsg(auteur," ".join(self.overops)) 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=="score": if len(message)>1: if len(message) in [3,4] and message[1].lower()=="transfert": @@ -320,6 +407,13 @@ class Saturnin(ircbot.SingleServerIRCBot): self.add_score(de,-asked) self.add_score(to,asked) serv.privmsg(auteur,"Transfert de %s points de %s à %s"%(asked,de,to)) + elif len(message) == 2: + scores = self.get_scores() + asked = message[1] + if asked in scores: + serv.privmsg(auteur, "Score de %s : %s" % (asked, scores[asked])) + else: + serv.privmsg(auteur, "%s n'a encore jamais joué (avec succès)." % (asked,)) else: serv.privmsg(auteur,"Syntaxe : SCORE TRANSFERT []") else: @@ -367,25 +461,41 @@ class Saturnin(ircbot.SingleServerIRCBot): def sendscore(self, to): self.serv.privmsg(to, "Votre score : %s"%(self.get_scores().get(to,0)) ) + def _sendscores(self, to, scores): + message = '' + for [nick, score] in scores: + score = '{}: {}'.format(nick, score) + if len(message) < config.MAX_LENGTH - len(score): + message += ' ' + score + else: + self.serv.privmsg(to, message) + message = score + if message: + self.serv.privmsg(to, message) + def sendscores(self, to): scores=self.get_scores().items() # trie par score scores.sort(lambda x,y:cmp(x[1],y[1])) scores.reverse() - self.serv.privmsg(auteur,"Scores by score : "+" ; ".join(["%s %s"%(i[0],i[1]) for i in scores])) + self.serv.privmsg(to, "Scores by score : ") + self._sendscores(to, scores) # trie par pseudo scores.sort(lambda x,y:cmp(x[0].lower(),y[0].lower())) - self.serv.privmsg(auteur,"Scores by pseudo : "+" ; ".join(["%s %s"%(i[0],i[1]) for i in scores])) + self.serv.privmsg(to, "Scores by pseudo : ") + self._sendscores(to, scores) def on_pubmsg(self, serv, ev): + if ignore_event(serv, ev): + return auteur = irclib.nm_to_n(ev.source()) - canal = ev.target() + channel = ev.target() message = ev.arguments()[0] try: test=bot_unicode(message) except UnicodeBotError: - 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")) + if config.utf8_trigger and not channel in self.quiet_channels: + serv.privmsg(channel, (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()!=[]: @@ -396,30 +506,40 @@ class Saturnin(ircbot.SingleServerIRCBot): args="" if cmd in ["meurs","die","crève"]: if auteur in self.overops: - log(self.serveur,canal,auteur,message+"[successful]") + log(self.serveur,channel,auteur,message+"[successful]") self.mourir() else: - serv.privmsg(canal,("%s: %s"%(auteur,random.choice(config.quit_fail_messages))).encode("utf8")) - log(self.serveur,canal,auteur,message+"[failed]") + serv.privmsg(channel,("%s: %s"%(auteur,random.choice(config.quit_fail_messages))).encode("utf8")) + log(self.serveur,channel,auteur,message+"[failed]") + elif cmd == "reload": + if auteur in self.ops: + log(self.serveur, channel, auteur, message+"[successful]") + self.reload(channel) elif cmd in ["part","leave","dégage","va-t-en","tut'tiresailleurs,c'estmesgalets"]: - if auteur in self.ops and (not (canal in self.stay_channels) + if auteur in self.ops and (not (channel in self.stay_channels) or auteur in self.overops): - self.quitter(canal) - log(self.serveur,canal,auteur,message+"[successful]") - if canal in self.chanlist: - self.chanlist.remove(canal) + self.quitter(channel) + log(self.serveur,channel,auteur,message+"[successful]") + if channel in self.chanlist: + self.chanlist.remove(channel) else: - serv.privmsg(canal,("%s: %s"%(auteur,random.choice(config.leave_fail_messages))).encode("utf8")) - log(self.serveur,canal,auteur,message+"[failed]") + serv.privmsg(channel,("%s: %s"%(auteur,random.choice(config.leave_fail_messages))).encode("utf8")) + log(self.serveur,channel,auteur,message+"[failed]") elif cmd == "score": self.sendscore(auteur) elif cmd == "scores": self.sendscores(auteur) - else: - if is_pan(message): - self.shot(auteur) + elif cmd in [u"deviens", u"pseudo"]: + if auteur in self.ops: + become = args + serv.nick(become) + log(self.serveur, channel, auteur, message + "[successful]") + if is_pan(message): + self.shot(channel, auteur) 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() @@ -430,8 +550,7 @@ class Saturnin(ircbot.SingleServerIRCBot): #~ serv.privmsg(channel, (u"%s: %s"%(auteur,random.choice(config.utf8_fail_answers))).encode("utf8")) #~ return #~ mypseudo=self.nick - - + def on_kick(self,serv,ev): auteur = irclib.nm_to_n(ev.source()) channel = ev.target() @@ -449,6 +568,43 @@ class Saturnin(ircbot.SingleServerIRCBot): #~ else: #~ serv.privmsg(channel,l1[i].format(auteur).encode("utf8")) + def spawn(self, channel, timestamp, delay=0): + if channel in self.play_channels: + if delay>0: + self.serv.execute_delayed(delay, self.spawn, (channel, timestamp)) + self.status[channel] = [1, timestamp] + else: + # on teste le timestamp pour pas s'emmêler dans les spawn + infos = self.status.get(channel, [0,0]) + if infos == [1, timestamp]: + spawn_sentence = random.choice(config.canards) + random.choice(config.spawn_sentences) + self.serv.privmsg(channel, spawn_sentence.encode("utf8")) + self.status[channel] = [2, timestamp] + times_up_delay = random.randrange(*config.times_up_delays) + self.serv.execute_delayed(times_up_delay, self.too_slow, (channel, timestamp)) + + def too_slow(self, channel, timestamp): + infos = self.status.get(channel, [0,0]) + if infos == [2, timestamp]: + self.serv.privmsg(channel, random.choice(config.times_up_sentences).encode("utf8")) + respawn_delay = random.randrange(*config.spawn_delays) + self.spawn(channel, time.time(), respawn_delay) + + def shot(self, channel, auteur): + if self.status.get(channel, [0, 0])[0] == 2: + succeed = random.randrange(0,101) > config.proba_miss + if succeed: + self.serv.privmsg(channel, random.choice(config.killed_templates).format(auteur).encode("utf8")) + self.add_score(auteur, 1) + if random.randrange(0, 101) < config.proba_killed_sentence: + self.serv.privmsg(channel, random.choice(config.killed_sentences).encode("utf8")) + respawn_delay = random.randrange(*config.spawn_delays) + self.spawn(channel, time.time(), respawn_delay) + else: + self.serv.privmsg(channel, random.choice(config.miss_templates).format(auteur).encode("utf8")) + if random.randrange(0,101) < config.proba_miss_sentence: + self.serv.privmsg(channel, random.choice(config.miss_sentences).encode("utf8")) + def quitter(self,chan,leave_message=None): if leave_message==None: leave_message=random.choice(config.leave_messages) @@ -464,7 +620,7 @@ class Saturnin(ircbot.SingleServerIRCBot): f.close() return scores - def add_score(self,pseudo,value): + def add_score(self, pseudo, value): scores=self.get_scores() if scores.has_key(pseudo): scores[pseudo]+=value @@ -480,26 +636,88 @@ class Saturnin(ircbot.SingleServerIRCBot): def _getnick(self): return self.serv.get_nickname() nick=property(_getnick) + + 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="saturnin.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 : saturnin.py [--debug]" + print "Usage : saturnin.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/saturnin.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) if "--quiet" in sys.argv: config.debug_stdout=False 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", + "local":"localhost"} try: serveur=serveurs[serveur] except KeyError: print "Server Unknown : %s"%(serveur) exit(404) - bot = Saturnin(serveur,debug) - bot.start() + saturnin = Saturnin(serveur,debug) + # Si on reçoit un SIGHUP, on reload la config + def sighup_handler(signum, frame): + saturnin.reload("SIGHUP") + signal.signal(signal.SIGHUP, sighup_handler) + if daemon: + child_pid = os.fork() + if child_pid == 0: + os.setsid() + saturnin.start_as_daemon(outfile) + else: + # on enregistre le pid de saturnin + pidfile = "/var/run/bots/saturnin.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: + saturnin.start() +