API Endpoint setup

vidsplit/api/urls.py

@@ -1,6 +1,9 @@
 from django.urls import path
-from .views import DownloadPage
+from .views import Initialize, Generate, Download, Session
 
 urlpatterns = [
-    path("download", DownloadPage.as_view(), name="download-page"),
+    path("initialize", Initialize.as_view(), name="initialize"),
+    path("generate", Generate.as_view(), name="generate"),
+    path("download", Download.as_view(), name="download"),
+    path("session", Session.as_view(), name="session"),
 ]

vidsplit/api/views.py

@@ -1,26 +1,49 @@
-from django.shortcuts import render
 from rest_framework import generics, status
 from .serializers import VideoSerializer
 from .models import Video
-from rest_framework.views import APIView
 from rest_framework.response import Response
 
 
-class DownloadPage(generics.ListAPIView):
-    def get(self, request, *args, **kwargs):
-        session_id = self.
-        video_id = self.kwargs.get("video_id")
-        video = Video.objects.get(id=video_id)
-        serializer = VideoSerializer(video)
-        return Response(serializer.data, status=status.HTTP_200_OK)
-
+class Initialize(generics.ListAPIView):
+    http_method_names = ["post"]
+
     def post(self, request, *args, **kwargs):
         video_id = self.kwargs.get("video_id")
-        session_id = request.session.session_key
-        video = Video.objects.get(id=video_id, sesh_id = request.session.session_key)
+        video = Video.objects.get(id=video_id, session_id=request.session.session_key)
         serializer = VideoSerializer(video, data=request.data)
         if serializer.is_valid(raise_exception=True):
             serializer.save()
             return Response(serializer.data, status=status.HTTP_201_CREATED)
         return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)
 
+
+class Generate(generics.ListAPIView):
+    http_method_names = ["put"]
+
+    def put(self, request, *args, **kwargs):
+        video_id = self.kwargs.get("video_id")
+        video = Video.objects.get(id=video_id, session_id=request.session.session_key)
+        serializer = VideoSerializer(video, data=request.data)
+        if serializer.is_valid(raise_exception=True):
+            serializer.save()
+            return Response(serializer.data, status=status.HTTP_201_CREATED)
+        return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)
+
+
+class Download(generics.ListAPIView):
+    http_method_names = ["get"]
+
+    def get(self, request, *args, **kwargs):
+        video_id = self.kwargs.get("video_id")
+        video = Video.objects.get(id=video_id, session_id=request.session.session_key)
+        serializer = VideoSerializer(video)
+        return Response(serializer.data)
+
+
+class Session(generics.ListAPIView):
+    http_method_names = ["get"]
+
+    def get(self, request, *args, **kwargs):
+        video = Video.objects.get(session_id=request.session.session_key)
+        serializer = VideoSerializer(video)
+        return Response(serializer.data)