X-Git-Url: https://git.mdrn.pl/redakcja.git/blobdiff_plain/1b3b64f7afe8ee6944f1f2d82ee5dabf3d8bb48b..73de75f6bdf2c2deb04f7c94ae62a2684af95759:/src/catalogue/models.py diff --git a/src/catalogue/models.py b/src/catalogue/models.py index 4a236b62..2ceb110b 100644 --- a/src/catalogue/models.py +++ b/src/catalogue/models.py @@ -1,38 +1,278 @@ +from collections import Counter +import decimal +from django.apps import apps from django.db import models +from django.urls import reverse from django.utils.translation import gettext_lazy as _ +from wikidata.client import Client +from .constants import WIKIDATA +from .wikidata import WikidataMixin -class Author(models.Model): - name = models.CharField(max_length=255) - year_of_death = models.SmallIntegerField(null=True, blank=True) - status = models.PositiveSmallIntegerField(null=True, blank=True, choices=[ - (1, _('Alive')), - (2, _('Dead')), - (3, _('Long dead')), - (4, _('Unknown')), - ]) +class Author(WikidataMixin, models.Model): + slug = models.SlugField(max_length=255, null=True, blank=True, unique=True) + first_name = models.CharField(_("first name"), max_length=255, blank=True) + last_name = models.CharField(_("last name"), max_length=255, blank=True) + + name_de = models.CharField(_("name (de)"), max_length=255, blank=True) + name_lt = models.CharField(_("name (lt)"), max_length=255, blank=True) + + gender = models.CharField(_("gender"), max_length=255, blank=True) + nationality = models.CharField(_("nationality"), max_length=255, blank=True) + year_of_death = models.SmallIntegerField(_("year of death"), null=True, blank=True) + status = models.PositiveSmallIntegerField( + _("status"), + null=True, + blank=True, + choices=[ + (1, _("Alive")), + (2, _("Dead")), + (3, _("Long dead")), + (4, _("Unknown")), + ], + ) + notes = models.TextField(_("notes"), blank=True) + gazeta_link = models.CharField(_("gazeta link"), max_length=255, blank=True) + culturepl_link = models.CharField(_("culture.pl link"), max_length=255, blank=True) + + description = models.TextField(_("description"), blank=True) + description_de = models.TextField(_("description (de)"), blank=True) + description_lt = models.TextField(_("description (lt)"), blank=True) + + priority = models.PositiveSmallIntegerField( + _("priority"), + default=0, choices=[(0, _("Low")), (1, _("Medium")), (2, _("High"))] + ) + collections = models.ManyToManyField("Collection", blank=True, verbose_name=_("collections")) + + class Meta: + verbose_name = _('author') + verbose_name_plural = _('authors') + ordering = ("last_name", "first_name", "year_of_death") + + class Wikidata: + first_name = WIKIDATA.GIVEN_NAME + last_name = WIKIDATA.LAST_NAME + year_of_death = WIKIDATA.DATE_OF_DEATH + gender = WIKIDATA.GENDER + notes = "description" + + def __str__(self): + return f"{self.first_name} {self.last_name}" + + def get_absolute_url(self): + return reverse("catalogue_author", args=[self.slug]) + + @property + def pd_year(self): + if self.year_of_death: + return self.year_of_death + 71 + elif self.year_of_death == 0: + return 0 + else: + return None + + +class Category(WikidataMixin, models.Model): + name = models.CharField(_("name"), max_length=255) + slug = models.SlugField(max_length=255, unique=True) + + class Meta: + abstract = True def __str__(self): return self.name +class Epoch(Category): + class Meta: + verbose_name = _('epoch') + verbose_name_plural = _('epochs') + + +class Genre(Category): + class Meta: + verbose_name = _('genre') + verbose_name_plural = _('genres') + + +class Kind(Category): + class Meta: + verbose_name = _('kind') + verbose_name_plural = _('kinds') + + +class Book(WikidataMixin, models.Model): + slug = models.SlugField(max_length=255, blank=True, null=True, unique=True) + authors = models.ManyToManyField(Author, blank=True, verbose_name=_("authors")) + translators = models.ManyToManyField( + Author, + related_name="translated_book_set", + related_query_name="translated_book", + blank=True, + verbose_name=_("translators") + ) + epochs = models.ManyToManyField(Epoch, blank=True, verbose_name=_("epochs")) + kinds = models.ManyToManyField(Kind, blank=True, verbose_name=_("kinds")) + genres = models.ManyToManyField(Genre, blank=True, verbose_name=_("genres")) + title = models.CharField(_("title"), max_length=255, blank=True) + language = models.CharField(_("language"), max_length=255, blank=True) + based_on = models.ForeignKey( + "self", models.PROTECT, related_name="translation", null=True, blank=True, + verbose_name=_("based on") + ) + scans_source = models.CharField(_("scans source"), max_length=255, blank=True) + text_source = models.CharField(_("text source"), max_length=255, blank=True) + notes = models.TextField(_("notes"), blank=True) + priority = models.PositiveSmallIntegerField( + _("priority"), + default=0, choices=[(0, _("Low")), (1, _("Medium")), (2, _("High"))] + ) + pd_year = models.IntegerField(_('year of entry into PD'), null=True, blank=True) + gazeta_link = models.CharField(_("gazeta link"), max_length=255, blank=True) + collections = models.ManyToManyField("Collection", blank=True, verbose_name=_("collections")) + + estimated_chars = models.IntegerField(_("estimated number of characters"), null=True, blank=True) + estimated_verses = models.IntegerField(_("estimated number of verses"), null=True, blank=True) + estimate_source = models.CharField(_("source of estimates"), max_length=2048, blank=True) + + class Meta: + ordering = ("title",) + verbose_name = _('book') + verbose_name_plural = _('books') + + class Wikidata: + authors = WIKIDATA.AUTHOR + translators = WIKIDATA.TRANSLATOR + title = WIKIDATA.TITLE + language = WIKIDATA.LANGUAGE + based_on = WIKIDATA.BASED_ON + notes = "description" -class Book(models.Model): - uri = models.CharField(max_length=255) + def __str__(self): + txt = self.title + astr = self.authors_str() + if astr: + txt = f"{astr} – {txt}" + tstr = self.translators_str() + if tstr: + txt = f"{txt} (tłum. {tstr})" + return txt + + def get_absolute_url(self): + return reverse("catalogue_book", args=[self.slug]) + + def authors_str(self): + return ", ".join(str(author) for author in self.authors.all()) + + def translators_str(self): + return ", ".join(str(author) for author in self.translators.all()) - authors = models.ManyToManyField(Author, blank=True) - translators = models.ManyToManyField(Author, related_name='translated_book_set', related_query_name='translated_book', blank=True) - title = models.CharField(max_length=255) - language = models.CharField(max_length=3) - based_on = models.ForeignKey('self', models.PROTECT, related_name='translation', null=True, blank=True) + def get_estimated_costs(self): + return { + work_type: work_type.calculate(self) + for work_type in WorkType.objects.all() + } - scans_source = models.CharField(max_length=255, blank=True) - text_source = models.CharField(max_length=255, blank=True) - notes = models.TextField(blank=True) - priority = models.PositiveSmallIntegerField(default=0, choices=[ - (0, _('Low')), - (1, _('Medium')), - (2, _('High')), - ]) + +class CollectionCategory(models.Model): + name = models.CharField(_("name"), max_length=255) + parent = models.ForeignKey('self', models.SET_NULL, related_name='children', null=True, blank=True, verbose_name=_("parent")) + notes = models.TextField(_("notes"), blank=True) + + class Meta: + ordering = ('parent__name', 'name') + verbose_name = _('collection category') + verbose_name_plural = _('collection categories') def __str__(self): - return self.title + if self.parent: + return f"{self.parent} / {self.name}" + else: + return self.name + + +class Collection(models.Model): + name = models.CharField(_("name"), max_length=255) + slug = models.SlugField(max_length=255, unique=True) + category = models.ForeignKey(CollectionCategory, models.SET_NULL, null=True, blank=True, verbose_name=_("category")) + notes = models.TextField(_("notes"), blank=True) + + class Meta: + ordering = ('category', 'name') + verbose_name = _('collection') + verbose_name_plural = _('collections') + + def __str__(self): + if self.category: + return f"{self.category} / {self.name}" + else: + return self.name + + def get_estimated_costs(self): + costs = Counter() + for book in self.book_set.all(): + for k, v in book.get_estimated_costs().items(): + costs[k] += v or 0 + + for author in self.author_set.all(): + for book in author.book_set.all(): + for k, v in book.get_estimated_costs().items(): + costs[k] += v or 0 + for book in author.translated_book_set.all(): + for k, v in book.get_estimated_costs().items(): + costs[k] += v or 0 + return costs + + +class WorkType(models.Model): + name = models.CharField(_("name"), max_length=255) + + class Meta: + ordering = ('name',) + verbose_name = _('work type') + verbose_name_plural = _('work types') + + def get_rate_for(self, book): + for workrate in self.workrate_set.all(): + if workrate.matches(book): + return workrate + + def calculate(self, book): + workrate = self.get_rate_for(book) + if workrate is not None: + return workrate.calculate(book) + + + +class WorkRate(models.Model): + priority = models.IntegerField(_("priority"), default=1) + per_normpage = models.DecimalField(_("per normalized page"), decimal_places=2, max_digits=6, null=True, blank=True) + per_verse = models.DecimalField(_("per verse"), decimal_places=2, max_digits=6, null=True, blank=True) + work_type = models.ForeignKey(WorkType, models.CASCADE, verbose_name=_("work type")) + epochs = models.ManyToManyField(Epoch, blank=True, verbose_name=_("epochs")) + kinds = models.ManyToManyField(Kind, blank=True, verbose_name=_("kinds")) + genres = models.ManyToManyField(Genre, blank=True, verbose_name=_("genres")) + collections = models.ManyToManyField(Collection, blank=True, verbose_name=_("collections")) + + class Meta: + ordering = ('priority',) + verbose_name = _('work rate') + verbose_name_plural = _('work rates') + + def matches(self, book): + for category in 'epochs', 'kinds', 'genres', 'collections': + oneof = getattr(self, category).all() + if oneof: + if not set(oneof).intersection( + getattr(book, category).all()): + return False + return True + + def calculate(self, book): + if self.per_verse: + if book.estimated_verses: + return book.estimated_verses * self.per_verse + elif self.per_normpage: + if book.estimated_chars: + return (decimal.Decimal(book.estimated_chars) / 1800 * self.per_normpage).quantize(decimal.Decimal('1.00'), rounding=decimal.ROUND_HALF_UP) +