X-Git-Url: http://gitweb.pimeys.fr/?a=blobdiff_plain;f=parrot.py;h=17a24a0264c17758dfdd6ee9aff83a8a867678c3;hb=b9063a8c634b789e417f2fde616294ac3849377c;hp=e58b04d97ffb8ce9b0c2234d465516f5c6a6261c;hpb=f7e96ecef213b35fa0c1e9e4078088d1e30d2641;p=bots%2Fparrot.git

diff --git a/parrot.py b/parrot.py
index e58b04d..17a24a0 100755
--- a/parrot.py
+++ b/parrot.py
@@ -26,7 +26,7 @@ import errors
 #: Module de gestion des quotes
 import quotes
 
-quote_pattern = re.compile(config.quote_regexp)
+quote_pattern = re.compile(config.quote_regexp, flags=re.UNICODE)
 
 def get_config_logfile(serveur):
     """Renvoie le nom du fichier de log en fonction du ``serveur`` et de la config."""
@@ -85,7 +85,8 @@ class Parrot(ircbot.SingleServerIRCBot):
         self.last_perdu = 0
         
         self.quotedb = quotes.QuoteDB()
-        self.quotedb.load()
+        self.reload_quotes()
+
     
     ### Utilitaires
     def _getnick(self):
@@ -95,7 +96,7 @@ 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", "GHOST %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)
@@ -122,10 +123,15 @@ class Parrot(ircbot.SingleServerIRCBot):
         quit_message = random.choice(config.quit_messages)
         self.die(msg=quit_message.encode("utf8"))
     
+    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,24 @@ 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é).")
     
     ### Surcharge des events du Bot
     def on_welcome(self, serv, ev):
@@ -343,6 +362,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:
@@ -394,6 +417,10 @@ class Parrot(ircbot.SingleServerIRCBot):
     
             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)
         else:
             # Vu que ce bot est prévu pour parser des quotes il va falloir bosser ici
             match = quote_pattern.match(message)
@@ -439,7 +466,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))
             time.sleep(2)
             serv.join(channel)