X-Git-Url: http://gitweb.pimeys.fr/?a=blobdiff_plain;f=parrot.py;h=6229e8d4a5c4189d82a0a6945fe5d8c3a52e4e50;hb=9cdf643cd7cb6713b470f54a09f963816101fca0;hp=9d0927412e99cca6d6e4eaff98f3d602ebecffd0;hpb=9ff959bfabe1a0e9db2d4e0c44dd907b7edde4b3;p=bots%2Fparrot.git

diff --git a/parrot.py b/parrot.py
index 9d09274..6229e8d 100755
--- a/parrot.py
+++ b/parrot.py
@@ -26,7 +26,8 @@ import errors
 #: Module de gestion des quotes
 import quotes
 
-quote_pattern = re.compile(config.quote_regexp, flags=re.UNICODE)
+# Je veux pouvoir éditer ce que crée ce bot
+os.umask(002)
 
 def get_config_logfile(serveur):
     """Renvoie le nom du fichier de log en fonction du ``serveur`` et de la config."""
@@ -161,11 +162,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):
@@ -349,6 +363,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:
@@ -400,9 +418,15 @@ 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)
+            elif cmd in [u"display"]:
+                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 = quote_pattern.match(message)
+            match = self.quote_pattern.match(message)
             if match:
                 d = match.groupdict()
                 if self.quotedb.store(d["author"], d["content"]):
@@ -422,6 +446,18 @@ class Parrot(ircbot.SingleServerIRCBot):
                         serv.privmsg(canal, (u"Pas de quote de %s en mémoire." % author).encode("utf-8"))
                         return
                     serv.privmsg(canal, str(q))
+            elif message.startswith(u"!author"):
+                words = message.split()
+                cmd = words[0].lstrip("!")
+                regexp = cmd in ["author-r", "authorreg", "author-reg", "author-regex"]
+                search = u" ".join(words[1:])
+                authors = self.quotedb.search_authors(search, regexp)
+                if len(authors) > config.search_max_authors:
+                    authors = authors[:config.search_max_authors+1] + [unicode(len(authors) - config.search_max_authors)]
+                if authors:
+                    serv.privmsg(canal, "%s: %s" % (auteur, (u" ".join(authors)).encode("utf-8")))
+                else:
+                    serv.privmsg(canal, "%s: Pas d'auteur correspondant à la recherche.")
 
     def on_action(self, serv, ev):
         """À la réception d'une action."""