X-Git-Url: https://git.mdrn.pl/audio.git/blobdiff_plain/d7186e88fcc0cf5d5449a16d3907e01aaafc484e..e9e0d773e29755d3ab10945cc7582e491c0cdb9c:/src/youtube/models.py?ds=sidebyside diff --git a/src/youtube/models.py b/src/youtube/models.py index 423f338..6306a49 100644 --- a/src/youtube/models.py +++ b/src/youtube/models.py @@ -1,29 +1,35 @@ +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 ( - video_from_image, - cut_video, + concat_audio, concat_videos, + cut_video, get_duration, get_framerate, mux, + standardize_video, + video_from_image, ) +from .thumbnail import create_thumbnail class YouTube(models.Model): title_template = models.CharField(max_length=1024, blank=True) description_template = models.TextField(blank=True) category = models.IntegerField(null=True, blank=True) # get categories - intro_card = models.FileField(blank=True) - intro_card_duration = models.FloatField(null=True, blank=True) - card = models.FileField(blank=True) - loop_video = models.FileField(blank=True) - outro_card = models.FileField(blank=True) - outro_card_duration = models.FloatField(null=True, blank=True) + intro_flac = models.FileField(upload_to='youtube/intro_flac', blank=True) + 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) + genres = models.CharField(max_length=2048, blank=True) class Meta: verbose_name = _("YouTube configuration") @@ -59,65 +65,125 @@ class YouTube(models.Model): "POST", "https://www.googleapis.com/upload/youtube/v3/videos", params={'part': part}, - data=data, - media_data=f.read(), + json=data, + resumable_data=f.read(), ) data = response.json() audiobook.youtube_id = data['id'] audiobook.save(update_fields=['youtube_id']) + + self.update_thumbnail(audiobook) return response def prepare_file(self, input_path, output_path=None): - duration = get_duration(input_path) + audio = self.prepare_audio(input_path) + duration = self.get_duration(input_path) video = self.prepare_video(duration) - output = mux([video, input_path], output_path=output_path) + 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) + 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): + files = [] + if self.intro_flac: + files.append(self.intro_flac.path) + files.append(input_path) + if self.outro_flac: + files.append(self.outro_flac.path) + return concat_audio(files) + def prepare_video(self, duration): concat = [] + outro = [] delete = [] if self.loop_video: fps = get_framerate(self.loop_video.path) + loop_video = standardize_video(self.loop_video.path) else: fps = 25 loop_duration = duration - if self.intro_card and self.intro_card_duration: - loop_duration -= self.intro_card_duration - intro = video_from_image( - self.intro_card.path, self.intro_card_duration, fps=fps + for card in self.card_set.filter(duration__gt=0): + loop_duration -= card.duration + card_video = video_from_image( + card.image.path, card.duration, fps=fps ) - concat.append(intro) - delete.append(intro) - - if self.outro_card and self.outro_card_duration: - loop_duration -= self.outro_card_duration - outro = video_from_image( - self.outro_card.path, self.outro_card_duration, fps=fps - ) - concat.append(outro) - delete.append(outro) + (concat if card.order < 0 else outro).append(card_video) + delete.append(card_video) if self.loop_video: - loop_video_duration = get_duration(self.loop_video.path) + loop_video_duration = get_duration(loop_video) times_loop = int(loop_duration // loop_video_duration) leftover_duration = loop_duration % loop_video_duration - leftover = cut_video(self.loop_video.path, leftover_duration) - concat[1:1] = [self.loop_video.path] * times_loop + [leftover] + leftover = cut_video(loop_video, leftover_duration) + concat.extend([loop_video] * times_loop + [leftover]) delete.append(leftover) else: - leftover = video_from_image(self.card.path, loop_duration) - concat.insert(1, video_from_image(self.card.path, loop_duration, fps=fps)) + leftover = video_from_image(self.loop_card.path, loop_duration) + concat.append(video_from_image(self.loop_card.path, loop_duration, fps=fps)) delete.append(leftover) + concat.extend(outro) output = concat_videos(concat) for p in delete: unlink(p) + unlink(loop_video) return output # tags # license # selfDeclaredMadeForKids + + 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(), + ) + + def prepare_thumbnail(self, audiobook): + slug = audiobook.url.rstrip('/').rsplit('/', 1)[-1] + apidata = requests.get(f'https://wolnelektury.pl/api/books/{slug}/').json() + img = create_thumbnail( + self.thumbnail_template.path, + self.thumbnail_definition, + { + "author": ', '.join((a['name'] for a in apidata['authors'])), + "title": apidata['title'], + }, + lambda name: Font.objects.get(name=name).truetype.path + ) + buf = io.BytesIO() + img.save(buf, format='PNG') + return buf + + +class Card(models.Model): + youtube = models.ForeignKey(YouTube, models.CASCADE) + order = models.SmallIntegerField() + image = models.FileField(upload_to='youtube/card') + duration = models.FloatField() + + class Meta: + ordering = ('order', ) + + +class Font(models.Model): + name = models.CharField(max_length=255, unique=True) + truetype = models.FileField(upload_to='youtube/font') + + def __str__(self): + return self.name