src/pdcounter/models.py

   1 # This file is part of Wolnelektury, licensed under GNU Affero GPLv3 or later.
   2 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
   3 #
   4 from django.conf import settings
   5 from django.db import models
   6 from django.urls import reverse
   7 from django.utils.translation import ugettext_lazy as _
   8 from datetime import datetime
   9 from django.db.models.signals import post_save, post_delete
  10
  11
  12 class Author(models.Model):
  13     name = models.CharField(_('name'), max_length=50, db_index=True)
  14     slug = models.SlugField(_('slug'), max_length=120, db_index=True, unique=True)
  15     sort_key = models.CharField(_('sort key'), max_length=120, db_index=True)
  16     description = models.TextField(_('description'), blank=True)
  17     death = models.IntegerField(_('year of death'), blank=True, null=True)
  18     gazeta_link = models.CharField(blank=True, max_length=240)
  19     wiki_link = models.CharField(blank=True, max_length=240)
  20
  21     class Meta:
  22         ordering = ('sort_key',)
  23         verbose_name = _('author')
  24         verbose_name_plural = _('authors')
  25
  26     @property
  27     def category(self):
  28         return "author"
  29
  30     def __str__(self):
  31         return self.name
  32
  33     def __repr__(self):
  34         return "Author(slug=%r)" % self.slug
  35
  36     def get_absolute_url(self):
  37         return reverse('tagged_object_list', args=[self.url_chunk])
  38
  39     def has_description(self):
  40         return len(self.description) > 0
  41     has_description.short_description = _('description')
  42     has_description.boolean = True
  43
  44     def alive(self):
  45         return self.death is None
  46
  47     def in_pd(self):
  48         """ tests whether an author is in public domain """
  49         return self.death is not None and self.goes_to_pd() <= datetime.now().year
  50
  51     def goes_to_pd(self):
  52         """ calculates the year of public domain entry for an author """
  53         return self.death + 71 if self.death is not None else None
  54
  55     @property
  56     def url_chunk(self):
  57         return '/'.join(('autor', self.slug))
  58
  59
  60 class BookStub(models.Model):
  61     title = models.CharField(_('title'), max_length=120)
  62     author = models.CharField(_('author'), max_length=120)
  63     pd = models.IntegerField(_('goes to public domain'), null=True, blank=True)
  64     slug = models.SlugField(_('slug'), max_length=120, unique=True, db_index=True)
  65     translator = models.TextField(_('translator'), blank=True)
  66
  67     class Meta:
  68         ordering = ('title',)
  69         verbose_name = _('book stub')
  70         verbose_name_plural = _('book stubs')
  71
  72     def __str__(self):
  73         return self.title
  74
  75     def get_absolute_url(self):
  76         return reverse('book_detail', args=[self.slug])
  77
  78     def in_pd(self):
  79         return self.pd is not None and self.pd <= datetime.now().year
  80
  81     @property
  82     def name(self):
  83         return self.title
  84
  85     def pretty_title(self, html_links=False):
  86         return ', '.join((self.author, self.title))
  87
  88
  89 if not settings.NO_SEARCH_INDEX:
  90     def update_index(sender, instance, **kwargs):
  91         from search.index import Index
  92         idx = Index()
  93         idx.index_tags(instance, remove_only='created' not in kwargs)
  94
  95     post_delete.connect(update_index, Author)
  96     post_delete.connect(update_index, BookStub)
  97     post_save.connect(update_index, Author)
  98     post_save.connect(update_index, BookStub)