124 lines
4.0 KiB
Python
124 lines
4.0 KiB
Python
from rest_framework import status
|
|
from rest_framework.decorators import api_view
|
|
from rest_framework.response import Response
|
|
from rest_framework.views import APIView
|
|
from rest_framework.permissions import IsAuthenticatedOrReadOnly
|
|
from django.http import Http404
|
|
|
|
from django.contrib.auth.models import User
|
|
|
|
from snippets.models import Snippet
|
|
from snippets.serializers import SnippetSerializer, UserSerializer
|
|
|
|
# @api_view(["GET", "POST"])
|
|
# def snippet_list(request):
|
|
|
|
# if request.method == "GET":
|
|
# snippets = Snippet.objects.all()
|
|
# serializer = SnippetSerializer(snippets, many=True)
|
|
# return Response(serializer.data)
|
|
|
|
# elif request.method == "POST":
|
|
# data = JSONParser().parse(request)
|
|
# serializer = SnippetSerializer(data=data)
|
|
# if serializer.is_valid():
|
|
# serializer.save()
|
|
# return Response(serializer.data, status=status.HTTP_201_CREATED)
|
|
# return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)
|
|
|
|
|
|
class SnippetList(APIView):
|
|
|
|
permission_classes = [IsAuthenticatedOrReadOnly]
|
|
|
|
def get(self, request, format=None):
|
|
snippets = Snippet.objects.all()
|
|
serializer = SnippetSerializer(snippets, many=True)
|
|
return Response(serializer.data)
|
|
|
|
def post(self, request, format=None):
|
|
serializer = SnippetSerializer(data=request.data)
|
|
if serializer.is_valid():
|
|
serializer.save(owner=request.user)
|
|
return Response(serializer.data, status=status.HTTP_201_CREATED)
|
|
return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)
|
|
|
|
|
|
# @api_view(["GET", "PUT", "DELETE"])
|
|
# def snippet_detail(request, pk):
|
|
|
|
# try:
|
|
# snippet = Snippet.objects.get(pk=pk)
|
|
# except Snippet.DoesNotExist:
|
|
# return Response(status=status.HTTP_404_NOT_FOUND)
|
|
|
|
# if request.method == "GET":
|
|
# serializer = SnippetSerializer(snippet)
|
|
# return Response(serializer.data)
|
|
|
|
# elif request.method == "PUT":
|
|
# data = JSONParser().parse(request)
|
|
# serializer = SnippetSerializer(snippet, data=data)
|
|
# if serializer.is_valid():
|
|
# serializer.save()
|
|
# return Response(serializer.data)
|
|
# return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)
|
|
|
|
# elif request.method == "DELETE":
|
|
# snippet.delete()
|
|
# return Response(status=status.HTTP_204_NO_CONTENT)
|
|
|
|
class SnippetDetail(APIView):
|
|
|
|
permission_classes = [IsAuthenticatedOrReadOnly]
|
|
|
|
def get_object(self, pk):
|
|
try:
|
|
return Snippet.objects.get(pk=pk)
|
|
except Snippet.DoesNotExist:
|
|
raise Http404
|
|
|
|
def get(self, request, pk, format=None):
|
|
snippet = self.get_object(pk)
|
|
serializer = SnippetSerializer(snippet)
|
|
return Response(serializer.data)
|
|
|
|
def put(self, request, pk):
|
|
snippet = self.get_object(pk)
|
|
serializer = SnippetSerializer(snippet, data=request.data)
|
|
if serializer.is_valid():
|
|
serializer.save()
|
|
return Response(serializer.data, status=status.HTTP_201_CREATED)
|
|
return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)
|
|
|
|
def delete(self, request, pk):
|
|
snippet = self.get_object(pk)
|
|
snippet.delete()
|
|
return Response(status=status.HTTP_204_NO_CONTENT)
|
|
|
|
|
|
class UserList(APIView):
|
|
permission_classes = [IsAuthenticatedOrReadOnly]
|
|
|
|
def get(self, request, format=None):
|
|
user = User.objects.all()
|
|
serializer = UserSerializer(user, many=True)
|
|
return Response(serializer.data, status=status.HTTP_200_OK)
|
|
|
|
class UserDetail(APIView):
|
|
|
|
permission_classes = [IsAuthenticatedOrReadOnly]
|
|
|
|
def get_user(self, pk):
|
|
try:
|
|
return User.objects.get(pk=pk)
|
|
except User.DoesNotExist:
|
|
raise Http404
|
|
|
|
def get(self, request, pk, format=None):
|
|
user = self.get_user(pk=pk)
|
|
serializer = UserSerializer(user)
|
|
return Response(serializer.data)
|
|
|
|
|