Isbns, and minor fixes.

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

diff --git a/src/isbn/models.py b/src/isbn/models.py
new file mode 100644 (file)
index 0000000..9db0513
--- /dev/null
+++ b/src/isbn/models.py
@@ -0,0 +1,150 @@
+from django.apps import apps
+from django.db import models
+from lxml import etree
+import requests
+from .product_forms import FORMS
+
+
+class IsbnPool(models.Model):
+    PURPOSE_GENERAL = 'GENERAL'
+    PURPOSE_WL = 'WL'
+    PURPOSE_CHOICES = (
+        (PURPOSE_WL, 'Wolne Lektury'),
+        (PURPOSE_GENERAL, 'Ogólne'),
+    )
+
+    prefix = models.CharField(max_length=10)
+    suffix_from = models.IntegerField()
+    suffix_to = models.IntegerField()
+    ref_from = models.IntegerField()
+    purpose = models.CharField(max_length=8, choices=PURPOSE_CHOICES)
+
+    def __str__(self):
+        return '-'.join((
+            self.prefix[:3],
+            self.prefix[3:5],
+            self.prefix[5:],
+            'X' * (12 - len(self.prefix)),
+            'X'
+        ))
+
+    @staticmethod
+    def check_digit(prefix12):
+        digits = [int(d) for d in prefix12]
+        return str((-sum(digits[0::2]) + 7 * sum(digits[1::2])) % 10)
+    
+    def get_code(self, suffix, dashes=False):
+        suffix_length = 12 - len(self.prefix)
+        suffix_str = f'{suffix:0{suffix_length}d}'
+        prefix12 = self.prefix + suffix_str
+        check_digit = self.check_digit(prefix12)
+        if dashes:
+            isbn = '-'.join((
+                self.prefix[:3],
+                self.prefix[3:5],
+                self.prefix[5:],
+                suffix_str,
+                check_digit
+            ))
+        else:
+            isbn = ''.join((
+                prefix12, check_digit
+            ))
+        return isbn
+
+    
+    @property
+    def size(self):
+        return self.suffix_to - self.suffix_from + 1
+
+    @property
+    def entries(self):
+        return self.isbn_set.count()
+    
+    @property
+    def fill_percentage(self):
+        return 100 * self.entries / self.size
+
+    def bn_record_id_for(self, suffix):
+        return self.ref_from + suffix
+    
+    def import_all_bn_data(self):
+        for suffix in range(self.suffix_from, self.suffix_to + 1):
+            print(suffix)
+            self.import_bn_data_for(suffix)
+    
+    def import_bn_data_for(self, suffix):
+        record_id = self.bn_record_id_for(suffix)
+        content = requests.get(
+            f'https://e-isbn.pl/IsbnWeb/record/export_onix.xml?record_id={record_id}').content
+        elem = etree.fromstring(content)
+        product = elem.find('{http://ns.editeur.org/onix/3.0/reference}Product')
+        if product is not None:
+            isbn, created = self.isbn_set.get_or_create(
+                suffix=suffix
+            )
+            isbn.bn_data = etree.tostring(product, pretty_print=True, encoding='unicode')
+            isbn.save(update_fields=['bn_data'])
+
+
+class Isbn(models.Model):
+    pool = models.ForeignKey(IsbnPool, models.PROTECT)
+    suffix = models.IntegerField()
+    datestamp = models.DateField(blank=True, null=True)
+    book = models.ForeignKey(
+        'catalogue.Book', models.PROTECT, null=True, blank=True
+    )
+    form = models.CharField(
+        max_length=32, choices=[
+            (form, form)
+            for form, config in FORMS
+        ], null=True, blank=True
+    )
+    bn_data = models.TextField(blank=True)
+    wl_data = models.TextField(blank=True)
+    notes = models.TextField(blank=True)
+
+    class Meta:
+        ordering = ['pool', 'suffix']
+        unique_together = ['pool', 'suffix']
+
+    def __str__(self):
+        return self.get_code(True)
+        
+    def get_code(self, dashes=True):
+        return self.pool.get_code(self.suffix, dashes=dashes)
+
+    @classmethod
+    def import_from_documents(cls):
+        Book = apps.get_model('documents', 'Book')
+        for book in Book.objects.all():
+            try:
+                catalogue_book = book.catalogue_book
+                if catalogue_book is None:
+                    continue
+            except:
+                continue
+            try:
+                meta = book.wldocument(publishable=False, librarian2=True).meta
+            except:
+                continue
+            for form in ('html', 'txt', 'pdf', 'epub', 'mobi'):
+                isbn = getattr(meta, f'isbn_{form}')
+                if isbn is not None:
+                    parts = isbn.split('-')
+                    assert parts[0] == 'ISBN'
+                    suffix = int(parts[-2])
+                    prefix = ''.join(parts[1:-2])
+                    pool = IsbnPool.objects.get(prefix=prefix)
+                    isbn, created = pool.isbn_set.get_or_create(
+                        suffix=suffix,
+                    )
+                    if isbn.book is None:
+                        isbn.book = catalogue_book
+                    else:
+                        assert isbn.book is catalogue_book
+                    if isbn.form is None:
+                        isbn.form = form
+                    else:
+                        assert isbn.form == form
+                    isbn.save(update_fields=['book', 'form'])