38 lines
1.2 KiB
Python
38 lines
1.2 KiB
Python
from django.contrib.auth.models import User, Group
|
|
from rest_framework import viewsets
|
|
from rest_framework import permissions
|
|
from base.serializers import UserSerializer, GroupSerializer, LayerSerializer
|
|
from base.models import Layer
|
|
from rest_framework.response import Response
|
|
|
|
class UserViewSet(viewsets.ModelViewSet):
|
|
"""
|
|
API endpoint that allows users to be viewed or edited.
|
|
"""
|
|
queryset = User.objects.all().order_by('-date_joined')
|
|
serializer_class = UserSerializer
|
|
permission_classes = [permissions.IsAuthenticated]
|
|
|
|
|
|
class GroupViewSet(viewsets.ModelViewSet):
|
|
"""
|
|
API endpoint that allows groups to be viewed or edited.
|
|
"""
|
|
queryset = Group.objects.all()
|
|
serializer_class = GroupSerializer
|
|
permission_classes = [permissions.IsAuthenticated]
|
|
|
|
|
|
class LayerViewSet(viewsets.ViewSet):
|
|
"""
|
|
API endpoint that allows layers to be viewed or edited.
|
|
"""
|
|
queryset = Layer.objects.all()
|
|
serializer_class = LayerSerializer
|
|
permission_classes = [permissions.IsAuthenticated]
|
|
|
|
def list(self, request):
|
|
user = request.user
|
|
data = Layer.objects.filter(owner=user)
|
|
return Response(LayerSerializer(data,many=True).data)
|