forked from wiktor/spejstore-new
54 lines
1.7 KiB
Python
54 lines
1.7 KiB
Python
from rest_framework import viewsets, generics, filters
|
|
from rest_framework.response import Response
|
|
from rest_framework.decorators import detail_route
|
|
|
|
from storage.models import Item, Label
|
|
from storage.serializers import ItemSerializer, LabelSerializer
|
|
from django.shortcuts import get_object_or_404
|
|
|
|
class LabelViewSet(viewsets.ModelViewSet):
|
|
"""
|
|
API endpoint that allows items to be viewed or edited.
|
|
"""
|
|
queryset = Label.objects
|
|
serializer_class = LabelSerializer
|
|
|
|
|
|
class ItemViewSet(viewsets.ModelViewSet):
|
|
"""
|
|
API endpoint that allows items to be viewed or edited.
|
|
"""
|
|
queryset = Item.objects
|
|
serializer_class = ItemSerializer
|
|
|
|
def get_queryset(self):
|
|
return Item.get_roots()
|
|
|
|
def get_object(self):
|
|
lookup_url_kwarg = self.lookup_url_kwarg or self.lookup_field
|
|
|
|
obj = get_object_or_404(Item, pk=self.kwargs[lookup_url_kwarg])
|
|
self.check_object_permissions(self.request, obj)
|
|
|
|
return obj
|
|
|
|
@detail_route()
|
|
def children(self, request, pk):
|
|
item = self.get_object()
|
|
return Response(self.serializer_class(item.get_children().all(), many=True).data)
|
|
|
|
@detail_route()
|
|
def ancestors(self, request, pk):
|
|
item = self.get_object()
|
|
return Response(self.serializer_class(item.get_ancestors().all(), many=True).data)
|
|
|
|
@detail_route()
|
|
def descendants(self, request, pk):
|
|
item = self.get_object()
|
|
return Response(self.serializer_class(item.get_descendants().all(), many=True).data)
|
|
|
|
@detail_route()
|
|
def siblings(self, request, pk):
|
|
item = self.get_object()
|
|
return Response(self.serializer_class(item.get_siblings().all(), many=True).data)
|