Add notes to collections in catalogue.

[redakcja.git] / src / documents / models / book.py

diff --git a/src/documents/models/book.py b/src/documents/models/book.py
index e7c3481..75e707e 100644 (file)
--- a/src/documents/models/book.py
+++ b/src/documents/models/book.py
@@ -1,6 +1,7 @@
 # This file is part of FNP-Redakcja, licensed under GNU Affero GPLv3 or later.
 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
 #
+from django.apps import apps
 from django.contrib.sites.models import Site
 from django.db import models, transaction
 from django.template.loader import render_to_string
@@ -29,7 +30,6 @@ class Book(models.Model):
     gallery = models.CharField(_('scan gallery name'), max_length=255, blank=True)
     project = models.ForeignKey(Project, models.SET_NULL, null=True, blank=True)
 
-    #wl_slug = models.CharField(_('title'), max_length=255, null=True, db_index=True, editable=False)
     parent = models.ForeignKey('self', models.SET_NULL, null=True, blank=True, verbose_name=_('parent'), related_name="children", editable=False)
     parent_number = models.IntegerField(_('parent number'), null=True, blank=True, db_index=True, editable=False)
 
@@ -40,8 +40,16 @@ class Book(models.Model):
     _on_track = models.IntegerField(null=True, blank=True, db_index=True, editable=False)
     dc_cover_image = models.ForeignKey(Image, blank=True, null=True,
         db_index=True, on_delete=models.SET_NULL, editable=False)
-    dc_slug = models.CharField(max_length=128, null=True, blank=True,
-            editable=False, db_index=True)
+    catalogue_book = models.ForeignKey(
+        'catalogue.Book',
+        models.DO_NOTHING,
+        to_field='slug',
+        null=True, blank=True,
+        db_constraint=False,
+        editable=False, db_index=True,
+        related_name='document_books',
+        related_query_name='document_book',
+    )
 
     class NoTextError(BaseException):
         pass
@@ -337,13 +345,15 @@ class Book(models.Model):
 
     def refresh_dc_cache(self):
         update = {
-            'dc_slug': None,
+            'catalogue_book_id': None,
             'dc_cover_image': None,
         }
 
         info = self.book_info()
+        print(info)
         if info is not None:
-            update['dc_slug'] = info.url.slug
+            update['catalogue_book_id'] = info.url.slug
+            print(info.url.slug)
             if info.cover_source:
                 try:
                     image = Image.objects.get(pk=int(info.cover_source.rstrip('/').rsplit('/', 1)[-1]))
@@ -352,6 +362,7 @@ class Book(models.Model):
                 else:
                     if info.cover_source == image.get_full_url():
                         update['dc_cover_image'] = image
+        print(update)
         Book.objects.filter(pk=self.pk).update(**update)
 
     def touch(self):