src/cover/models.py

   1 # This file is part of FNP-Redakcja, licensed under GNU Affero GPLv3 or later.
   2 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
   3 #
   4 from io import BytesIO
   5 from django.core.files.base import ContentFile
   6 from django.core.files.storage import FileSystemStorage
   7 from django.db import models
   8 from django.db.models.signals import post_save
   9 from django.dispatch import receiver
  10 from django.urls import reverse
  11 from django.utils.translation import gettext_lazy as _
  12 from django.contrib.sites.models import Site
  13 from PIL import Image as PILImage
  14 from librarian.dcparser import BookInfo
  15 from librarian.meta.types.person import Person
  16 from librarian.cover import make_cover
  17 from cover.utils import URLOpener
  18
  19
  20 class OverwriteStorage(FileSystemStorage):
  21
  22     def get_available_name(self, name, max_length=None):
  23         self.delete(name)
  24         return name
  25
  26
  27 class Image(models.Model):
  28     title = models.CharField(max_length=255, verbose_name=_('title'))
  29     author = models.CharField(max_length=255, verbose_name=_('author'))
  30     license_name = models.CharField(max_length=255, verbose_name=_('license name'))
  31     license_url = models.URLField(max_length=255, blank=True, verbose_name=_('license URL'))
  32     source_url = models.URLField(verbose_name=_('source URL'), null=True, blank=True)
  33     download_url = models.URLField(max_length=4096, unique=True, verbose_name=_('image download URL'), null=True, blank=True)
  34     file = models.ImageField(
  35         upload_to='cover/image',
  36         max_length=1024,
  37         storage=OverwriteStorage(),
  38         verbose_name=_('file')
  39     )
  40     use_file = models.ImageField(
  41         upload_to='cover/use',
  42         storage=OverwriteStorage(),
  43         editable=True,
  44         verbose_name=_('file for use')
  45     )
  46
  47     example = models.ImageField(
  48         upload_to='cover/example',
  49         storage=OverwriteStorage(),
  50         editable=False,
  51     )
  52
  53     cut_top = models.IntegerField(default=0, )
  54     cut_bottom = models.IntegerField(default=0)
  55     cut_left = models.IntegerField(default=0)
  56     cut_right = models.IntegerField(default=0)
  57
  58     class Meta:
  59         verbose_name = _('cover image')
  60         verbose_name_plural = _('cover images')
  61
  62     def __str__(self):
  63         return u"%s - %s" % (self.author, self.title)
  64
  65     def save(self, **kwargs):
  66         super().save(**kwargs)
  67         img = self.file
  68         if self.cut_top or self.cut_bottom or self.cut_left or self.cut_right:
  69             img = PILImage.open(img)
  70             img = img.crop((
  71                 self.cut_left,
  72                 self.cut_top,
  73                 img.size[0] - self.cut_right,
  74                 img.size[1] - self.cut_bottom,
  75             ))
  76             buffer = BytesIO()
  77             img.save(
  78                 buffer,
  79                 format='JPEG',
  80             )
  81             img = ContentFile(buffer.getvalue())
  82         self.use_file.save(
  83             "%d.jpg" % self.pk,
  84             img,
  85             save=False
  86         )
  87
  88         super().save(update_fields=['use_file'])
  89
  90         self.example.save(
  91             "%d.jpg" % self.pk,
  92             ContentFile(self.build_example().get_bytes()),
  93             save=False
  94         )
  95         super().save(update_fields=['example'])
  96
  97
  98     def build_example(self):
  99         class A:
 100             pass
 101         info = A()
 102         info.authors = []
 103         info.translators = []
 104         info.cover_class = None
 105         info.cover_box_position = None
 106         info.title = '?'
 107         info.cover_url = 'file://' + self.use_file.path
 108         return make_cover(info, width=200).output_file()
 109
 110     def get_absolute_url(self):
 111         return reverse('cover_image', args=[self.id])
 112
 113     def get_full_url(self):
 114         return "http://%s%s" % (Site.objects.get_current().domain, self.get_absolute_url())
 115
 116     def cut_percentages(self):
 117         img = PILImage.open(self.file)
 118         max_w, max_h = 600, 600
 119         w, h = img.size
 120         scale = min(max_w / w, max_h / h)
 121         ws, hs = round(w * scale), round(h * scale)
 122
 123         return {
 124             'left': 100 * self.cut_left / w,
 125             'right': 100 * self.cut_right / w,
 126             'top': 100 * self.cut_top / h,
 127             'bottom': 100 * self.cut_bottom / h,
 128             'width': ws,
 129             'height': hs,
 130             'th': f'{ws}x{hs}',
 131         }
 132
 133     @property
 134     def etag(self):
 135         return f'{self.cut_top}.{self.cut_bottom}.{self.cut_left}.{self.cut_right}'
 136
 137     @property
 138     def attribution(self):
 139         pieces = []
 140         if self.title:
 141             pieces.append(self.title)
 142         if self.author:
 143             pieces.append(self.author)
 144         if self.license_name:
 145             pieces.append(self.license_name)
 146         if self.source_url:
 147             pieces.append(self.source_url.split('/')[2])
 148         return ', '.join(pieces)
 149
 150
 151
 152 @receiver(post_save, sender=Image)
 153 def download_image(sender, instance, **kwargs):
 154     if instance.pk and not instance.file:
 155         t = URLOpener().open(instance.download_url).read()
 156         instance.file.save("%d.jpg" % instance.pk, ContentFile(t))