from __future__ import unicode_literals, print_function
from rest_framework_bulk import generics

from .models import SimpleModel
from .serializers import SimpleSerializer


class SimpleMixin(object):
    model = SimpleModel
    queryset = SimpleModel.objects.all()
    serializer_class = SimpleSerializer


class SimpleBulkAPIView(SimpleMixin, generics.ListBulkCreateUpdateDestroyAPIView):
    pass


class FilteredBulkAPIView(SimpleMixin, generics.ListBulkCreateUpdateDestroyAPIView):
    def filter_queryset(self, queryset):
        return queryset.filter(number__gt=5)


class SimpleViewSet(SimpleMixin, generics.BulkModelViewSet):
    def filter_queryset(self, queryset):
        return queryset.filter(number__gt=5)