X-Git-Url: https://git.mdrn.pl/audio.git/blobdiff_plain/7a7629297f382c47a84ec5786c8cdd52224887ae..7a728f387a0969d0ca4cfc7b7877bf4fa255eef9:/src/youtube/models.py diff --git a/src/youtube/models.py b/src/youtube/models.py index 6306a49..5fe3c2b 100644 --- a/src/youtube/models.py +++ b/src/youtube/models.py @@ -4,7 +4,6 @@ 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, @@ -13,61 +12,75 @@ from .utils import ( get_duration, get_framerate, mux, + standardize_audio, standardize_video, video_from_image, ) from .thumbnail import create_thumbnail +YOUTUBE_TITLE_LIMIT = 100 + + 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 + category = models.IntegerField(null=True, blank=True, choices=[ + (27, 'Edukacja'), + ]) 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) + privacy_status = models.CharField(max_length=16, choices=[ + ('public', _('public')), + ('unlisted', _('unlisted')), + ('private', _('private')), + ]) genres = models.CharField(max_length=2048, blank=True) class Meta: verbose_name = _("YouTube configuration") verbose_name_plural = _("YouTube configurations") - def publish(self, audiobook, path): - ctx = Context(dict(audiobook=audiobook)) - description = Template(self.description_template).render(ctx) - title = Template(self.title_template).render(ctx) - privacy = 'private' + def get_context(self, audiobook): + return Context(dict( + audiobook=audiobook, + )) + + def get_description(self, audiobook): + return Template(self.description_template).render(self.get_context(audiobook)) + + def get_title(self, audiobook): + return Template(self.title_template).render(self.get_context(audiobook))[:YOUTUBE_TITLE_LIMIT] - data = dict( + def get_data(self, audiobook): + return dict( snippet=dict( - title=title, - description=description, - # tags=tags, - # categoryId=category, - # defaultLanguage + title=self.get_title(audiobook), + description=self.get_description(audiobook), + categoryId=self.category, + defaultLanguage='pl', + defaultAudioLanguage='pl', ), status=dict( - privacyStatus=privacy, - # license - # selfDeclaredMadeForKids + privacyStatus=self.privacy_status, ), - # recordingDetails=dict( - # recordingDate - # ), ) + + def publish(self, audiobook, path): + data = self.get_data(audiobook) part = ",".join(data.keys()) - with open(path, "rb") as f: - response = youtube_call( - "POST", - "https://www.googleapis.com/upload/youtube/v3/videos", - params={'part': part}, - json=data, - resumable_data=f.read(), - ) + response = youtube_call( + "POST", + "https://www.googleapis.com/upload/youtube/v3/videos", + params={'part': part}, + json=data, + resumable_file_path=path, + ) data = response.json() audiobook.youtube_id = data['id'] audiobook.save(update_fields=['youtube_id']) @@ -75,32 +88,49 @@ class YouTube(models.Model): self.update_thumbnail(audiobook) return response - def prepare_file(self, input_path, output_path=None): - audio = self.prepare_audio(input_path) - duration = self.get_duration(input_path) + def update_data(self, audiobook): + data = self.get_data(audiobook) + data['id'] = audiobook.youtube_id + part = ",".join(data.keys()) + youtube_call( + "PUT", + "https://www.googleapis.com/youtube/v3/videos", + params={"part": part}, + json=data + ) + + 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(self.intro_flac.path) - files.append(input_path) + files.append(standardize_audio(self.intro_flac.path)) + for input_path in input_paths: + files.append(standardize_audio(input_path, cache=False)) if self.outro_flac: - files.append(self.outro_flac.path) - return concat_audio(files) - + files.append(standardize_audio(self.outro_flac.path)) + output = concat_audio(files) + for d in files: + unlink(d) + return output + def prepare_video(self, duration): concat = [] outro = [] @@ -155,21 +185,20 @@ class YouTube(models.Model): ) 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'], + "author": ', '.join((a['name'] for a in audiobook.book['authors'])), + "title": audiobook.book['title'], + "part": (audiobook.youtube_volume or audiobook.part_name).strip(), }, 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)