From 4c89a14f13fe5d7dd3a6d2488dd08067a652c10d Mon Sep 17 00:00:00 2001 From: Radek Czajka Date: Wed, 20 May 2020 13:54:14 +0200 Subject: [PATCH] More YT details. --- src/archive/models.py | 8 +++ .../migrations/0010_auto_20200520_1353.py | 24 +++++++ src/youtube/models.py | 62 ++++++++++++------- 3 files changed, 71 insertions(+), 23 deletions(-) create mode 100644 src/youtube/migrations/0010_auto_20200520_1353.py diff --git a/src/archive/models.py b/src/archive/models.py index 34fc1f4..4a805f2 100644 --- a/src/archive/models.py +++ b/src/archive/models.py @@ -3,8 +3,10 @@ import os.path from django.db import models from time import sleep +from django.utils.functional import cached_property from django.utils.translation import gettext_lazy as _ from django_pglocks import advisory_lock +import requests from archive.constants import status from archive.settings import FILES_SAVE_PATH, ADVERT, LICENSE, ORGANIZATION, PROJECT from archive.utils import OverwriteStorage, sha1_file @@ -153,3 +155,9 @@ class Audiobook(models.Model): tags['flac_sha1'] = self.source_sha1 return tags + @cached_property + def book(self): + slug = self.url.rstrip('/').rsplit('/', 1)[-1] + apidata = requests.get(f'https://wolnelektury.pl/api/books/{slug}/').json() + return apidata + diff --git a/src/youtube/migrations/0010_auto_20200520_1353.py b/src/youtube/migrations/0010_auto_20200520_1353.py new file mode 100644 index 0000000..adc1bae --- /dev/null +++ b/src/youtube/migrations/0010_auto_20200520_1353.py @@ -0,0 +1,24 @@ +# Generated by Django 3.0.4 on 2020-05-20 13:53 + +from django.db import migrations, models + + +class Migration(migrations.Migration): + + dependencies = [ + ('youtube', '0009_auto_20200518_1656'), + ] + + operations = [ + migrations.AddField( + model_name='youtube', + name='privacy_status', + field=models.CharField(choices=[('public', 'public'), ('unlisted', 'unlisted'), ('private', 'private')], default='unlisted', max_length=16), + preserve_default=False, + ), + migrations.AlterField( + model_name='youtube', + name='category', + field=models.IntegerField(blank=True, choices=[(27, 'Edukacja')], null=True), + ), + ] diff --git a/src/youtube/models.py b/src/youtube/models.py index 6306a49..ecc05e4 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, @@ -22,42 +21,50 @@ 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 + 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)) - data = dict( + def get_title(self, audiobook): + return Template(self.title_template).render(self.get_context(audiobook)) + + 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' ), status=dict( - privacyStatus=privacy, - # license - # selfDeclaredMadeForKids + privacyStatus=self.privacy_status, ), - # recordingDetails=dict( - # recordingDate - # ), ) + + def publish(self, audiobook, path): + data = self.get_data() part = ",".join(data.keys()) with open(path, "rb") as f: @@ -75,6 +82,17 @@ class YouTube(models.Model): self.update_thumbnail(audiobook) return response + def update_data(self, audiobook): + data = self.get_data() + 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_path, output_path=None): audio = self.prepare_audio(input_path) duration = self.get_duration(input_path) @@ -155,14 +173,12 @@ 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'], }, lambda name: Font.objects.get(name=name).truetype.path ) -- 2.20.1