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
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."""