X-Git-Url: http://gitweb.pimeys.fr/?p=bots%2Fparrot.git;a=blobdiff_plain;f=parrot.py;h=5b766d09a108659dff26dbc2aaa1727a69c25f77;hp=1072896cb8f98b687d94283d0b8cf66864f670b6;hb=HEAD;hpb=51b7d8da6804bd5018f5ecd169875fad92084ee6 diff --git a/parrot.py b/parrot.py index 1072896..5b766d0 100755 --- a/parrot.py +++ b/parrot.py @@ -64,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 @@ -97,8 +97,7 @@ class Parrot(ircbot.SingleServerIRCBot): def give_me_my_pseudo(self, serv): """Récupère le pseudo auprès de NickServ.""" - serv.privmsg("NickServ", "GHOST %s %s" % (config.irc_pseudo, config.irc_password)) - serv.privmsg("NickServ", "RELEASE %s %s" % (config.irc_pseudo, config.irc_password)) + serv.privmsg("NickServ", "RECOVER %s %s" % (config.irc_pseudo, config.irc_password)) time.sleep(0.3) serv.nick(config.irc_pseudo) @@ -117,12 +116,12 @@ 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 """ @@ -207,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() @@ -386,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()!=[]: @@ -400,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: @@ -416,7 +415,7 @@ 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: @@ -425,22 +424,31 @@ class Parrot(ircbot.SingleServerIRCBot): 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"]: + 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 = 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: @@ -448,7 +456,7 @@ 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("!") @@ -465,16 +473,21 @@ class Parrot(ircbot.SingleServerIRCBot): elif cmd.startswith("from"): quotes = sum([self.quotedb.quotesfrom(a) for a in authors], []) q = random.choice(quotes) - serv.privmsg(canal, str(q)) + 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, str(q)) + serv.privmsg(canal, q.display(show_context)) else: serv.privmsg(canal, "%s: Pas de quotes correspondant à la recherche." % (auteur,)) @@ -489,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 @@ -500,7 +513,7 @@ class Parrot(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, ("%s kické de %s par %s (raison : %s)" % (victime, channel, auteur, raison)).decode("utf-8")) time.sleep(2) serv.join(channel)