Configurable license.

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

diff --git a/src/youtube/models.py b/src/youtube/models.py
index ecc05e4..39a1d2d 100644 (file)
--- a/src/youtube/models.py
+++ b/src/youtube/models.py
@@ -12,12 +12,16 @@ 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)
@@ -42,13 +46,15 @@ class YouTube(models.Model):
         verbose_name_plural = _("YouTube configurations")
 
     def get_context(self, audiobook):
-        return Context(dict(audiobook=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))
+        return Template(self.title_template).render(self.get_context(audiobook))[:YOUTUBE_TITLE_LIMIT]
 
     def get_data(self, audiobook):
         return dict(
@@ -56,7 +62,8 @@ class YouTube(models.Model):
                 title=self.get_title(audiobook),
                 description=self.get_description(audiobook),
                 categoryId=self.category,
-                defaultLanguage='pl'
+                defaultLanguage='pl',
+                defaultAudioLanguage='pl',
             ),
             status=dict(
                 privacyStatus=self.privacy_status,
@@ -64,17 +71,16 @@ class YouTube(models.Model):
         )
 
     def publish(self, audiobook, path):
-        data = self.get_data()
+        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'])
@@ -83,7 +89,7 @@ class YouTube(models.Model):
         return response
 
     def update_data(self, audiobook):
-        data = self.get_data()
+        data = self.get_data(audiobook)
         data['id'] = audiobook.youtube_id
         part = ",".join(data.keys())
         youtube_call(
@@ -113,11 +119,15 @@ class YouTube(models.Model):
     def prepare_audio(self, input_path):
         files = []
         if self.intro_flac:
-            files.append(self.intro_flac.path)
-        files.append(input_path)
+            files.append(standardize_audio(self.intro_flac.path))
+        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 = []