-# -*- coding: utf-8 -*-
# This file is part of Wolnelektury, licensed under GNU Affero GPLv3 or later.
# Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
#
from django.conf import settings
+from django.contrib.contenttypes.fields import GenericForeignKey
+from django.contrib.contenttypes.models import ContentType
from django.core.cache import caches
from django.contrib.auth.models import User
+from django.core.exceptions import ObjectDoesNotExist
from django.db import models
-from django.db.models import permalink
+from django.db.models.query import Prefetch
from django.dispatch import Signal
-from django.utils.translation import ugettext_lazy as _
-from newtagging.models import TagBase
-from ssify import flush_ssi_includes
+from django.urls import reverse
+from django.utils.translation import gettext_lazy as _
+
+from newtagging.models import TagManager, TaggedItemManager
# Those are hard-coded here so that makemessages sees them.
)
-class Tag(TagBase):
+class TagRelation(models.Model):
+
+ tag = models.ForeignKey('Tag', models.CASCADE, verbose_name=_('tag'), related_name='items')
+ content_type = models.ForeignKey(ContentType, models.CASCADE, verbose_name=_('content type'))
+ object_id = models.PositiveIntegerField(_('object id'), db_index=True)
+ content_object = GenericForeignKey('content_type', 'object_id')
+
+ objects = TaggedItemManager()
+
+ class Meta:
+ db_table = 'catalogue_tag_relation'
+ unique_together = (('tag', 'content_type', 'object_id'),)
+
+ def __str__(self):
+ try:
+ return '%s [%s]' % (self.content_type.get_object_for_this_type(pk=self.object_id), self.tag)
+ except ObjectDoesNotExist:
+ return '<deleted> [%s]' % self.tag
+
+
+class Tag(models.Model):
"""A tag attachable to books and fragments (and possibly anything).
Used to represent searchable metadata (authors, epochs, genres, kinds),
_('category'), max_length=50, blank=False, null=False, db_index=True, choices=TAG_CATEGORIES)
description = models.TextField(_('description'), blank=True)
- user = models.ForeignKey(User, blank=True, null=True)
+ for_books = models.BooleanField(default=False)
+ for_pictures = models.BooleanField(default=False)
+
+ user = models.ForeignKey(User, models.CASCADE, blank=True, null=True)
gazeta_link = models.CharField(blank=True, max_length=240)
culturepl_link = models.CharField(blank=True, max_length=240)
wiki_link = models.CharField(blank=True, max_length=240)
+ photo = models.FileField(blank=True, null=True, upload_to='catalogue/tag/')
+ photo_attribution = models.CharField(max_length=255, blank=True)
created_at = models.DateTimeField(_('creation date'), auto_now_add=True, db_index=True)
changed_at = models.DateTimeField(_('creation date'), auto_now=True, db_index=True)
- after_change = Signal(providing_args=['instance', 'languages'])
+ plural = models.CharField(
+ 'liczba mnoga', max_length=255, blank=True,
+ help_text='dotyczy gatunków'
+ )
+ genre_epoch_specific = models.BooleanField(
+ default=False,
+ help_text='Po wskazaniu tego gatunku, dodanie epoki byłoby nadmiarowe, np. „dramat romantyczny”'
+ )
+ adjective_feminine_singular = models.CharField(
+ 'przymiotnik pojedynczy żeński', max_length=255, blank=True,
+ help_text='twórczość … Adama Mickiewicza; dotyczy epok'
+ )
+ adjective_nonmasculine_plural = models.CharField(
+ 'przymiotnik mnogi niemęskoosobowy', max_length=255, blank=True,
+ help_text='utwory … Adama Mickiewicza; dotyczy epok'
+ )
+ genitive = models.CharField(
+ 'dopełniacz', max_length=255, blank=True,
+ help_text='utwory … (czyje?); dotyczy autorów'
+ )
+ collective_noun = models.CharField(
+ 'określenie zbiorowe', max_length=255, blank=True,
+ help_text='np. „Liryka” albo „Twórczość dramatyczna”; dotyczy rodzajów'
+ )
+
+ after_change = Signal()
+
+ intermediary_table_model = TagRelation
+ objects = TagManager()
class UrlDeprecationWarning(DeprecationWarning):
- pass
+ def __init__(self, tags=None):
+ super(Tag.UrlDeprecationWarning, self).__init__()
+ self.tags = tags
categories_rev = {
'autor': 'author',
'polka': 'set',
'obiekt': 'thing',
}
- categories_dict = dict((item[::-1] for item in categories_rev.iteritems()))
+ categories_dict = dict((item[::-1] for item in categories_rev.items()))
class Meta:
ordering = ('sort_key',)
unique_together = (("slug", "category"),)
app_label = 'catalogue'
- def save(self, *args, **kwargs):
- flush_cache = flush_all_includes = False
- if self.pk and self.category != 'set':
- # Flush the whole views cache.
- # Seem a little harsh, but changed tag names, descriptions
- # and links come up at any number of places.
- flush_cache = True
-
- # Find in which languages we need to flush related includes.
- old_self = type(self).objects.get(pk=self.pk)
- # Category shouldn't normally be changed, but just in case.
- if self.category != old_self.category:
- flush_all_includes = True
- languages_changed = self.languages_changed(old_self)
-
+ def save(self, *args, quick=False, **kwargs):
+ existing = self.pk and self.category != 'set'
ret = super(Tag, self).save(*args, **kwargs)
-
- if flush_cache:
- caches[settings.CACHE_MIDDLEWARE_ALIAS].clear()
- if flush_all_includes:
- flush_ssi_includes()
- else:
- self.flush_includes()
- self.after_change.send(sender=type(self), instance=self, languages=languages_changed)
-
+ if existing and not quick:
+ self.after_change.send(sender=type(self), instance=self)
return ret
- def languages_changed(self, old):
- all_langs = [lc for (lc, _ln) in settings.LANGUAGES]
- if (old.category, old.slug) != (self.category, self.slug):
- return all_langs
- languages = set()
- for lang in all_langs:
- name_field = 'name_%s' % lang
- if getattr(old, name_field) != getattr(self, name_field):
- languages.add(lang)
- return languages
-
- def flush_includes(self, languages=True):
- if not languages:
- return
- if languages is True:
- languages = [lc for (lc, _ln) in settings.LANGUAGES]
- flush_ssi_includes([
- template % (self.pk, lang)
- for template in [
- '/api/include/tag/%d.%s.json',
- '/api/include/tag/%d.%s.xml',
- ]
- for lang in languages
- ])
- flush_ssi_includes([
- '/katalog/%s.json' % lang for lang in languages])
-
- def __unicode__(self):
+ def __str__(self):
return self.name
def __repr__(self):
else:
return ''
- @permalink
+ @property
+ def category_plural(self):
+ return self.category + 's'
+
def get_absolute_url(self):
- return 'tagged_object_list', [self.url_chunk]
+ return reverse('tagged_object_list', args=[self.url_chunk])
- @permalink
def get_absolute_gallery_url(self):
- return 'tagged_object_list_gallery', [self.url_chunk]
+ return reverse('tagged_object_list_gallery', args=[self.url_chunk])
- @classmethod
- @permalink
- def create_url(cls, category, slug):
- return ('catalogue.views.tagged_object_list', [
- '/'.join((cls.categories_dict[category], slug))
- ])
+ def get_absolute_catalogue_url(self):
+ # TODO: remove magic.
+ return reverse(f'{self.category}_catalogue')
def has_description(self):
return len(self.description) > 0
has_description.boolean = True
@staticmethod
- def get_tag_list(tags):
- if isinstance(tags, basestring):
- if not tags:
- return []
- real_tags = []
- ambiguous_slugs = []
- category = None
- deprecated = False
- tags_splitted = tags.split('/')
- for name in tags_splitted:
- if category:
- real_tags.append(Tag.objects.get(slug=name, category=category))
- category = None
- elif name in Tag.categories_rev:
- category = Tag.categories_rev[name]
- else:
- try:
- real_tags.append(Tag.objects.get(slug=name))
- deprecated = True
- except Tag.MultipleObjectsReturned:
- ambiguous_slugs.append(name)
-
+ def get_tag_list(tag_str):
+ if not tag_str:
+ return []
+ tags = []
+ ambiguous_slugs = []
+ category = None
+ deprecated = False
+ tags_splitted = tag_str.split('/')
+ for name in tags_splitted:
if category:
- # something strange left off
- raise Tag.DoesNotExist()
- if ambiguous_slugs:
- # some tags should be qualified
- e = Tag.MultipleObjectsReturned()
- e.tags = real_tags
- e.ambiguous_slugs = ambiguous_slugs
- raise e
- if deprecated:
- e = Tag.UrlDeprecationWarning()
- e.tags = real_tags
- raise e
- return real_tags
- else:
- return TagBase.get_tag_list(tags)
+ tags.append(Tag.objects.get(slug=name, category=category))
+ category = None
+ elif name in Tag.categories_rev:
+ category = Tag.categories_rev[name]
+ else:
+ try:
+ tags.append(Tag.objects.get(slug=name))
+ deprecated = True
+ except Tag.MultipleObjectsReturned:
+ ambiguous_slugs.append(name)
+
+ if category:
+ # something strange left off
+ raise Tag.DoesNotExist()
+ if ambiguous_slugs:
+ # some tags should be qualified
+ e = Tag.MultipleObjectsReturned()
+ e.tags = tags
+ e.ambiguous_slugs = ambiguous_slugs
+ raise e
+ if deprecated:
+ raise Tag.UrlDeprecationWarning(tags=tags)
+ return tags
@property
def url_chunk(self):
@staticmethod
def tags_from_info(info):
- from fnpdjango.utils.text.slughifi import slughifi
+ from slugify import slugify
from sortify import sortify
meta_tags = []
categories = (('kinds', 'kind'), ('genres', 'genre'), ('authors', 'author'), ('epochs', 'epoch'))
for field_name, category in categories:
try:
tag_names = getattr(info, field_name)
- except KeyError:
+ except (AttributeError, KeyError): # TODO: shouldn't be KeyError here at all.
try:
tag_names = [getattr(info, category)]
except KeyError:
# For instance, Pictures do not have 'genre' field.
continue
for tag_name in tag_names:
- lang = getattr(tag_name, 'lang', settings.LANGUAGE_CODE)
+ lang = getattr(tag_name, 'lang', None) or settings.LANGUAGE_CODE
tag_sort_key = tag_name
if category == 'author':
- tag_sort_key = tag_name.last_name
+ tag_sort_key = ' '.join((tag_name.last_name,) + tag_name.first_names)
tag_name = tag_name.readable()
- if lang == settings.LANGUAGE_CODE:
- # Allow creating new tag, if it's in default language.
- tag, created = Tag.objects.get_or_create(slug=slughifi(tag_name), category=category)
- if created:
- tag_name = unicode(tag_name)
- tag.name = tag_name
- setattr(tag, "name_%s" % lang, tag_name)
- tag.sort_key = sortify(tag_sort_key.lower())
- tag.save()
- meta_tags.append(tag)
- else:
- # Ignore unknown tags in non-default languages.
- try:
- tag = Tag.objects.get(category=category, **{"name_%s" % lang: tag_name})
- except Tag.DoesNotExist:
- pass
- else:
+ try:
+ tag = Tag.objects.get(category=category, **{"name_%s" % lang: tag_name})
+ except Tag.DoesNotExist:
+ if lang == settings.LANGUAGE_CODE:
+ # Allow creating new tag, if it's in default language.
+ tag, created = Tag.objects.get_or_create(slug=slugify(tag_name), category=category)
+ if created:
+ tag_name = str(tag_name)
+ tag.name = tag_name
+ setattr(tag, "name_%s" % lang, tag_name)
+ tag.sort_key = sortify(tag_sort_key.lower())
+ tag.save()
+
meta_tags.append(tag)
+ else:
+ meta_tags.append(tag)
return meta_tags
-# Pickle complains about not having this.
-TagRelation = Tag.intermediary_table_model
+TagRelation.tag_model = Tag
+
+
+def prefetch_relations(objects, category, only_name=True):
+ queryset = TagRelation.objects.filter(tag__category=category).select_related('tag')
+ if only_name:
+ queryset = queryset.only('tag__name_pl', 'object_id')
+ return objects.prefetch_related(
+ Prefetch('tag_relations', queryset=queryset, to_attr='%s_relations' % category))
+
+
+def prefetched_relations(obj, category):
+ if hasattr(obj, '%s_relations' % category):
+ return getattr(obj, '%s_relations' % category)
+ else:
+ return None