X-Git-Url: https://git.mdrn.pl/wolnelektury.git/blobdiff_plain/357027375ff8867f42ca34bcbfb5a78b5b185fc3..14eaee7aef2e0f89f0210f087d10eeaed0bedf43:/src/catalogue/models/bookmedia.py?ds=sidebyside diff --git a/src/catalogue/models/bookmedia.py b/src/catalogue/models/bookmedia.py index 1ef3fd7b9..14483eb83 100644 --- a/src/catalogue/models/bookmedia.py +++ b/src/catalogue/models/bookmedia.py @@ -1,4 +1,3 @@ -# -*- coding: utf-8 -*- # This file is part of Wolnelektury, licensed under GNU Affero GPLv3 or later. # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information. # @@ -7,14 +6,21 @@ import json from collections import namedtuple from django.db import models from django.utils.translation import ugettext_lazy as _ -import jsonfield -from fnpdjango.utils.text.slughifi import slughifi -from catalogue.fields import OverwritingFileField +from slugify import slugify +from mutagen import MutagenError + +from catalogue.fields import OverwriteStorage def _file_upload_to(i, _n): - return 'book/%(ext)s/%(name)s.%(ext)s' % { - 'ext': i.ext(), 'name': slughifi(i.name)} + name = i.book.slug + if i.index: + name += f'_{i.index:03d}' + if i.part_name: + name += f'_' + slugify(i.part_name) + ext = i.ext() + return f'book/{ext}/{name}.{ext}' + class BookMedia(models.Model): """Represents media attached to a book.""" @@ -24,38 +30,53 @@ class BookMedia(models.Model): ('ogg', FileFormat(name='Ogg Vorbis', ext='ogg')), ('daisy', FileFormat(name='DAISY', ext='daisy.zip')), ]) - format_choices = [(k, _('%s file' % t.name)) - for k, t in formats.items()] + format_choices = [(k, _('%s file' % t.name)) for k, t in formats.items()] type = models.CharField(_('type'), db_index=True, choices=format_choices, max_length=20) name = models.CharField(_('name'), max_length=512) - file = OverwritingFileField(_('file'), max_length=600, - upload_to=_file_upload_to) + part_name = models.CharField(_('part name'), default='', blank=True, max_length=512) + index = models.IntegerField(_('index'), default=0) + file = models.FileField(_('file'), max_length=600, upload_to=_file_upload_to, storage=OverwriteStorage()) uploaded_at = models.DateTimeField(_('creation date'), auto_now_add=True, editable=False, db_index=True) - extra_info = jsonfield.JSONField(_('extra information'), default={}, editable=False) - book = models.ForeignKey('Book', related_name='media') + project_description = models.CharField(max_length=2048, blank=True) + project_icon = models.CharField(max_length=2048, blank=True) + extra_info = models.TextField(_('extra information'), default='{}', editable=False) + book = models.ForeignKey('Book', models.CASCADE, related_name='media') source_sha1 = models.CharField(null=True, blank=True, max_length=40, editable=False) - def __unicode__(self): - return "%s (%s)" % (self.name, self.file.name.split("/")[-1]) + def __str__(self): + return self.file.name.split("/")[-1] class Meta: - ordering = ('type', 'name') - verbose_name = _('book media') + ordering = ('type', 'index') + verbose_name = _('book media') verbose_name_plural = _('book media') app_label = 'catalogue' - def save(self, *args, **kwargs): + def get_extra_info_json(self): + return json.loads(self.extra_info or '{}') + + def save(self, parts_count=None, *args, **kwargs): from catalogue.utils import ExistingFile, remove_zip + if not parts_count: + parts_count = 1 + BookMedia.objects.filter(book=self.book, type=self.type).exclude(pk=self.pk).count() + if parts_count == 1: + self.name = self.book.pretty_title() + else: + no = ('%02d' if parts_count < 100 else '%03d') % self.index + self.name = '%s. %s' % (no, self.book.pretty_title()) + if self.part_name: + self.name += ', ' + self.part_name + try: old = BookMedia.objects.get(pk=self.pk) except BookMedia.DoesNotExist: old = None else: # if name changed, change the file name, too - if slughifi(self.name) != slughifi(old.name): - self.file.save(None, ExistingFile(self.file.path), save=False, leave=True) + if slugify(self.name) != slugify(old.name): + self.file.save(None, ExistingFile(self.file.path), save=False) super(BookMedia, self).save(*args, **kwargs) @@ -64,12 +85,9 @@ class BookMedia(models.Model): remove_zip("%s_%s" % (old.book.slug, old.type)) remove_zip("%s_%s" % (self.book.slug, self.type)) - extra_info = self.extra_info - if isinstance(extra_info, basestring): - # Walkaround for weird jsonfield 'no-decode' optimization. - extra_info = json.loads(extra_info) + extra_info = self.get_extra_info_json() extra_info.update(self.read_meta()) - self.extra_info = extra_info + self.extra_info = json.dumps(extra_info) self.source_sha1 = self.read_source_sha1(self.file.path, self.type) return super(BookMedia, self).save(*args, **kwargs) @@ -80,31 +98,35 @@ class BookMedia(models.Model): import mutagen from mutagen import id3 - artist_name = director_name = project = funded_by = '' + artist_name = director_name = project = funded_by = license = '' if self.type == 'mp3': try: audio = id3.ID3(self.file.path) artist_name = ', '.join(', '.join(tag.text) for tag in audio.getall('TPE1')) director_name = ', '.join(', '.join(tag.text) for tag in audio.getall('TPE3')) - project = ", ".join([t.data for t in audio.getall('PRIV') - if t.owner == 'wolnelektury.pl?project']) - funded_by = ", ".join([t.data for t in audio.getall('PRIV') - if t.owner == 'wolnelektury.pl?funded_by']) - except: + license = ', '.join(tag.url for tag in audio.getall('WCOP')) + project = ", ".join([ + t.data.decode('utf-8') for t in audio.getall('PRIV') + if t.owner == 'wolnelektury.pl?project']) + funded_by = ", ".join([ + t.data.decode('utf-8') for t in audio.getall('PRIV') + if t.owner == 'wolnelektury.pl?funded_by']) + except MutagenError: pass elif self.type == 'ogg': try: audio = mutagen.File(self.file.path) artist_name = ', '.join(audio.get('artist', [])) director_name = ', '.join(audio.get('conductor', [])) + license = ', '.join(audio.get('license', [])) project = ", ".join(audio.get('project', [])) funded_by = ", ".join(audio.get('funded_by', [])) - except: + except (MutagenError, AttributeError): pass else: return {} return {'artist_name': artist_name, 'director_name': director_name, - 'project': project, 'funded_by': funded_by} + 'project': project, 'funded_by': funded_by, 'license': license} def ext(self): return self.formats[self.type].ext @@ -120,15 +142,26 @@ class BookMedia(models.Model): if filetype == 'mp3': try: audio = id3.ID3(filepath) - return [t.data for t in audio.getall('PRIV') + return [t.data.decode('utf-8') for t in audio.getall('PRIV') if t.owner == 'wolnelektury.pl?flac_sha1'][0] - except: + except (MutagenError, IndexError): return None elif filetype == 'ogg': try: audio = mutagen.File(filepath) return audio.get('flac_sha1', [None])[0] - except: + except (MutagenError, AttributeError, IndexError): return None else: return None + + @property + def director(self): + return self.get_extra_info_json().get('director_name', None) + + @property + def artist(self): + return self.get_extra_info_json().get('artist_name', None) + + def file_url(self): + return self.file.url