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,
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'])
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 = []
)
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)
fnp / audio.git / blobdiff