src/catalogue/models.py

   1 import decimal
   2 from django.apps import apps
   3 from django.db import models
   4 from django.urls import reverse
   5 from django.utils.translation import gettext_lazy as _
   6 from wikidata.client import Client
   7 from .constants import WIKIDATA
   8 from .utils import UnrelatedManager
   9 from .wikidata import WikidataMixin
  10
  11
  12 class Author(WikidataMixin, models.Model):
  13     slug = models.SlugField(max_length=255, null=True, blank=True, unique=True)
  14     first_name = models.CharField(max_length=255, blank=True)
  15     last_name = models.CharField(max_length=255, blank=True)
  16
  17     name_de = models.CharField(max_length=255, blank=True)
  18     name_lt = models.CharField(max_length=255, blank=True)
  19
  20     gender = models.CharField(max_length=255, blank=True)
  21     nationality = models.CharField(max_length=255, blank=True)
  22     year_of_death = models.SmallIntegerField(null=True, blank=True)
  23     status = models.PositiveSmallIntegerField(
  24         null=True,
  25         blank=True,
  26         choices=[
  27             (1, _("Alive")),
  28             (2, _("Dead")),
  29             (3, _("Long dead")),
  30             (4, _("Unknown")),
  31         ],
  32     )
  33     notes = models.TextField(blank=True)
  34     gazeta_link = models.CharField(max_length=255, blank=True)
  35     culturepl_link = models.CharField(max_length=255, blank=True)
  36
  37     description = models.TextField(blank=True)
  38     description_de = models.TextField(blank=True)
  39     description_lt = models.TextField(blank=True)
  40
  41     priority = models.PositiveSmallIntegerField(
  42         default=0, choices=[(0, _("Low")), (1, _("Medium")), (2, _("High"))]
  43     )
  44     collections = models.ManyToManyField("Collection", blank=True)
  45
  46     class Meta:
  47         verbose_name = _('author')
  48         verbose_name_plural = _('authors')
  49         ordering = ("last_name", "first_name", "year_of_death")
  50
  51     class Wikidata:
  52         first_name = WIKIDATA.GIVEN_NAME
  53         last_name = WIKIDATA.LAST_NAME
  54         year_of_death = WIKIDATA.DATE_OF_DEATH
  55         gender = WIKIDATA.GENDER
  56         notes = "description"
  57
  58     def __str__(self):
  59         return f"{self.first_name} {self.last_name}"
  60
  61     def get_absolute_url(self):
  62         return reverse("catalogue_author", args=[self.slug])
  63
  64     @property
  65     def pd_year(self):
  66         if self.year_of_death:
  67             return self.year_of_death + 71
  68         elif self.year_of_death == 0:
  69             return 0
  70         else:
  71             return None
  72
  73
  74 class Category(WikidataMixin, models.Model):
  75     name = models.CharField(max_length=255)
  76     slug = models.SlugField(max_length=255, unique=True)
  77
  78     class Meta:
  79         abstract = True
  80
  81     def __str__(self):
  82         return self.name
  83
  84 class Epoch(Category):
  85     class Meta:
  86         verbose_name = _('epoch')
  87         verbose_name_plural = _('epochs')
  88
  89
  90 class Genre(Category):
  91     class Meta:
  92         verbose_name = _('genre')
  93         verbose_name_plural = _('genres')
  94
  95
  96 class Kind(Category):
  97     class Meta:
  98         verbose_name = _('kind')
  99         verbose_name_plural = _('kinds')
 100
 101
 102 class Book(WikidataMixin, models.Model):
 103     slug = models.SlugField(max_length=255, blank=True, null=True, unique=True)
 104     authors = models.ManyToManyField(Author, blank=True)
 105     translators = models.ManyToManyField(
 106         Author,
 107         related_name="translated_book_set",
 108         related_query_name="translated_book",
 109         blank=True,
 110     )
 111     epochs = models.ManyToManyField(Epoch, blank=True)
 112     kinds = models.ManyToManyField(Kind, blank=True)
 113     genres = models.ManyToManyField(Genre, blank=True)
 114     title = models.CharField(max_length=255, blank=True)
 115     language = models.CharField(max_length=255, blank=True)
 116     based_on = models.ForeignKey(
 117         "self", models.PROTECT, related_name="translation", null=True, blank=True
 118     )
 119     scans_source = models.CharField(max_length=255, blank=True)
 120     text_source = models.CharField(max_length=255, blank=True)
 121     notes = models.TextField(blank=True)
 122     priority = models.PositiveSmallIntegerField(
 123         default=0, choices=[(0, _("Low")), (1, _("Medium")), (2, _("High"))]
 124     )
 125     pd_year = models.IntegerField(null=True, blank=True)
 126     gazeta_link = models.CharField(max_length=255, blank=True)
 127     collections = models.ManyToManyField("Collection", blank=True)
 128
 129     estimated_chars = models.IntegerField(null=True, blank=True)
 130     estimated_verses = models.IntegerField(null=True, blank=True)
 131     estimate_source = models.CharField(max_length=2048, blank=True)
 132
 133     objects = UnrelatedManager()
 134
 135     class Meta:
 136         ordering = ("title",)
 137         verbose_name = _('book')
 138         verbose_name_plural = _('books')
 139
 140     class Wikidata:
 141         authors = WIKIDATA.AUTHOR
 142         translators = WIKIDATA.TRANSLATOR
 143         title = WIKIDATA.TITLE
 144         language = WIKIDATA.LANGUAGE
 145         based_on = WIKIDATA.BASED_ON
 146         notes = "description"
 147
 148     def __str__(self):
 149         txt = self.title
 150         astr = self.authors_str()
 151         if astr:
 152             txt = f"{astr} – {txt}"
 153         tstr = self.translators_str()
 154         if tstr:
 155             txt = f"{txt} (tłum. {tstr})"
 156         return txt
 157
 158     def get_absolute_url(self):
 159         return reverse("catalogue_book", args=[self.slug])
 160
 161     def authors_str(self):
 162         return ", ".join(str(author) for author in self.authors.all())
 163
 164     def translators_str(self):
 165         return ", ".join(str(author) for author in self.translators.all())
 166
 167     def get_document_books(self):
 168         DBook = apps.get_model("documents", "Book")
 169         return DBook.objects.filter(dc_slug=self.slug)
 170
 171     def get_estimated_costs(self):
 172         return {
 173             work_type: work_type.calculate(self)
 174             for work_type in WorkType.objects.all()
 175         }
 176
 177
 178 class Collection(models.Model):
 179     name = models.CharField(max_length=255)
 180     slug = models.SlugField(max_length=255, unique=True)
 181
 182     class Meta:
 183         verbose_name = _('collection')
 184         verbose_name_plural = _('collections')
 185
 186     def __str__(self):
 187         return self.name
 188
 189
 190 class WorkType(models.Model):
 191     name = models.CharField(max_length=255)
 192
 193     def get_rate_for(self, book):
 194         for workrate in self.workrate_set.all():
 195             if workrate.matches(book):
 196                 return workrate
 197
 198     def calculate(self, book):
 199         workrate = self.get_rate_for(book)
 200         if workrate is not None:
 201             return workrate.calculate(book)
 202
 203
 204
 205 class WorkRate(models.Model):
 206     priority = models.IntegerField(default=1)
 207     per_normpage = models.DecimalField(decimal_places=2, max_digits=6, null=True, blank=True)
 208     per_verse = models.DecimalField(decimal_places=2, max_digits=6, null=True, blank=True)
 209     work_type = models.ForeignKey(WorkType, models.CASCADE)
 210     epochs = models.ManyToManyField(Epoch, blank=True)
 211     kinds = models.ManyToManyField(Kind, blank=True)
 212     genres = models.ManyToManyField(Genre, blank=True)
 213     collections = models.ManyToManyField(Collection, blank=True)
 214
 215     class Meta:
 216         ordering = ('priority',)
 217
 218     def matches(self, book):
 219         for category in 'epochs', 'kinds', 'genres', 'collections':
 220             oneof = getattr(self, category).all()
 221             if oneof:
 222                 if not set(oneof).intersection(
 223                         getattr(book, category).all()):
 224                     return False
 225         return True
 226
 227     def calculate(self, book):
 228         if self.per_verse:
 229             if book.estimated_verses:
 230                 return book.estimated_verses * self.per_verse
 231         elif self.per_normpage:
 232             if book.estimated_chars:
 233                 return (decimal.Decimal(book.estimated_chars) / 1800 * self.per_normpage).quantize(decimal.Decimal('1.00'), rounding=decimal.ROUND_HALF_UP)
 234