Start replacing Piston in OAuth flow with OAuthLib.
[wolnelektury.git] / src / isbn / models.py
1 # -*- coding: utf-8 -*-
2 # This file is part of Wolnelektury, licensed under GNU Affero GPLv3 or later.
3 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
4 #
5 from django.db import models
6 from django.db.models import F
7 from jsonfield import JSONField
8
9
10 class ISBNPool(models.Model):
11     PURPOSE_WL = 'WL'
12     PURPOSE_FNP = 'FNP'
13     PURPOSE_CHOICES = (
14         (PURPOSE_WL, 'Wolne Lektury'),
15         (PURPOSE_FNP, 'Fundacja Nowoczesna Polska'),
16     )
17
18     prefix = models.CharField(max_length=10)
19     suffix_from = models.IntegerField()
20     suffix_to = models.IntegerField()
21     ref_from = models.IntegerField()
22     next_suffix = models.IntegerField()
23     purpose = models.CharField(max_length=4, choices=PURPOSE_CHOICES)
24
25     def __unicode__(self):
26         return self.prefix
27
28     @classmethod
29     def active_pool(cls, purpose):
30         pools = cls.objects.filter(purpose=purpose)
31         pools = pools.exclude(next_suffix__gt=F('suffix_to'))
32         if len(pools) == 1:
33             return pools.get()
34         else:
35             pools.exclude(next_suffix=F('suffix_from'))
36             return pools.get()
37
38     @staticmethod
39     def check_digit(prefix12):
40         digits = [int(d) for d in prefix12]
41         return str((-sum(digits[0::2]) + 7 * sum(digits[1::2])) % 10)
42
43     def isbn(self, suffix, dashes=False):
44         prefix_length = len(self.prefix)
45         suffix_length = 12 - prefix_length
46         suffix_str = ('%%0%dd' % suffix_length) % suffix
47         prefix12 = self.prefix + suffix_str
48         if dashes:
49             prefix12_final = '%s-%s-%s-%s-' % (self.prefix[:3], self.prefix[3:5], self.prefix[5:], suffix_str)
50         else:
51             prefix12_final = prefix12
52         return prefix12_final + self.check_digit(prefix12)
53
54
55 class ONIXRecord(models.Model):
56     isbn_pool = models.ForeignKey(ISBNPool)
57     datestamp = models.DateField(auto_now=True)
58     suffix = models.IntegerField()
59     product_form = models.CharField(max_length=4)
60     product_form_detail = models.CharField(max_length=8, blank=True)
61     title = models.CharField(max_length=256)
62     part_number = models.CharField(max_length=64, blank=True)
63     contributors = JSONField()  # roles, names, optional: ISNI, date of birth/death
64     edition_type = models.CharField(max_length=4)
65     edition_number = models.IntegerField(default=1)
66     language = models.CharField(max_length=4)
67     imprint = models.CharField(max_length=256)
68     publishing_date = models.DateField()
69     dc_slug = models.CharField(max_length=256, default='', db_index=True)
70
71     class Meta:
72         ordering = ['isbn_pool__id', 'suffix']
73         unique_together = ['isbn_pool', 'suffix']
74
75     @classmethod
76     def new_record(cls, purpose, data):
77         pool = ISBNPool.active_pool(purpose)
78         fields = {
79             'isbn_pool': pool,
80             'suffix': pool.next_suffix,
81         }
82         fields_to_copy = [
83             'product_form',
84             'product_form_detail',
85             'title',
86             'part_number',
87             'contributors',  # ???
88             'edition_type',
89             'edition_number',
90             'language',
91             'imprint',
92             'publishing_date',
93             'dc_slug',
94         ]
95         for field in fields_to_copy:
96             if field in data:
97                 fields[field] = data[field]
98         cls.objects.create(**fields)
99         pool.next_suffix += 1
100         pool.save()
101
102     def isbn(self, dashes=False):
103         return self.isbn_pool.isbn(self.suffix, dashes=dashes)
104
105     def reference(self):
106         return 'pl-eisbn-%s' % (self.isbn_pool.ref_from + self.suffix)