Regexp avec le flag UNICODE histoire de matcher correctement les espaces insécables

[bots/parrot.git] / quotes.py

diff --git a/quotes.py b/quotes.py
index 73f711c08b5dc6de3143f4d84802f24ad403c27e..1365b1211868bc51a235c4e9f2d50d6aa4cdb93e 100644 (file)
--- a/quotes.py
+++ b/quotes.py
@@ -11,6 +11,9 @@ import random
 
 import config
 
+quote_matcher = re.compile(config.quote_regexp, flags=re.UNICODE)
+quote_matcher_with_timestamp = re.compile(config.quote_regexp_with_timestamp, flags=re.UNICODE)
+
 def get_now():
     """ Renvoie la date actuelle """
     return datetime.datetime(*time.localtime()[:6])
@@ -42,7 +45,6 @@ class Quote(object):
             Indépendamment de la date. """
         return [self.author, self.content] == [otherquote.author, otherquote.content]
 
-quote_matcher = re.compile(config.quote_regexp)
 
 def parse(text, date=None):
     """ Parse le ``text`` et renvoie une quote ou None. """
@@ -97,3 +99,25 @@ class QuoteDB(object):
     def randomfrom(self, author):
         """ Sort une quote aléatoire de ``author`` """
         return random.choice([q for q in self.quotelist if q.author == author])
+
+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"
+    with open(dump_file, "w") as f:
+        f.write(t)
+
+def restore(dump_file=None):
+    """Crée un DB de quotes en parsant le contenu d'un fichier de dump."""
+    if dump_file is None:
+        dump_file = config.quote_dump_file
+    with open(dump_file) as f:
+        t = f.read()
+    t = t.decode("utf-8") # Oui, ça peut fail, mais on ne doit alors pas continuer
+    l = [m.groupdict() for m in quote_matcher_with_timestamp.finditer(t)]
+    # On instancie les quotes grâce aux dicos qui ont déjà la bonne tronche
+    l = [Quote(**q) for q in l]
+    newquotedb = QuoteDB()
+    newquotedb.quotelist = l
+    return newquotedb