X-Git-Url: https://git.mdrn.pl/audio.git/blobdiff_plain/0678dd50bd89ef1ad0f0d95c1e7907e31187a0d3..5715ba5816c2773949689461f715a08caa8c5d11:/src/youtube/models.py?ds=inline diff --git a/src/youtube/models.py b/src/youtube/models.py index 39a1d2d..35899b6 100644 --- a/src/youtube/models.py +++ b/src/youtube/models.py @@ -1,9 +1,9 @@ import io from os import unlink -from tempfile import NamedTemporaryFile from django.db import models from django.utils.translation import gettext_lazy as _ from django.template import Template, Context +import requests from apiclient import youtube_call from .utils import ( concat_audio, @@ -23,6 +23,7 @@ YOUTUBE_TITLE_LIMIT = 100 class YouTube(models.Model): + name = models.CharField(max_length=255) title_template = models.CharField(max_length=1024, blank=True) description_template = models.TextField(blank=True) category = models.IntegerField(null=True, blank=True, choices=[ @@ -32,8 +33,6 @@ class YouTube(models.Model): outro_flac = models.FileField(upload_to='youtube/outro_flac', blank=True) loop_card = models.FileField(upload_to='youtube/card', blank=True) loop_video = models.FileField(upload_to='youtube/loop_video', blank=True) - thumbnail_template = models.FileField(upload_to='youtube/thumbnail', blank=True) - thumbnail_definition = models.TextField(blank=True) privacy_status = models.CharField(max_length=16, choices=[ ('public', _('public')), ('unlisted', _('unlisted')), @@ -45,6 +44,9 @@ class YouTube(models.Model): verbose_name = _("YouTube configuration") verbose_name_plural = _("YouTube configurations") + def __str__(self): + return self.name + def get_context(self, audiobook): return Context(dict( audiobook=audiobook, @@ -97,30 +99,33 @@ class YouTube(models.Model): "https://www.googleapis.com/youtube/v3/videos", params={"part": part}, json=data - ) + ) - def prepare_file(self, input_path, output_path=None): - audio = self.prepare_audio(input_path) - duration = self.get_duration(input_path) + def prepare_file(self, input_paths, output_path=None): + audio = self.prepare_audio(input_paths) + duration = self.get_duration(input_paths) video = self.prepare_video(duration) output = mux([video, audio], output_path=output_path) unlink(audio) unlink(video) return output - def get_duration(self, input_path): - d = get_duration(input_path) + def get_duration(self, input_paths): + d = 0 + for input_path in input_paths: + d += get_duration(input_path) if self.intro_flac: d += get_duration(self.intro_flac.path) if self.outro_flac: d += get_duration(self.outro_flac.path) return d - - def prepare_audio(self, input_path): + + def prepare_audio(self, input_paths): files = [] if self.intro_flac: files.append(standardize_audio(self.intro_flac.path)) - files.append(standardize_audio(input_path, cache=False)) + for input_path in input_paths: + files.append(standardize_audio(input_path, cache=False)) if self.outro_flac: files.append(standardize_audio(self.outro_flac.path)) output = concat_audio(files) @@ -128,7 +133,6 @@ class YouTube(models.Model): unlink(d) return output - def prepare_video(self, duration): concat = [] outro = [] @@ -175,27 +179,22 @@ class YouTube(models.Model): def update_thumbnail(self, audiobook): thumbnail = self.prepare_thumbnail(audiobook) - response = youtube_call( - "POST", - "https://www.googleapis.com/upload/youtube/v3/thumbnails/set", - params={'videoId': audiobook.youtube_id}, - data=thumbnail.getvalue(), - ) + if thumbnail is not None: + response = youtube_call( + "POST", + "https://www.googleapis.com/upload/youtube/v3/thumbnails/set", + params={'videoId': audiobook.youtube_id}, + data=thumbnail.getvalue(), + ) def prepare_thumbnail(self, audiobook): - img = create_thumbnail( - self.thumbnail_template.path, - self.thumbnail_definition, - { - "author": ', '.join((a['name'] for a in audiobook.book['authors'])), - "title": audiobook.book['title'], - }, - lambda name: Font.objects.get(name=name).truetype.path - ) - buf = io.BytesIO() - img.save(buf, format='PNG') - return buf - + for thumbnail_template in ThumbnailTemplate.objects.filter(is_active=True).order_by('order'): + if not thumbnail_template.is_for_audiobook(audiobook): + continue + thumbnail = thumbnail_template.generate(audiobook) + if thumbnail is not None: + return thumbnail + class Card(models.Model): youtube = models.ForeignKey(YouTube, models.CASCADE) @@ -213,3 +212,71 @@ class Font(models.Model): def __str__(self): return self.name + + +class ThumbnailTemplate(models.Model): + youtube = models.ForeignKey(YouTube, models.CASCADE) + order = models.SmallIntegerField() + is_active = models.BooleanField() + background = models.FileField(upload_to='youtube/thumbnail') + definition = models.TextField() + authors = models.CharField(max_length=255, blank=True) + epochs = models.CharField(max_length=255, blank=True) + kinds = models.CharField(max_length=255, blank=True) + genres = models.CharField(max_length=255, blank=True) + collections = models.CharField(max_length=255, blank=True) + + class Meta: + ordering = ('order', ) + + def generate(self, audiobook): + try: + title = audiobook.book['title'] + if audiobook.book.get('parent'): + parent_title = audiobook.book['parent']['title'] + if not title.startswith(parent_title): + title = ", ".join((parent_title, title)) + + img = create_thumbnail( + self.background.path, + self.definition, + { + "author": ', '.join((a['name'] for a in audiobook.book['authors'])), + "title": title, + "part": (audiobook.youtube_volume or audiobook.part_name).strip() if audiobook.youtube_volume_count > 1 else '', + }, + lambda name: Font.objects.get(name=name).truetype.path + ) + except Exception as e: + print(e) + return + else: + buf = io.BytesIO() + img.save(buf, format='PNG') + return buf + + def is_for_audiobook(self, audiobook): + for category in 'authors', 'epochs', 'kinds', 'genres': + if getattr(self, category): + book_slugs = set([g['slug'] for g in audiobook.book[category]]) + template_slugs = set([g.strip() for g in getattr(self, category).split(',')]) + if not book_slugs.intersection(template_slugs): + return False + + if self.collections: + template_collections = set([g.strip() for g in self.collections.split(',')]) + in_any = False + for collection in template_collections: + apidata = requests.get( + f'https://wolnelektury.pl/api/collections/{collection}/' + ).json() + for book in apidata['books']: + if book['slug'] == audiobook.slug: + in_any = True + break + if in_any: + break + if not in_any: + return False + + return True