From: Vincent Le Gallic <legallic@crans.org>
Date: Fri, 18 Apr 2014 02:55:40 +0000 (+0200)
Subject: gestion du lieu et du quoter
X-Git-Url: http://gitweb.pimeys.fr/?p=bots%2Fparrot.git;a=commitdiff_plain;h=e16477da6ca556666492b839f5a79ec84c41018d

gestion du lieu et du quoter
---

diff --git a/config.py b/config.py
index 552f24f..fce38cb 100644
--- a/config.py
+++ b/config.py
@@ -125,8 +125,11 @@ helpmsg_overops = u" SAY DO STAY NOSTAY OPS OVEROPS KICK DIE CRASH RESTORE"
 #: et aucun élément de ``exceptlist``
 blacklisted_masks = [("Flo!*@*", [])]
 
+#: Regexp pour le lieu et le quoter
+quote_optionnal_fields = ur"\|\s*(?P<place>[^|]+?)?\s*\|\s*(?P<quoter>[^|]+?)?\s*"
+
 #: Regexp utilisée pour parser des quotes
-quote_regexp = ur"""\s*(?:"|«|')\s*(?P<content>.+?)\s*(?:"|»|')\s*~\s*(?P<author>.+?)\s*(?:$|\n)"""
+quote_regexp = ur"""\s*(?:"|«|')\s*(?P<content>.+?)\s*(?:"|»|')\s*~\s*(?P<author>[^|]+?)\s*(?:%s)?(?:$|\n)""" % (quote_optionnal_fields,)
 
 #: Regexp utlisée pour parser une quote avec timestamp
 quote_regexp_with_timestamp = ur"(?P<timestamp>.{19}) " + quote_regexp
diff --git a/parrot.py b/parrot.py
index 1072896..0cd5be9 100755
--- a/parrot.py
+++ b/parrot.py
@@ -432,7 +432,9 @@ class Parrot(ircbot.SingleServerIRCBot):
             match = self.quote_pattern.match(message)
             if match:
                 d = match.groupdict()
-                if self.quotedb.store(d["author"], d["content"]):
+                # On n'autorise pas les gens à déclarer le quoter
+                d["quoter"] = auteur.decode("utf-8")
+                if self.quotedb.store(**d):
                     serv.privmsg(canal, (u"%s: Ce sera retenu, répété, amplifié" % (auteur,)).encode("utf-8"))
                     self.quotedb.save()
                 else:
diff --git a/quotes.py b/quotes.py
index 2ff299e..f6e4bc3 100644
--- a/quotes.py
+++ b/quotes.py
@@ -40,7 +40,7 @@ def sanitize_author(raw):
 
 class Quote(object):
     """ Une citation """
-    def __init__(self, author, content, timestamp=None):
+    def __init__(self, author, content, timestamp=None, place=None, quoter=None):
         if timestamp is None:
             timestamp = get_now()
         elif isinstance(timestamp, basestring):
@@ -48,10 +48,13 @@ class Quote(object):
         self.author = sanitize_author(author)
         self.content = content
         self.timestamp = timestamp
+        self.place = place
+        self.quoter = quoter
     
     def jsonize(self):
         d = {"author" : self.author, "content" : self.content,
-             "timestamp" : self.timestamp.strftime(u"%F_%T")}
+             "timestamp" : self.timestamp.strftime(u"%F_%T"),
+             "place" : self.place, "quoter" : self.quoter}
         return d
     
     def __unicode__(self):
@@ -60,6 +63,13 @@ class Quote(object):
     def __str__(self):
         return unicode(self).encode("utf-8")
     
+    def full_str(self):
+        """ Retourne une chaîne représentant la totalité des infos de la quote,
+            tout en étant parsable et human-readable. """
+        place = self.place if self.place else ""
+        quoter = self.quoter if self.quoter else ""
+        return (u"%s %s | %s | %s" % (self.timestamp.strftime("%F_%T"), unicode(self), place, quoter)).encode("utf-8")
+    
     def __eq__(self, otherquote):
         """ Vérifie si cette phrase n'a pas déjà été dite par la même personne.
             Indépendamment de la date et de la casse. """
@@ -119,11 +129,13 @@ class QuoteDB(object):
         authors = [list(c) for c in authors if len(c) > 1]
         return authors
     
-    def store(self, author, content, timestamp=None):
+    def store(self, timestamp=None, **kwargs):
         """ Enregistre une nouvelle quote, sauf si elle existe déjà.
             Force l'auteur à utiliser la même casse si un auteur de casse différente existait déjà.
             Renvoie ``True`` si elle a été ajoutée, ``False`` si elle existait. """
-        newquote = Quote(self._collapse_author(author), content, timestamp)
+        kwargs["author"] = self._collapse_author(kwargs["author"])
+        kwargs["timestamp"] = timestamp
+        newquote = Quote(**kwargs)
         if not newquote in self.quotelist:
             self.search
             self.quotelist.append(newquote)
@@ -178,7 +190,7 @@ def dump(quotedb, dump_file=None):
     """Pour exporter les quotes dans un format readable vers un fichier."""
     if dump_file is None:
         dump_file = config.quote_dump_file
-    t = "\n".join(["%s %s" % (q.timestamp.strftime("%F_%T"), q) for q in quotedb.quotelist]) + "\n"
+    t = "\n".join([q.full_str() for q in quotedb.quotelist]) + "\n"
     with open(dump_file, "w") as f:
         f.write(t)