add generate video duration

apps/core/tasks.py

@@ -5,7 +5,12 @@ from django.conf import settings
 
 from .models import Video, READY, FAIL
 from .s3_uploader import upload_file
-from .utils import add_video_to_playlist, generate_thumbnail, download_file
+from .utils import (
+    add_video_to_playlist,
+    generate_thumbnail,
+    download_file,
+    get_video_duration,
+)
 
 from cp_video.celery import app
 from cp_video.utils import asset_upload
@@ -32,6 +37,7 @@ def send_video_to_s3(video_id=None):
         local_video.local_file = None
         local_video.status = READY
         local_video.s3_file = upload_file(video_file, key, 'video/mp4')
+        local_video.duration = get_video_duration(video_name)
 
         try:
             log.info('Generating thumbnail...')

apps/core/utils.py

-import requests
-import logging
+import subprocess, json, requests, logging
 
 from django.contrib.contenttypes.models import ContentType
 
@@ -50,6 +49,22 @@ def generate_thumbnail(video_path, video_name):
     return thumbnail, thumbnail_name
 
 
+def get_video_duration(filename):
+    result = subprocess.check_output(
+        f'ffprobe -v quiet -show_streams -select_streams v:0 -of json "{filename}"',
+        shell=True
+    ).decode()
+
+    duration = json.loads(result)['streams'][0]['duration']
+
+    try:
+        duration = int(duration)
+    except ValueError:
+        duration = float(duration)
+
+    return duration
+
+
 def add_video_to_playlist(video):
     video_content_type = ContentType.objects.get_for_model(Video)
     playlist_content_type = ContentType.objects.get_for_model(Playlist)