Catalogue improvements.

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

diff --git a/src/catalogue/models.py b/src/catalogue/models.py
index 1897846..4568af1 100644 (file)
--- a/src/catalogue/models.py
+++ b/src/catalogue/models.py
@@ -19,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,
@@ -58,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:
@@ -135,6 +145,9 @@ 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)
 
+    free_license = models.BooleanField(_('free license'), default=False)
+    polona_missing = models.BooleanField(_('missing on Polona'), default=False)
+
     class Meta:
         ordering = ("title",)
         verbose_name = _('book')
@@ -160,12 +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())
+    translators_str.admin_order_field = 'translators__last_name'
+    translators_str.short_description = _('Translator')
 
     def get_estimated_costs(self):
         return {
@@ -177,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')
@@ -194,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')