from django.shortcuts import get_object_or_404

from rest_framework import viewsets, status
from rest_framework.response import Response
from rest_framework.decorators import action


from .models import Video
from .serializers import VideoSerializer
from .tasks import generate_subtitles

class VideoViewSet(viewsets.ModelViewSet):
    queryset = Video.objects.all()
    serializer_class = VideoSerializer

    def create(self, request, *args, **kwargs):
        serializer = self.get_serializer(data=request.data)
        serializer.is_valid(raise_exception=True)
        video = serializer.save()
        # generate_subtitles.delay(video.id)
        return Response(serializer.data, status=status.HTTP_201_CREATED)

    # @action(detail=True, methods=['get'])
    # def status(self, request, pk=None):
    #     video = get_object_or_404(Video, pk=pk)
    #     return Response({'status': video.status})

    # @action(detail=True, methods=['get'])
    # def subtitles(self, request, pk=None):
    #     video = get_object_or_404(Video, pk=pk)
    #     return Response({'subtitles': video.subtitles})