Badges on audiobook list.

[audio.git] / src / youtube / models.py

diff --git a/src/youtube/models.py b/src/youtube/models.py
index 6306a49..5fe3c2b 100644 (file)
--- 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
 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,
 from apiclient import youtube_call
 from .utils import (
     concat_audio,


@@ -13,61 +12,75 @@ from .utils import (
     get_duration,
     get_framerate,
     mux,
     get_duration,
     get_framerate,
     mux,

+    standardize_audio,

     standardize_video,
     video_from_image,
 )
 from .thumbnail import create_thumbnail
 
 
     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)
 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)
     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")
 
     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(
             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(
             ),
             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())
 
         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'])
         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
 
         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
 
         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
         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 = []
         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:
         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_video(self, duration):
         concat = []
         outro = []


@@ -155,21 +185,20 @@ class YouTube(models.Model):
         )
 
     def prepare_thumbnail(self, audiobook):
         )
 
     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,
             {
         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
             },
             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)
 
 class Card(models.Model):
     youtube = models.ForeignKey(YouTube, models.CASCADE)