X-Git-Url: http://gitweb.pimeys.fr/?p=bots%2Fbasile.git;a=blobdiff_plain;f=basile.py;h=2fedf6ed76a4a0aa8dc9dfb3a644b892edd0dcc9;hp=75c634199ccb1e4f40947d13acbba5b816143118;hb=4c600255d74b7e058ffa4d0c729c7e495e61b484;hpb=fc1ad7795762c50f2eeadfcb14bd3058b3bfe8c2 diff --git a/basile.py b/basile.py index 75c6341..2fedf6e 100755 --- a/basile.py +++ b/basile.py @@ -52,7 +52,19 @@ def log(serveur, channel, auteur=None, message=None): f.write((chain + u"\n").encode("utf-8")) f.close() if config.debug_stdout: - print chain + print chain.encode("utf-8") + +def ignore_event(serv, ev): + """Retourne ``True`` si il faut ignorer cet évènement.""" + for (blackmask, exceptlist) in config.blacklisted_masks: + usermask = ev.source() + blackit = bool(irclib.mask_matches(usermask, blackmask)) + exceptit = any([bool(irclib.mask_matches(usermask, exceptmask)) for exceptmask in exceptlist]) + if exceptit: # Il est exempté + return False + else: + if blackit: # Il n'est pas exempté et matche la blacklist + return True def regex_join(liste, avant=u".*(?:^| )", apres=u"(?:$|\.| |,|;).*"): """Fabrique une regexp à partir d'une liste d'éléments à matcher.""" @@ -62,8 +74,10 @@ def is_something(chain, regexp=None, matches=[], avant=u".*(?:^| )", apres=u"(?: case_sensitive=False): """Vérifie si chain contient un des éléments de ``matches``. Si ``regexp`` est fournie, c'est simplement elle qui est testée""" - if case_sensitive: + if not case_sensitive: chain = chain.lower() + apres = apres.lower() + avant = avant.lower() if regexp == None: regexp = regex_join(matches, avant, apres) regexp = re.compile(regexp) @@ -205,6 +219,7 @@ class Basile(ircbot.SingleServerIRCBot): self.quiet_channels = config.quiet_channels self.last_perdu = 0 + ### Communication NK def new_connection_NK(self, serv, username, password, typ="bdd"): """Renvoie (``True``, ) ou bien (``False``, None)""" @@ -235,7 +250,13 @@ class Basile(ircbot.SingleServerIRCBot): return (True, info, sock) else: return (False, None, None) - + + ### Utilitaires + def _getnick(self): + """Récuère le nick effectif du bot sur le serveur.""" + return self.serv.get_nickname() + nick = property(_getnick) + def give_me_my_pseudo(self, serv): """Récupère le pseudo auprès de NickServ.""" serv.privmsg("NickServ", "RECOVER %s %s" % (config.irc_pseudo, config.irc_password)) @@ -243,6 +264,73 @@ class Basile(ircbot.SingleServerIRCBot): time.sleep(0.3) serv.nick(config.irc_pseudo) + def pourmoi(self, serv, message): + """Renvoie (False, lemessage) ou (True, le message amputé de "pseudo: ")""" + pseudo = self.nick + pseudo = pseudo.decode("utf-8") + size = len(pseudo) + if message[:size] == pseudo and len(message) > size and message[size] == ":": + return (True, message[size+1:].lstrip(" ")) + else: + return (False, message) + + ### Exécution d'actions + def lost(self, serv, channel, forced=False): + """Réaction à un trigger de perdu. + Annonce "J'ai perdu" sur le channel si on n'a pas perdu depuis un certain temps.""" + if self.last_perdu + config.time_between_perdu < time.time() or forced: + if not channel in self.quiet_channels or forced: + serv.privmsg(channel, "J'ai perdu !") + self.last_perdu=time.time() + delay=config.time_between_perdu_trigger + delta=config.time_between_perdu_trigger_delta + serv.execute_delayed(random.randrange(delay-delta,delay+delta),self.lost,(serv,channel)) + + def quitter(self, chan, leave_message=None): + """Quitter un channel avec un message customisable.""" + if leave_message == None: + leave_message = random.choice(config.leave_messages) + self.serv.part(chan, message=leave_message.encode("utf8")) + + def mourir(self): + """Se déconnecter du serveur IRC avec un message customisable.""" + quit_message = random.choice(config.quit_messages) + self.die(msg=quit_message.encode("utf8")) + + def execute_reload(self, auteur=None): + """Recharge la config.""" + reload(config) + regexp_compile() + 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) + return True, None + else: + return True, u"Config reloaded" + + def crash(self, chan="nowhere", who="nobody"): + """Fait crasher le bot.""" + where = "en privé" if chan == "priv" else "sur le chan %s" % chan + raise CrashError((u"Crash demandé par %s %s" % (who, where)).encode("utf-8")) + + ACTIONS = { + "reload" : execute_reload, + } + + def execute_something(self, something, params, place=None, auteur=None): + """Exécute une action et répond son résultat à ``auteur`` + sur un chan ou en privé en fonction de ``place``""" + action = self.ACTIONS[something] + success, message = action(self, **params) + if message: + if irclib.is_channel(place): + message = "%s: %s" % (auteur, message.encode("utf-8")) + self.serv.privmsg(place, message) + log(self.serveur, place, auteur, something + "%r" % params + ("[successful]" if success else "[failed]")) + + ### Surcharge des events du Bot def on_welcome(self, serv, ev): """À l'arrivée sur le serveur.""" self.serv = serv # ça serv ira :) @@ -260,29 +348,10 @@ class Basile(ircbot.SingleServerIRCBot): for report in self.report_bugs_to: serv.privmsg(report, "Connection to NK2015 failed, invalid password ?") - def lost(self, serv, channel, forced=False): - """Réaction à un trigger de perdu. - Annonce "J'ai perdu" sur le channel si on n'a pas perdu depuis un certain temps.""" - if self.last_perdu + config.time_between_perdu < time.time() or forced: - if not channel in self.quiet_channels or forced: - serv.privmsg(channel, "J'ai perdu !") - self.last_perdu=time.time() - delay=config.time_between_perdu_trigger - delta=config.time_between_perdu_trigger_delta - serv.execute_delayed(random.randrange(delay-delta,delay+delta),self.lost,(serv,channel)) - - def pourmoi(self, serv, message): - """Renvoie (False, lemessage) ou (True, le message amputé de "pseudo: ")""" - pseudo = self.nick - pseudo = pseudo.decode("utf-8") - size = len(pseudo) - if message[:size] == pseudo and len(message) > size and message[size] == ":": - return (True, message[size+1:].lstrip(" ")) - else: - return (False, message) - def on_privmsg(self, serv, ev): """À la réception d'un message en privé.""" + if ignore_event(serv, ev): + return message = ev.arguments()[0] auteur = irclib.nm_to_n(ev.source()) try: @@ -372,7 +441,7 @@ class Basile(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: - self.quitter(message[1], " ".join(message[2:])) + self.quitter(message[1].encode("utf-8"), " ".join(message[2:])) self.chanlist.remove(message[1]) log(self.serveur, "priv", auteur, " ".join(message) + "[successful]") else: @@ -422,8 +491,7 @@ class Basile(ircbot.SingleServerIRCBot): notunderstood = True elif cmd == u"reload": if auteur in self.ops: - self.reload(auteur) - log(self.serveur, "priv", auteur, " ".join(message) + "[successful]") + self.execute_something("reload", {"auteur" : auteur}, place=auteur, auteur=auteur) else: notunderstood = True elif cmd == u"reconnect": @@ -472,7 +540,7 @@ class Basile(ircbot.SingleServerIRCBot): notunderstood = True elif cmd == u"say": if auteur in self.overops and len(message) > 2: - serv.privmsg(message[1], " ".join(message[2:])) + serv.privmsg(message[1].encode("utf-8"), (u" ".join(message[2:])).encode("utf-8")) log(self.serveur, "priv", auteur, " ".join(message)) elif len(message) <= 2: serv.privmsg(auteur, "Syntaxe : SAY ") @@ -488,7 +556,7 @@ class Basile(ircbot.SingleServerIRCBot): notunderstood = True elif cmd == u"kick": if auteur in self.overops and len(message) > 2: - serv.kick(message[1], message[2], " ".join(message[3:])) + serv.kick(message[1].encode("utf-8"), message[2].encode("utf-8"), " ".join(message[3:]).encode("utf-8")) log(self.serveur, "priv", auteur, " ".join(message)) elif len(message) <= 2: serv.privmsg(auteur, "Syntaxe : KICK []") @@ -507,8 +575,8 @@ class Basile(ircbot.SingleServerIRCBot): if self.identities.has_key(auteur): success, solde, pseudo = nk.get_solde(self.nk, self.identities[auteur]["idbde"], serv, auteur) if success: - serv.privmsg(auteur, "%s (%s)" % (float(solde)/100, pseudo.encode("utf8"))) - log(self.serveur, "priv", auteur, " ".join(message) + "[successful]" if success else "[failed]") + serv.privmsg(auteur, "%.2f (%s)" % (solde/100.0, pseudo.encode("utf8"))) + log(self.serveur, "priv", auteur, " ".join(message) + ("[successful]" if success else "[failed]")) else: serv.privmsg(canal, "Je ne connais pas ton pseudo note.") elif cmd == u"ops": @@ -528,6 +596,8 @@ class Basile(ircbot.SingleServerIRCBot): def on_pubmsg(self, serv, ev): """À la réception d'un message sur un channel.""" + if ignore_event(serv, ev): + return auteur = irclib.nm_to_n(ev.source()) canal = ev.target() message = ev.arguments()[0] @@ -553,8 +623,7 @@ class Basile(ircbot.SingleServerIRCBot): log(self.serveur, canal, auteur, message + "[failed]") elif cmd == u"reload": if auteur in self.ops: - log(self.serveur, canal, auteur, message + "[successful]") - self.reload(canal) + self.execute_something("reload", {"auteur" : auteur}, place=canal, auteur=auteur) elif cmd == u"crash": if auteur in self.overops: self.crash(auteur, message) @@ -600,20 +669,23 @@ class Basile(ircbot.SingleServerIRCBot): elif cmd in [u"ping"] and not canal in self.quiet_channels: serv.privmsg(canal, "%s: pong" % (auteur)) - elif cmd in [u"solde", u"!solde"]: + elif cmd in [u"solde", u"!solde", u"!coca"] or cmd.startswith("!"): if self.identities.has_key(auteur): idbde = self.identities[auteur]["idbde"] - success, solde, pseudo = nk.get_solde(self.nk, self.identities[auteur]["idbde"], serv, canal) - if success: - serv.privmsg(canal, "%s: %s (%s)" % (auteur, float(solde)/100, pseudo.encode("utf8"))) - log(self.serveur, canal, auteur, message + "[successful]" if success else "[failed]") + if cmd in [u"solde", u"!solde"]: + success, solde, pseudo = nk.get_solde(self.nk, self.identities[auteur]["idbde"], serv, canal) + if success: + serv.privmsg(canal, "%s: %s (%s)" % (auteur, float(solde)/100, pseudo.encode("utf8"))) + elif cmd in [u"!coca"] or cmd.startswith("!"): + success = nk.consomme(self.nk, self.identities[auteur]["idbde"], message[1:], serv, canal) + log(self.serveur, canal, auteur, message + ("[successful]" if success else "[failed]")) else: serv.privmsg(canal, "%s: Je ne connais pas votre pseudo note." % (auteur)) log(self.serveur, canal, auteur, message + "[unknown]") - elif (re.match("!?(pain au chocolat|chocolatine)", message.lower()) + elif (re.match("(pain au chocolat|chocolatine)", message.lower()) and not canal in self.quiet_channels): serv.action(canal, "sert un pain au chocolat à %s" % (auteur)) - elif re.match("!?manzana",message.lower()) and not canal in self.quiet_channels: + elif re.match("manzana",message.lower()) and not canal in self.quiet_channels: if auteur in config.manzana: serv.action(canal, "sert une bouteille de manzana à %s" % (auteur)) elif auteur in config.manzana_bis: @@ -739,6 +811,8 @@ class Basile(ircbot.SingleServerIRCBot): def on_action(self, serv, ev): """À la réception d'une action.""" + if ignore_event(serv, ev): + return action = ev.arguments()[0] auteur = irclib.nm_to_n(ev.source()) channel = ev.target() @@ -774,7 +848,7 @@ class Basile(ircbot.SingleServerIRCBot): 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)) + log(self.serveur, u"%s kické de %s par %s (raison : %s)" % (victime, channel.decode("utf-8"), auteur, raison)) time.sleep(2) serv.join(channel) l1, l2 = config.kick_answers, config.kick_actions @@ -785,39 +859,7 @@ class Basile(ircbot.SingleServerIRCBot): else: serv.privmsg(channel, l1[i].format(auteur).encode("utf8")) - def quitter(self, chan, leave_message=None): - """Quitter un channel avec un message customisable.""" - if leave_message == None: - leave_message = random.choice(config.leave_messages) - self.serv.part(chan, message=leave_message.encode("utf8")) - - def mourir(self): - """Se déconnecter du serveur IRC avec un message customisable.""" - quit_message = random.choice(config.quit_messages) - self.die(msg=quit_message.encode("utf8")) - - def _getnick(self): - """Récuère le nick effectif du bot sur le serveur.""" - return self.serv.get_nickname() - nick = property(_getnick) - - def reload(self, auteur=None): - """Recharge la config.""" - reload(config) - regexp_compile() - 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 crash(self, chan="nowhere", who="nobody"): - """Fait crasher le bot.""" - where = "en privé" if chan == "priv" else "sur le chan %s" % chan - raise CrashError("Crash demandé par %s %s" % (who, where)) - + ### .fork trick def start_as_daemon(self, outfile): sys.stderr = Logger(outfile) self.start() @@ -874,7 +916,7 @@ def main(): basile = Basile(serveur,debug) # Si on reçoit un SIGHUP, on reload la config def sighup_handler(signum, frame): - basile.reload("SIGHUP") + basile.execute_reload(auteur="SIGHUP") signal.signal(signal.SIGHUP, sighup_handler) # Daemonization if daemon: