This should allmost work
This commit is contained in:
1 parent
7787d0abc6
commit
9bbfea8971
21 files changed
+250
-34
No files matched your search
@@ -0,0 +1,57 @@
|
||||
from __future__ import unicode_literals
|
||||
from rest_framework.decorators import list_route
|
||||
from rest_framework.permissions import IsAuthenticated
|
||||
from rest_framework.response import Response
|
||||
from rest_framework import status, filters, viewsets
|
||||
|
||||
from .models import Alias, Domain, User
|
||||
from .serializers import AliasSerializer, AliasIdSerializer, DomainSerializer, UserSerializer
|
||||
|
||||
|
||||
class AliasViewSet(viewsets.ModelViewSet):
|
||||
""" Aliases """
|
||||
queryset = Alias.objects.all()
|
||||
serializer_class = AliasSerializer
|
||||
permission_classes = [IsAuthenticated]
|
||||
filter_backends = (filters.DjangoFilterBackend,)
|
||||
filter_fields = ('domain', 'domain__name')
|
||||
|
||||
@list_route(methods=['post'])
|
||||
def create_bulk(self, request):
|
||||
serializer = AliasSerializer(data=request.data, many=True)
|
||||
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)
|
||||
|
||||
@list_route(methods=['delete'])
|
||||
def delete_bulk(self, request):
|
||||
# TODO how to lookup instances and delete them by id?
|
||||
serializer = AliasIdSerializer(data=request.data, many=True)
|
||||
if not serializer.is_valid():
|
||||
return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)
|
||||
|
||||
ids = map(lambda x: x['id'], serializer.initial_data) # FIXME this is not how to do it
|
||||
aliases = Alias.objects.filter(id__in=ids)
|
||||
diff = set(ids).difference(set(aliases.values_list('id', flat=True)))
|
||||
aliases.delete()
|
||||
|
||||
if diff:
|
||||
result = "ID's '{}' could not be found and were not deleted.".format(", ".join(map(str, diff)))
|
||||
return Response({"result": result}, status=status.HTTP_200_OK)
|
||||
|
||||
return Response(status=status.HTTP_204_NO_CONTENT)
|
||||
|
||||
|
||||
class DomainViewSet(viewsets.ModelViewSet):
|
||||
""" Domains """
|
||||
queryset = Domain.objects.all()
|
||||
serializer_class = DomainSerializer
|
||||
permission_classes = [IsAuthenticated]
|
||||
|
||||
|
||||
class UserViewSet(viewsets.ModelViewSet):
|
||||
""" Users """
|
||||
queryset = User.objects.all()
|
||||
serializer_class = UserSerializer
|
||||
permission_classes = [IsAuthenticated]
|
||||
Reference in new issue
Block a user