27 lines
906 B
Python
27 lines
906 B
Python
from rest_framework import generics, status
|
|
from rest_framework.response import Response
|
|
from rest_framework.permissions import IsAuthenticated
|
|
from .models import File
|
|
from .serializers import FileSerializer
|
|
|
|
class FileListView(generics.ListAPIView):
|
|
serializer_class = FileSerializer
|
|
permission_classes = [IsAuthenticated]
|
|
|
|
def get_queryset(self):
|
|
return File.objects.filter(user=self.request.user)
|
|
|
|
class FileUploadView(generics.CreateAPIView):
|
|
serializer_class = FileSerializer
|
|
permission_classes = [IsAuthenticated]
|
|
|
|
def perform_create(self, serializer):
|
|
serializer.save(user=self.request.user)
|
|
|
|
class FileDetailView(generics.RetrieveUpdateDestroyAPIView):
|
|
serializer_class = FileSerializer
|
|
permission_classes = [IsAuthenticated]
|
|
lookup_field = 'id'
|
|
|
|
def get_queryset(self):
|
|
return File.objects.filter(user=self.request.user) |