X-Git-Url: https://git.mdrn.pl/wolnelektury.git/blobdiff_plain/87e084d47c335cd6f0b3e91d614999f55d148044..7e07f9fa2258e13475e7f9940af952c8d68d8c03:/src/catalogue/models/collection.py diff --git a/src/catalogue/models/collection.py b/src/catalogue/models/collection.py index bae1cc418..15a4e2adc 100644 --- a/src/catalogue/models/collection.py +++ b/src/catalogue/models/collection.py @@ -18,7 +18,8 @@ class Collection(models.Model): models.SlugField(_('slug'), max_length=120, unique=True, db_index=True) book_slugs = models.TextField(_('book slugs')) - kind = models.CharField(_('kind'), max_length=10, blank=False, default='book', db_index=True, choices=((('book'), _('book')), (('picture'), ('picture')))) + kind = models.CharField(_('kind'), max_length=10, blank=False, default='book', db_index=True, + choices=(('book', _('book')), ('picture', _('picture')))) class Meta: ordering = ('title',) @@ -33,17 +34,17 @@ class Collection(models.Model): try: return re.search(r'\w', self.title, re.U).group(0) except AttributeError: - return None + return '' @models.permalink def get_absolute_url(self): - return ("collection", [self.slug]) + return "collection", [self.slug] def get_query(self): slugs = self.book_slugs.split() # allow URIs - slugs = [slug.rstrip('/').rsplit('/', 1)[-1] if '/' in slug else slug - for slug in slugs] + # WTF + slugs = [slug.rstrip('/').rsplit('/', 1)[-1] if '/' in slug else slug for slug in slugs] return models.Q(slug__in=slugs) def get_books(self):