29 lines
980 B
Python
29 lines
980 B
Python
from rest_framework import viewsets
|
|
from django_filters import rest_framework as filters
|
|
|
|
from .filters import CollectionFilter
|
|
from .models import Collection, CollectionMethod, CollectionType, Flaw
|
|
from .serializers import (CollectionSerializer, CollectionMethodSerializer,
|
|
CollectionTypeSerializer, FlawSerializer)
|
|
|
|
|
|
class CollectionViewSet(viewsets.ModelViewSet):
|
|
queryset = Collection.objects.all()
|
|
serializer_class = CollectionSerializer
|
|
filter_backends = (filters.DjangoFilterBackend,)
|
|
filter_class = CollectionFilter
|
|
|
|
|
|
class CollectionMethodViewSet(viewsets.ModelViewSet):
|
|
queryset = CollectionMethod.objects.all()
|
|
serializer_class = CollectionMethodSerializer
|
|
|
|
|
|
class CollectionTypeViewSet(viewsets.ModelViewSet):
|
|
queryset = CollectionType.objects.all()
|
|
serializer_class = CollectionTypeSerializer
|
|
|
|
|
|
class FlawViewSet(viewsets.ModelViewSet):
|
|
queryset = Flaw.objects.all()
|
|
serializer_class = FlawSerializer
|