75 lines
2.7 KiB
Python
75 lines
2.7 KiB
Python
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.auto_generate_representations()
|
|
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"}) |