Feature: Move viewsets to views.py

core/drf_api.py

@@ -1,11 +1,10 @@
-from rest_framework import serializers, viewsets, routers, mixins
-from rest_framework.viewsets import GenericViewSet
+from django.conf import settings
+from rest_framework import serializers
 from taggit.models import Tag
 
-from core.models import Image, Pin
-from core.permissions import IsOwnerOrReadOnly
+from core.models import Image
+from core.models import Pin
 from django_images.models import Thumbnail
-from django.conf import settings
 from users.models import User
 
 
@@ -19,11 +18,6 @@ class UserSerializer(serializers.HyperlinkedModelSerializer):
         )
 
 
-class UserViewSet(viewsets.ModelViewSet):
-    queryset = User.objects.all()
-    serializer_class = UserSerializer
-
-
 class ThumbnailSerializer(serializers.HyperlinkedModelSerializer):
     class Meta:
         model = Thumbnail
@@ -62,11 +56,6 @@ class ImageSerializer(serializers.ModelSerializer):
         return image
 
 
-class ImageViewSet(mixins.CreateModelMixin, mixins.RetrieveModelMixin, GenericViewSet):
-    queryset = Image.objects.all()
-    serializer_class = ImageSerializer
-
-
 class TagSerializer(serializers.ModelSerializer):
     class Meta:
         model = Tag
@@ -121,16 +110,3 @@ class PinSerializer(serializers.HyperlinkedModelSerializer):
             image = Image.objects.create(image=image_file['image'])
             instance.image = image
         return super(PinSerializer, self).update(instance, validated_data)
-
-
-class PinViewSet(viewsets.ModelViewSet):
-    queryset = Pin.objects.all()
-    serializer_class = PinSerializer
-    filter_fields = ('submitter__username',)
-    permission_classes = [IsOwnerOrReadOnly("submitter"), ]
-
-
-drf_router = routers.DefaultRouter()
-drf_router.register(r'users', UserViewSet)
-drf_router.register(r'pins', PinViewSet)
-drf_router.register(r'images', ImageViewSet)