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)
+spaces_matcher = re.compile(u"\s", flags=re.U)
+
def get_now():
""" Renvoie la date actuelle """
return datetime.datetime(*time.localtime()[:6])
+def sanitize_author(raw):
+ """Proprifie l'auteur : enlève les espaces insécables."""
+ return spaces_matcher.sub(u" ", raw)
+
class Quote(object):
""" Une citation """
def __init__(self, author, content, timestamp=None):
timestamp = get_now()
elif isinstance(timestamp, basestring):
timestamp = datetime.datetime(*time.strptime(timestamp, u"%Y-%m-%d_%H:%M:%S")[:6])
- self.author = author
+ self.author = sanitize_author(author)
self.content = content
self.timestamp = timestamp
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. """
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 search(self, inquote=None, author=None, regexp=False):
+ """Fait une recherche dans les quotes."""
+ if regexp:
+ if inquote is None:
+ inquote = ".*"
+ if author is None:
+ author = ".*"
+ qreg = re.compile(inquote, flags=re.UNICODE)
+ areg = re.compile(author, flags=re.UNICODE)
+ l = [q for q in self.quotelist if qreg.match(q.content) and areg.match(q.author)]
+ else:
+ if inquote is None:
+ inquote = ""
+ if author is None:
+ author = ""
+ l = [q for q in self.quotelist if inquote in q.content and author in q.author]
+ return l
+
+ def search_authors(self, author=None, regexp=False):
+ """Renvoie la liste des auteurs contenant ``author`` ou qui matchent la regexp."""
+ if regexp:
+ if author is None:
+ author = ".*"
+ areg = re.compile(author, flags=re.UNICODE)
+ l = list(set([q.author for q in self.quotelist if areg.match(q.author)]))
+ else:
+ if author is None:
+ author = ""
+ l = list(set([q.author for q in self.quotelist if author in q.author]))
+ return l
def dump(quotedb, dump_file=None):
"""Pour exporter les quotes dans un format readable vers un fichier."""
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
- regex = re.compile(config.quote_regexp_with_timestamp)
- l = [m.groupdict() for m in regex.finditer(t)]
+ 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()