X-Git-Url: http://gitweb.pimeys.fr/?p=bots%2Fbasile.git;a=blobdiff_plain;f=basile.py;h=4ee806783efa2306f1f0afc738f7c3ad7e3bdc46;hp=797717a905bfd733a0083356d39b6a00226ae9fe;hb=4d6cedfb164059e9c5c607db4752533ae7333b44;hpb=bf0ca5fa90df6a7c1c094c904ec824700dff4787 diff --git a/basile.py b/basile.py index 797717a..4ee8067 100755 --- a/basile.py +++ b/basile.py @@ -54,6 +54,17 @@ def log(serveur, channel, auteur=None, message=None): if config.debug_stdout: print chain +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)) + return blackit and not exceptit + def regex_join(liste, avant=u".*(?:^| )", apres=u"(?:$|\.| |,|;).*"): """Fabrique une regexp à partir d'une liste d'éléments à matcher.""" return avant + u"(" + u"|".join(liste) + u")" + apres @@ -205,37 +216,44 @@ 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)""" try: login_result, sock = nk.login(username, password, typ) - droits, retcode, errmsg = login_result["msg"], login_result["retcode"], login_result["errmsg"] + info, retcode, errmsg = login_result["msg"], login_result["retcode"], login_result["errmsg"] except nk.NKRefused as exc: for report in self.report_bugs_to: serv.privmsg(report, "Le Serveur NK2015 est down.") - return (False, None) + return (False, None, None) except nk.NKHelloFailed as exc: for report in self.report_bugs_to: serv.privmsg(report, "La version du protocole utilisée n'est pas supportée par le serveur NK2015.") - return (False, None) + return (False, None, None) except nk.NKUnknownError as exc: erreurs = ["Une fucking erreur inconnue s'est produite"] erreurs += str(exc).split("\n") for report in self.report_bugs_to: for err in erreurs: serv.privmsg(report, err) - return (False, None) + return (False, None, None) except Exception as exc: # Exception qui ne vient pas de la communication avec le serveur NK2015 log(self.serveur, "Erreur dans new_connection_NK\n" + str(exc)) - return (False, None) + return (False, None, None) if retcode == 0: - return (True, sock) + return (True, info, sock) else: - return (False, None) - + 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 +261,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("Crash demandé par %s %s" % (who, where)) + + 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 :) @@ -255,32 +340,11 @@ class Basile(ircbot.SingleServerIRCBot): log(self.serveur, "JOIN %s" % (c)) serv.join(c) # on ouvre la connexion note de Basile, special user - self.nk = self.new_connection_NK(serv, config.note_pseudo, config.note_password, "special")[1] + self.nk = self.new_connection_NK(serv, config.note_pseudo, config.note_password, "special")[2] if self.nk == None: 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é.""" message = ev.arguments()[0] @@ -321,16 +385,16 @@ class Basile(ircbot.SingleServerIRCBot): if len(message) == 1: if self.identities.has_key(auteur): serv.privmsg(auteur, "Je vous connais sous le pseudo note %s." % ( - self.identities[auteur].encode("utf8"))) + self.identities[auteur]["pseudo"].encode("utf8"))) else: serv.privmsg(auteur, "Je ne connais pas votre pseudo note.") elif len(message) >= 3: username, password = message[1], " ".join(message[2:]) - success, _ = self.new_connection_NK(serv, username, password) + success, info, _ = self.new_connection_NK(serv, username, password) if success: log(self.serveur, "priv", auteur, " ".join(message) + "[successful]") serv.privmsg(auteur, "Identité enregistrée.") - self.identities[auteur] = username + self.identities[auteur] = info json.dump(self.identities, open(config.identities_file,"w")) else: log(self.serveur, "priv", auteur, " ".join(message) + "[failed]") @@ -341,7 +405,7 @@ class Basile(ircbot.SingleServerIRCBot): if len(message) > 1: if self.identities.has_key(auteur): password = " ".join(message[1:]) - success, _ = self.new_connection_NK(serv, self.identities[auteur], password) + success, _, _ = self.new_connection_NK(serv, self.identities[auteur], password) if success: del self.identities[auteur] log(self.serveur, "priv", auteur, " ".join(message) + "[successful]") @@ -372,7 +436,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,15 +486,14 @@ 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": if auteur in self.ops: try: self.nk = self.new_connection_NK(serv, config.note_pseudo, - config.note_password, "special")[1] + config.note_password, "special")[2] except Exception as exc: self.nk = None log(self.serveur, 'Erreur dans on_pubmsg/"cmd in ["reconnect"]\n' + str(exc)) @@ -505,31 +568,12 @@ class Basile(ircbot.SingleServerIRCBot): elif cmd == u"solde": if len(message) == 1: if self.identities.has_key(auteur): - try: - self.nk.write('["search", ["x",["pseudo"],%s]]' % (json.dumps(self.identities[auteur]))) - ret = json.loads(self.nk.read()) - solde = ret["msg"][0]["solde"] - pseudo = ret["msg"][0]["pseudo"] - except Exception as exc: - print exc - serv.privmsg(auteur, "failed") - log(self.serveur, "priv", auteur, " ".join(message) + "[failed]") - return - serv.privmsg(auteur, "%s (%s)" % (float(solde)/100, pseudo.encode("utf8"))) - log(self.serveur, "priv", auteur, " ".join(message) + "[successful]") + success, solde, pseudo = nk.get_solde(self.nk, self.identities[auteur]["idbde"], serv, auteur) + if success: + 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 auteur in self.ops: - try: - self.nk.write('["search", ["x",["pseudo"],%s]]' % (json.dumps(message[1]))) - ret = json.loads(self.nk.read()) - solde = ret["msg"][0]["solde"] - pseudo = ret["msg"][0]["pseudo"] - except Exception as exc: - serv.privmsg(auteur, "failed") - log(self.serveur, "priv", auteur, " ".join(message) + "[failed]") - return - serv.privmsg(auteur, "%s (%s)" % (float(solde)/100, pseudo.encode("utf8"))) elif cmd == u"ops": if auteur in self.overops: serv.privmsg(auteur, " ".join(self.ops)) @@ -572,8 +616,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) @@ -592,7 +635,7 @@ class Basile(ircbot.SingleServerIRCBot): if auteur in self.ops: try: self.nk = self.new_connection_NK(serv, config.note_pseudo, - config.note_password, "special")[1] + config.note_password, "special")[2] except Exception as exc: self.nk = None log(self.serveur, 'Erreur dans on_pubmsg/"cmd in ["reconnect"]\n' + str(exc)) @@ -619,20 +662,16 @@ 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): - pseudo = self.identities[auteur] - try: - self.nk.write(json.dumps(["search", ["x", ["pseudo"], pseudo]])) - ret = json.loads(self.nk.read()) - solde = ret["msg"][0]["solde"] - pseudo = ret["msg"][0]["pseudo"] - except Exception as exc: - serv.privmsg(canal, "%s: failed"%(auteur)) - log(self.serveur, canal, auteur, message + "[failed]") - else: - serv.privmsg(canal, "%s: %s (%s)" % (auteur, float(solde)/100, pseudo.encode("utf8"))) - log(self.serveur, canal, auteur, message + "[successful]") + idbde = self.identities[auteur]["idbde"] + 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]") @@ -811,39 +850,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() @@ -900,7 +907,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: