X-Git-Url: http://gitweb.pimeys.fr/?p=bots%2Fparrot.git;a=blobdiff_plain;f=parrot.py;h=5b766d09a108659dff26dbc2aaa1727a69c25f77;hp=4ee3b23a6692399a93c3258ffa15aa7b3141628b;hb=HEAD;hpb=5dc10fbec825b8bd297a6cf6f3065242963968f0 diff --git a/parrot.py b/parrot.py index 4ee3b23..5b766d0 100755 --- a/parrot.py +++ b/parrot.py @@ -26,7 +26,8 @@ import errors #: Module de gestion des quotes import quotes -quote_pattern = re.compile(config.quote_regexp) +# Je veux pouvoir éditer ce que crée ce bot +os.umask(002) def get_config_logfile(serveur): """Renvoie le nom du fichier de log en fonction du ``serveur`` et de la config.""" @@ -63,7 +64,7 @@ def bot_unicode(chain): """Essaye de décoder ``chain`` en UTF-8. Lève une py:class:`errors.UnicodeBotError` en cas d'échec.""" try: - return chain.decode("utf8") + return chain.decode("utf-8") except UnicodeDecodeError as exc: raise errors.UnicodeBotError @@ -85,7 +86,8 @@ class Parrot(ircbot.SingleServerIRCBot): self.last_perdu = 0 self.quotedb = quotes.QuoteDB() - self.quotedb.load() + self.reload_quotes() + ### Utilitaires def _getnick(self): @@ -96,7 +98,6 @@ class Parrot(ircbot.SingleServerIRCBot): 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)) - serv.privmsg("NickServ", "RELEASE %s %s" % (config.irc_pseudo, config.irc_password)) time.sleep(0.3) serv.nick(config.irc_pseudo) @@ -115,17 +116,22 @@ class Parrot(ircbot.SingleServerIRCBot): """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")) + self.serv.part(chan, message=leave_message.encode("utf-8")) 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")) + self.die(msg=quit_message.encode("utf-8")) + def reload_quotes(self): + """ Recharge la base de données des quotes et recompile la regexp de quote """ + self.quotedb.load() + self.quote_pattern = re.compile(config.quote_regexp, flags=re.UNICODE) + def execute_reload(self, auteur=None): """Recharge la config.""" reload(config) - isit.regexp_compile() + self.reload_quotes() if auteur in [None, "SIGHUP"]: towrite = "Config reloaded" + " (SIGHUP received)" * (auteur == "SIGHUP") for to in config.report_bugs_to: @@ -155,11 +161,27 @@ class Parrot(ircbot.SingleServerIRCBot): self.serv.privmsg(place, message) log(self.serveur, place, auteur, something + "%r" % params + ("[successful]" if success else "[failed]")) - ### Gestion des quotes - def enregistrer(self, author, content, timestamp): - - quotes.save(self.quotelist, config.quote_file) + def acknowledge(self, asked_by, asked_where, message): + """Répond quelque chose au demandeur d'une action. + ``asked_where=None`` signifie en privé.""" + if asked_where is None: + self.serv.privmsg(asked_by, message) + else: + self.serv.privmsg(asked_where, "%s: %s" % (asked_by, message)) + + def dump(self, asked_by, asked_where=None): + """Dumpe les quotes. ``asked_where=None`` signifie en privé.""" + quotes.dump(self.quotedb) + self.acknowledge(asked_by, asked_where, "Quotes dumpées") + + def restore(self, asked_by, asked_where=None): + """Restaure les quotes à partir du dump. ``asked_where=None`` signifie en privé.""" + self.quotedb = quotes.restore() + self.acknowledge(asked_by, asked_where, "Quotes restaurées à partir du dump (pas de backup effectué).") + many = self.quotedb.get_clash_authors() + if many: + self.acknowledge(asked_by, asked_where, "Auteurs de casse différente : %s" % (many)) ### Surcharge des events du Bot def on_welcome(self, serv, ev): @@ -184,7 +206,7 @@ class Parrot(ircbot.SingleServerIRCBot): message = bot_unicode(message) except errors.UnicodeBotError: if config.utf8_trigger: - serv.privmsg(auteur, random.choice(config.utf8_fail_answers).encode("utf8")) + serv.privmsg(auteur, random.choice(config.utf8_fail_answers).encode("utf-8")) return message = message.split() cmd = message[0].lower() @@ -343,6 +365,10 @@ class Parrot(ircbot.SingleServerIRCBot): serv.privmsg(auteur, " ".join(self.overops)) else: notunderstood = True + elif cmd == u"dump" and auteur in self.ops: + self.dump(asked_by=auteur) + elif cmd == u"restore" and auteur in self.overops: + self.restore(asked_by=auteur) else: notunderstood = True if notunderstood: @@ -359,7 +385,7 @@ class Parrot(ircbot.SingleServerIRCBot): message = bot_unicode(message) except errors.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")) + serv.privmsg(canal, (u"%s: %s"% ( auteur, random.choice(config.utf8_fail_answers))).encode("utf-8")) return pour_moi, message = self.pourmoi(serv, message) if pour_moi and message.split()!=[]: @@ -373,7 +399,7 @@ class Parrot(ircbot.SingleServerIRCBot): log(self.serveur, canal, auteur, message + "[successful]") self.mourir() else: - serv.privmsg(canal,(u"%s: %s"%(auteur, random.choice(config.quit_fail_messages))).encode("utf8")) + serv.privmsg(canal,(u"%s: %s"%(auteur, random.choice(config.quit_fail_messages))).encode("utf-8")) log(self.serveur, canal, auteur, message + "[failed]") elif cmd == u"reload": if auteur in self.ops: @@ -389,25 +415,40 @@ class Parrot(ircbot.SingleServerIRCBot): if canal in self.chanlist: self.chanlist.remove(canal) else: - serv.privmsg(canal,(u"%s: %s" % (auteur, random.choice(config.leave_fail_messages))).encode("utf8")) + serv.privmsg(canal,(u"%s: %s" % (auteur, random.choice(config.leave_fail_messages))).encode("utf-8")) log(self.serveur, canal, auteur, message + "[failed]") elif cmd in [u"ping"] and not canal in self.quiet_channels: serv.privmsg(canal, "%s: pong" % (auteur)) + elif cmd in [u"dump"]: + self.dump(asked_by=auteur, asked_where=canal) + elif cmd in [u"restore"] and auteur in self.overops: + self.restore(asked_by=auteur, asked_where=canal) + elif cmd in [u"display", u"link", u"url"]: + self.serv.privmsg(canal, "%s: %s" % (auteur, config.quote_display_url.encode("utf-8"))) else: # Vu que ce bot est prévu pour parser des quotes il va falloir bosser ici - match = quote_pattern.match(message) + match = self.quote_pattern.match(message) if match: d = match.groupdict() - if self.quotedb.store(d["author"], d["content"]): + # On n'autorise pas les gens à déclarer le quoter + d["quoter"] = auteur.decode("utf-8") + if self.quotedb.store(**d): serv.privmsg(canal, (u"%s: Ce sera retenu, répété, amplifié" % (auteur,)).encode("utf-8")) self.quotedb.save() else: serv.privmsg(canal, (u"%s: Je le savais déjà." % (auteur,)).encode("utf-8")) + # Whou, attention, hack dégueu + # on enlève context- au début des !commands si il y est, + # et on passe à True le paramètre show_context pour s'en souvenir + show_context = False + if message.startswith(u"!context-"): + show_context = True + message = u"!" + message[9:] if message.startswith(u"!quote"): if message.strip() == u"!quote": q = self.quotedb.random() - serv.privmsg(canal, str(q)) + serv.privmsg(canal, q.display(show_context)) elif message.startswith("!quote "): author = message[7:].strip() try: @@ -415,7 +456,40 @@ class Parrot(ircbot.SingleServerIRCBot): except IndexError: serv.privmsg(canal, (u"Pas de quote de %s en mémoire." % author).encode("utf-8")) return - serv.privmsg(canal, str(q)) + serv.privmsg(canal, q.display(show_context)) + elif message.startswith(u"!author") or message.startswith(u"!from"): + words = message.split() + cmd = words[0].lstrip("!") + regexp = any([cmd.endswith(suffix) for suffix in config.regex_suffixes]) + search = u" ".join(words[1:]) + authors = self.quotedb.search_authors(search, regexp) + if not authors: + serv.privmsg(canal, "%s: Pas d'auteur correspondant à la recherche." % (auteur,)) + return + if cmd.startswith("author"): + if len(authors) > config.search_max_authors: + authors = authors[:config.search_max_authors+1] + ["+%s" % (len(authors) - config.search_max_authors)] + serv.privmsg(canal, "%s: %s" % (auteur, (u", ".join(authors)).encode("utf-8"))) + elif cmd.startswith("from"): + quotes = sum([self.quotedb.quotesfrom(a) for a in authors], []) + q = random.choice(quotes) + serv.privmsg(canal, q.display(show_context)) + elif message.startswith(u"!search"): + words = message.split() + cmd = words[0].lstrip("!") + regexp = cmd in ["search" + suffix for suffix in config.regex_suffixes] + search = u" ".join(words[1:]) + quotes = self.quotedb.search(inquote=search, regexp=regexp) + # On recherche également sur le contexte si on est en !context-search + if show_context: + quotes += self.quotedb.search(place=search, regexp=regexp) + # Pour pas biaiser le choix aléatoire, on enlève les doublons + quotes = list(set(quotes)) + if quotes: + q = random.choice(quotes) + serv.privmsg(canal, q.display(show_context)) + else: + serv.privmsg(canal, "%s: Pas de quotes correspondant à la recherche." % (auteur,)) def on_action(self, serv, ev): """À la réception d'une action.""" @@ -428,7 +502,7 @@ class Parrot(ircbot.SingleServerIRCBot): action = bot_unicode(action) except errors.UnicodeBotError: 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")) + serv.privmsg(channel, (u"%s: %s"%(auteur,random.choice(config.utf8_fail_answers))).encode("utf-8")) return mypseudo = self.nick @@ -439,7 +513,7 @@ class Parrot(ircbot.SingleServerIRCBot): victime = ev.arguments()[0] raison = ev.arguments()[1] if victime == self.nick: - log(self.serveur, u"%s kické de %s par %s (raison : %s)" % (victime, channel.decode("utf-8"), auteur, raison)) + log(self.serveur, ("%s kické de %s par %s (raison : %s)" % (victime, channel, auteur, raison)).decode("utf-8")) time.sleep(2) serv.join(channel) @@ -508,7 +582,7 @@ def main(): parrot.start_as_daemon(outfile) else: # on enregistre le pid de parrot - pidfile = "/var/run/bots/parror.pid" + pidfile = "/var/run/bots/parrot.pid" for arg in sys.argv: arg = arg.split("=") if arg[0].strip('-') in ["pidfile"]: