]> gitweb.pimeys.fr Git - bots/parrot.git/blobdiff - parrot.py
Changement d'ops
[bots/parrot.git] / parrot.py
index dd5053023f4d0bb2ccd0943077de43bd5ad78e28..5b766d09a108659dff26dbc2aaa1727a69c25f77 100755 (executable)
--- 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)