X-Git-Url: http://gitweb.pimeys.fr/?p=bots%2Fparrot.git;a=blobdiff_plain;f=quotes.py;h=5c1a58ebbd4330230334cb91c821a4296dda9a1a;hp=f6e4bc3dabbcf471f46cd6e123d9a403a5bac9b7;hb=598d5fb9c8929682e5785a0ec19747a153ed9e48;hpb=e16477da6ca556666492b839f5a79ec84c41018d diff --git a/quotes.py b/quotes.py index f6e4bc3..5c1a58e 100644 --- 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 """ @@ -59,16 +59,20 @@ class Quote(object): 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 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") + 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. @@ -123,7 +127,7 @@ 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] @@ -137,7 +141,6 @@ class QuoteDB(object): kwargs["timestamp"] = timestamp newquote = Quote(**kwargs) if not newquote in self.quotelist: - self.search self.quotelist.append(newquote) return True return False @@ -148,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]