On fait le ménage en dumpant pas des espaces

[bots/parrot.git] / quotes.py

diff --git a/quotes.py b/quotes.py
index 2ff299e65a19b779e9b2db4f402c8021d4fdcdf6..5c1a58ebbd4330230334cb91c821a4296dda9a1a 100644 (file)
--- a/quotes.py
+++ b/quotes.py
@@ -16,19 +16,19 @@ quote_matcher_with_timestamp = re.compile(config.quote_regexp_with_timestamp, fl
 spaces_matcher = re.compile(u"\s", flags=re.U)
 
 def equivalence_partition(iterable, relation):
-       """ Partitionne l'itérable en classes d'équivalences. """
-       classes = []
-       for o in iterable:
-               # find the class it is in
-               found = False
-               for c in classes:
-                       if relation( iter(c).next(), o ): # is it equivalent to this class?
-                               c.add( o )
-                               found = True
-                               break
-               if not found: # it is in a new class
-                       classes.append( set( [ o ] ) )
-       return classes
+    """ Partitionne l'itérable en classes d'équivalences. """
+    classes = []
+    for o in iterable:
+        # find the class it is in
+        found = False
+        for c in classes:
+            if relation( iter(c).next(), o ): # is it equivalent to this class?
+                c.add( o )
+                found = True
+                break
+        if not found: # it is in a new class
+            classes.append( set( [ o ] ) )
+    return classes
 
 def get_now():
     """ Renvoie la date actuelle """
@@ -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,18 +48,32 @@ 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):
         """ Retourne la quote affichable """
-        return config.quote_template % self.__dict__
+        if self.place and self.place.strip() != u"":
+            return config.quote_template_withcontext % self.__dict__
+        else:
+            return config.quote_template % self.__dict__
     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 and self.place.strip() != u"" else u""
+        quoter = self.quoter if self.quoter and self.quoter.strip() != u"" else u""
+        return (u"%s %s | %s | %s" % (self.timestamp.strftime("%F_%T"), config.quote_template % self.__dict__
+                  , 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. """
@@ -113,19 +127,20 @@ class QuoteDB(object):
             return author
     
     def get_clash_authors(self):
-        """ Renvoie une liste de liste d'auteurs qui sont enresgitrés avec des casses différentes. """
+        """ Renvoie une liste de liste d'auteurs qui sont enregistrés avec des casses différentes. """
         authors = list(set([q.author for q in self.quotelist]))
         authors = equivalence_partition(authors, lambda x,y: x.lower() == y.lower())
         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)
             return True
         return False
@@ -136,6 +151,7 @@ class QuoteDB(object):
     def random(self):
         """ Sort une quote aléatoire """
         return random.choice(self.quotelist)
+
     def quotesfrom(self, author):
         """ Sort toutes les quotes de ``author`` """
         return [q for q in self.quotelist if q.author == author]
@@ -178,7 +194,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)