Catalogue improvements.

[redakcja.git] / src / catalogue / models.py

diff --git a/src/catalogue/models.py b/src/catalogue/models.py
index cd64f84..4568af1 100644 (file)
--- a/src/catalogue/models.py
+++ b/src/catalogue/models.py
@@ -6,7 +6,6 @@ from django.urls import reverse
 from django.utils.translation import gettext_lazy as _
 from wikidata.client import Client
 from .constants import WIKIDATA
-from .utils import UnrelatedManager
 from .wikidata import WikidataMixin
 
 
@@ -20,7 +19,10 @@ class Author(WikidataMixin, models.Model):
 
     gender = models.CharField(_("gender"), max_length=255, blank=True)
     nationality = models.CharField(_("nationality"), max_length=255, blank=True)
+    year_of_birth = models.SmallIntegerField(_("year of birth"), null=True, blank=True)
+    place_of_birth = models.CharField(_('place of birth'), max_length=255, blank=True)
     year_of_death = models.SmallIntegerField(_("year of death"), null=True, blank=True)
+    place_of_death = models.CharField(_('place of death'), max_length=255, blank=True)
     status = models.PositiveSmallIntegerField(
         _("status"), 
         null=True,
@@ -59,11 +61,18 @@ class Author(WikidataMixin, models.Model):
         notes = "description"
 
     def __str__(self):
-        return f"{self.first_name} {self.last_name}"
+        name = f"{self.first_name} {self.last_name}"
+        if self.year_of_death is not None:
+            name += f' (zm. {self.year_of_death})'
+        return name
 
     def get_absolute_url(self):
         return reverse("catalogue_author", args=[self.slug])
 
+    @property
+    def name(self):
+        return f"{self.last_name}, {self.first_name}"
+    
     @property
     def pd_year(self):
         if self.year_of_death:
@@ -136,7 +145,8 @@ class Book(WikidataMixin, models.Model):
     estimated_verses = models.IntegerField(_("estimated number of verses"), null=True, blank=True)
     estimate_source = models.CharField(_("source of estimates"), max_length=2048, blank=True)
 
-    objects = UnrelatedManager()
+    free_license = models.BooleanField(_('free license'), default=False)
+    polona_missing = models.BooleanField(_('missing on Polona'), default=False)
 
     class Meta:
         ordering = ("title",)
@@ -163,16 +173,20 @@ class Book(WikidataMixin, models.Model):
 
     def get_absolute_url(self):
         return reverse("catalogue_book", args=[self.slug])
+
+    @property
+    def wluri(self):
+        return f'https://wolnelektury.pl/katalog/lektura/{self.slug}/'
     
     def authors_str(self):
         return ", ".join(str(author) for author in self.authors.all())
+    authors_str.admin_order_field = 'authors__last_name'
+    authors_str.short_description = _('Author')
 
     def translators_str(self):
         return ", ".join(str(author) for author in self.translators.all())
-
-    def get_document_books(self):
-        DBook = apps.get_model("documents", "Book")
-        return DBook.objects.filter(dc_slug=self.slug)
+    translators_str.admin_order_field = 'translators__last_name'
+    translators_str.short_description = _('Translator')
 
     def get_estimated_costs(self):
         return {
@@ -184,6 +198,7 @@ class Book(WikidataMixin, models.Model):
 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')
@@ -201,6 +216,7 @@ 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')