init commit

video/views.py

@@ -0,0 +1,75 @@
+from .models import *
+from .serializers import *
+
+from django.shortcuts import render
+from rest_framework.decorators import api_view
+from rest_framework.decorators import api_view
+from rest_framework.response import Response
+from rest_framework import status
+from django.contrib.auth.models import User
+from django.contrib.auth import authenticate
+from rest_framework.authentication import TokenAuthentication
+from rest_framework.permissions import IsAuthenticated
+from rest_framework.decorators import authentication_classes
+from rest_framework.decorators import permission_classes
+from django.shortcuts import get_object_or_404
+from datetime import datetime
+import json
+import os
+from django.core.cache import cache
+from django.http import JsonResponse
+
+from django.conf import settings
+import ffmpeg_streaming
+from ffmpeg_streaming import Formats, Bitrate, Representation, Size
+from django.core.files.storage import default_storage
+from .tasks import create_hls_tasks
+
+
+# Create your views here.
+@api_view(['GET'])
+@authentication_classes([TokenAuthentication])
+@permission_classes([IsAuthenticated])
+def nsfw_videos(request):
+    if 'nsfw_videos' in cache:
+        data = cache.get('nsfw_videos')
+        if len(data) > 0:
+            return Response(data=data, status=status.HTTP_200_OK)
+    
+    users = Video.objects.all()
+    serializer = VideoSerializer(users, many=True)
+    data = serializer.data
+    cache.set('nsfw_videos', data, timeout=None)
+    return Response(data=data, status=status.HTTP_200_OK)
+
+
+@api_view(['GET'])
+@authentication_classes([TokenAuthentication])
+@permission_classes([IsAuthenticated])
+def hls_videos(request):
+    video_url = request.GET.get('video_name', '')
+    type = request.GET.get('video_type', '')
+
+    end_location = str(settings.MEDIA_ROOT) + '/hls/'+ video_url +'.m3u8'
+    end_file_name = '/media/hls/'+ video_url +'_720p.m3u8'
+
+    if default_storage.exists(end_location):
+        return JsonResponse({'file_location': end_file_name})
+
+    media_location = os.path.join(settings.MEDIA_ROOT) + "/" + type + "/" + video_url
+    video = ffmpeg_streaming.input(media_location)
+    
+    hls = video.hls(Formats.h264())
+    _720p  = Representation(Size(1280, 720), Bitrate(2048 * 1024, 320 * 1024))
+    hls.representations(_720p)
+    hls.output(end_location)
+    # {{url}}/videos/hls_video?video_name=Spiderman_Stretch.mp4&video_type=exercise_videos
+    # {{url}}/videos/hls_video?video_name=Recover_24.mp4&video_type=videos
+    return JsonResponse({'file_location': end_file_name})
+
+@api_view(['GET'])
+@authentication_classes([TokenAuthentication])
+@permission_classes([IsAuthenticated])
+def create_hls(request):
+    create_hls_tasks.delay()
+    return JsonResponse({'running': "running"})