X-Git-Url: http://gitweb.pimeys.fr/?p=bots%2Fparrot.git;a=blobdiff_plain;f=parrot.py;h=5b766d09a108659dff26dbc2aaa1727a69c25f77;hp=dd5053023f4d0bb2ccd0943077de43bd5ad78e28;hb=HEAD;hpb=ce4fddd82c5d8f1ee1ed3e919e0e7e8b2c689fd1 diff --git a/parrot.py b/parrot.py index dd50530..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 @@ -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 @@ -100,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) @@ -120,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 """ @@ -210,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() @@ -389,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()!=[]: @@ -403,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: @@ -419,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: @@ -428,7 +424,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 +438,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 +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("!") @@ -470,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,)) @@ -494,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 @@ -505,16 +513,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)