# 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
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
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):
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):
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: ")"""
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:
Me rend silencieux sur le channel.""",None],
"noquiet": [None,"""NOQUIET <channel>
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 <channel>
+ Me fait spawner sur le channel.""",None],
+ "reload": [None,"""RELOAD
+ Recharge la configuration.""",None],
"say": [None,None,"""SAY <channel> <message>
Me fait parler sur le channel."""],
"do": [None,None,"""DO <channel> <action>
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:
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":
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 <channel>")
+ 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," ".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":
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 <pseudo> [<n>]")
else:
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()!=[]:
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()
#~ 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()
#~ 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)
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
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 <serveur> [--debug]"
+ print "Usage : saturnin.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/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()
+
projects / bots / saturnin.git / blobdiff