!search pour avoir une quote contenant un terme de recherche

[bots/parrot.git] / parrot.py

diff --git a/parrot.py b/parrot.py
index 6229e8d4a5c4189d82a0a6945fe5d8c3a52e4e50..1f5b70def7f3b3aa91f987595253692ce024fdd7 100755 (executable)
--- a/parrot.py
+++ b/parrot.py
@@ -449,15 +449,26 @@ class Parrot(ircbot.SingleServerIRCBot):
             elif message.startswith(u"!author"):
                 words = message.split()
                 cmd = words[0].lstrip("!")
-                regexp = cmd in ["author-r", "authorreg", "author-reg", "author-regex"]
+                regexp = cmd in ["author" + suffix for suffix in config.regex_suffixes]
                 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)]
+                    authors = authors[:config.search_max_authors+1] + ["+%s" % (len(authors) - config.search_max_authors)]
                 if authors:
-                    serv.privmsg(canal, "%s: %s" % (auteur, (u" ".join(authors)).encode("utf-8")))
+                    serv.privmsg(canal, "%s: %s" % (auteur, (u", ".join(authors)).encode("utf-8")))
                 else:
-                    serv.privmsg(canal, "%s: Pas d'auteur correspondant à la recherche.")
+                    serv.privmsg(canal, "%s: Pas d'auteur correspondant à la recherche." % (auteur,))
+            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)
+                if quotes:
+                    q = random.choice(quotes)
+                    serv.privmsg(canal, str(q))
+                else:
+                    serv.privmsg(canal, "%s: Pas de quotes correspondant à la recherche." % (auteur,))
 
     def on_action(self, serv, ev):
         """À la réception d'une action."""