X-Git-Url: https://git.mdrn.pl/wolnelektury.git/blobdiff_plain/5aa7e3bcddd7bac5e9f553e7f6d22e20799293d7..f0c1ca86983a56a80796a10499033304dd3a8c0e:/src/catalogue/models/bookmedia.py diff --git a/src/catalogue/models/bookmedia.py b/src/catalogue/models/bookmedia.py index de493d747..acb1881e5 100644 --- a/src/catalogue/models/bookmedia.py +++ b/src/catalogue/models/bookmedia.py @@ -1,19 +1,24 @@ -# This file is part of Wolnelektury, licensed under GNU Affero GPLv3 or later. -# Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information. +# This file is part of Wolne Lektury, licensed under GNU Affero GPLv3 or later. +# Copyright © Fundacja Wolne Lektury. See NOTICE for more information. # from collections import OrderedDict import json from collections import namedtuple from django.db import models -from django.utils.translation import ugettext_lazy as _ from slugify import slugify -from mutagen import MutagenError - -from catalogue.fields import OverwriteStorage +import mutagen +from mutagen import id3 +from fnpdjango.storage import BofhFileSystemStorage def _file_upload_to(i, _n): - return 'book/%(ext)s/%(name)s.%(ext)s' % {'ext': i.ext(), 'name': slugify(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): @@ -23,31 +28,47 @@ class BookMedia(models.Model): ('mp3', FileFormat(name='MP3', ext='mp3')), ('ogg', FileFormat(name='Ogg Vorbis', ext='ogg')), ('daisy', FileFormat(name='DAISY', ext='daisy.zip')), + ('audio.epub', FileFormat(name='EPUB+audio', ext='audio.epub')), + ('sync', FileFormat(name='sync', ext='sync.txt')), ]) - 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) - 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 = models.TextField(_('extra information'), default='{}', editable=False) + format_choices = [(k, 'plik %s' % t.name) for k, t in formats.items()] + + type = models.CharField('typ', db_index=True, choices=format_choices, max_length=20) + name = models.CharField('nazwa', max_length=512) + part_name = models.CharField('nazwa części', default='', blank=True, max_length=512) + index = models.IntegerField('indeks', default=0) + file = models.FileField('plik', max_length=600, upload_to=_file_upload_to, storage=BofhFileSystemStorage()) + duration = models.FloatField(null=True, blank=True) + uploaded_at = models.DateTimeField('data utworzenia', auto_now_add=True, editable=False, db_index=True) + project_description = models.CharField(max_length=2048, blank=True) + project_icon = models.CharField(max_length=2048, blank=True) + extra_info = models.TextField('dodatkowe informacje', 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 __str__(self): - return "%s (%s)" % (self.name, self.file.name.split("/")[-1]) + return self.file.name.split("/")[-1] class Meta: - ordering = ('type', 'name') - verbose_name = _('book media') - verbose_name_plural = _('book media') + ordering = ('type', 'index') + verbose_name = 'media książki' + verbose_name_plural = 'media książek' app_label = 'catalogue' def get_extra_info_json(self): return json.loads(self.extra_info or '{}') + def get_nice_filename(self): + parts_count = 1 + type(self).objects.filter(book=self.book, type=self.type).exclude(pk=self.pk).count() + + name = self.book.slug + if parts_count > 0: + name += f'_{self.index:03d}' + if self.part_name: + name += f'_' + slugify(self.part_name) + ext = self.ext() + return f'{name}.{ext}' + def save(self, parts_count=None, *args, **kwargs): from catalogue.utils import ExistingFile, remove_zip @@ -65,13 +86,9 @@ class BookMedia(models.Model): old = BookMedia.objects.get(pk=self.pk) except BookMedia.DoesNotExist: old = None - else: - # if name changed, change the file name, too - if slugify(self.name) != slugify(old.name): - self.file.save(None, ExistingFile(self.file.path), save=False) super(BookMedia, self).save(*args, **kwargs) - + # remove the zip package for book with modified media if old: remove_zip("%s_%s" % (old.book.slug, old.type)) @@ -81,16 +98,20 @@ class BookMedia(models.Model): extra_info.update(self.read_meta()) self.extra_info = json.dumps(extra_info) self.source_sha1 = self.read_source_sha1(self.file.path, self.type) + self.duration = self.read_duration() return super(BookMedia, self).save(*args, **kwargs) + def read_duration(self): + try: + return mutagen.File(self.file.path).info.length + except: + return None + def read_meta(self): """ Reads some metadata from the audiobook. """ - 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) @@ -103,7 +124,7 @@ class BookMedia(models.Model): funded_by = ", ".join([ t.data.decode('utf-8') for t in audio.getall('PRIV') if t.owner == 'wolnelektury.pl?funded_by']) - except MutagenError: + except mutagen.MutagenError: pass elif self.type == 'ogg': try: @@ -113,7 +134,7 @@ class BookMedia(models.Model): license = ', '.join(audio.get('license', [])) project = ", ".join(audio.get('project', [])) funded_by = ", ".join(audio.get('funded_by', [])) - except (MutagenError, AttributeError): + except (mutagen.MutagenError, AttributeError): pass else: return {} @@ -128,21 +149,18 @@ class BookMedia(models.Model): """ Reads source file SHA1 from audiobok metadata. """ - import mutagen - from mutagen import id3 - if filetype == 'mp3': try: audio = id3.ID3(filepath) return [t.data.decode('utf-8') for t in audio.getall('PRIV') if t.owner == 'wolnelektury.pl?flac_sha1'][0] - except (MutagenError, IndexError): + except (mutagen.MutagenError, IndexError): return None elif filetype == 'ogg': try: audio = mutagen.File(filepath) return audio.get('flac_sha1', [None])[0] - except (MutagenError, AttributeError, IndexError): + except (mutagen.MutagenError, AttributeError, IndexError): return None else: return None