src/catalogue/models.py

   1 from django.db import models
   2 from django.utils.translation import gettext_lazy as _
   3 from .constants import WIKIDATA
   4 from .wikidata import WikidataMixin
   5
   6
   7 class Author(WikidataMixin, models.Model):
   8     slug = models.SlugField(null=True, blank=True, unique=True)
   9     first_name = models.CharField(max_length=255, blank=True)
  10     last_name = models.CharField(max_length=255, blank=True)
  11     year_of_death = models.SmallIntegerField(null=True, blank=True)
  12     status = models.PositiveSmallIntegerField(
  13         null=True,
  14         blank=True,
  15         choices=[
  16             (1, _("Alive")),
  17             (2, _("Dead")),
  18             (3, _("Long dead")),
  19             (4, _("Unknown")),
  20         ],
  21     )
  22     notes = models.TextField(blank=True)
  23     priority = models.PositiveSmallIntegerField(
  24         default=0, choices=[(0, _("Low")), (1, _("Medium")), (2, _("High"))]
  25     )
  26
  27     class Wikidata:
  28         first_name = WIKIDATA.GIVEN_NAME
  29         last_name = WIKIDATA.LAST_NAME
  30         year_of_death = WIKIDATA.DATE_OF_DEATH
  31         notes = "description"
  32
  33     def __str__(self):
  34         return f"{self.first_name} {self.last_name}"
  35
  36
  37 class Book(WikidataMixin, models.Model):
  38     slug = models.SlugField(max_length=255, blank=True, null=True, unique=True)
  39     authors = models.ManyToManyField(Author, blank=True)
  40     translators = models.ManyToManyField(
  41         Author,
  42         related_name="translated_book_set",
  43         related_query_name="translated_book",
  44         blank=True,
  45     )
  46     title = models.CharField(max_length=255, blank=True)
  47     language = models.CharField(max_length=3, blank=True)
  48     based_on = models.ForeignKey(
  49         "self", models.PROTECT, related_name="translation", null=True, blank=True
  50     )
  51     scans_source = models.CharField(max_length=255, blank=True)
  52     text_source = models.CharField(max_length=255, blank=True)
  53     notes = models.TextField(blank=True)
  54     priority = models.PositiveSmallIntegerField(
  55         default=0, choices=[(0, _("Low")), (1, _("Medium")), (2, _("High"))]
  56     )
  57
  58     class Wikidata:
  59         authors = WIKIDATA.AUTHOR
  60         translators = WIKIDATA.TRANSLATOR
  61         title = WIKIDATA.TITLE
  62         language = WIKIDATA.LANGUAGE
  63         based_on = WIKIDATA.BASED_ON
  64         notes = "description"
  65
  66     def __str__(self):
  67         return self.title