X-Git-Url: http://gitweb.pimeys.fr/?p=bots%2Fparrot.git;a=blobdiff_plain;f=parrot.py;h=e9f4637b626d1d20a99fa6618f6afa095ba948c0;hp=dd5053023f4d0bb2ccd0943077de43bd5ad78e28;hb=f3e614e5165dd316a66377f9a59ccaf301fde425;hpb=ce4fddd82c5d8f1ee1ed3e919e0e7e8b2c689fd1 diff --git a/parrot.py b/parrot.py index dd50530..e9f4637 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 @@ -87,9 +87,6 @@ class Parrot(ircbot.SingleServerIRCBot): self.quotedb = quotes.QuoteDB() self.reload_quotes() - - # Pour agir à la réception des whois 307 = registered nick - self.ircobj.add_global_handler("307", self.on_registered_nick) ### Utilitaires @@ -120,12 +117,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 """ @@ -210,7 +207,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() @@ -389,7 +386,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()!=[]: @@ -403,7 +400,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: @@ -419,7 +416,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: @@ -428,7 +425,7 @@ 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 @@ -442,10 +439,17 @@ class Parrot(ircbot.SingleServerIRCBot): 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: @@ -453,7 +457,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("!") @@ -470,7 +474,7 @@ 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("!") @@ -479,7 +483,7 @@ class Parrot(ircbot.SingleServerIRCBot): quotes = self.quotedb.search(inquote=search, regexp=regexp) 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,)) @@ -494,7 +498,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 @@ -505,16 +509,10 @@ 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) - def on_registered_nick(self, serv, ev): - """À la réception d'un résultat de whois.""" - pseudo, regis = ev.arguments() - if regis == 'is a registered nick': - print "%s est enregistré \o/" % pseudo - ### .fork trick def start_as_daemon(self, outfile): sys.stderr = Logger(outfile)