How to return list or retrieve object from single class in django rest?

Question

I'm using 2 different classes to perform get and retrieve operations.

class Stock(ListAPIView):
    serializer_class=StockSerializer
    queryset=Stock.objects.all()

class StockView(RetrieveAPIView):
    serializer_class = StockSerializer
    lookup_field = 'slug'

    def get_queryset(self, slug):                    
            collection = Stock.objects.get(slug=slug)            
        return collection

    def get(self, request, slug):
        collection = self.get_queryset(slug)
        serializer = CollectionSerializer(collection)
        
        return Response(
            serializer.data, 
            status=status.HTTP_200_OK
        )

but I want to perform both operations from a single class,i.e I don't want to write 2 different classes. So how to handle both (GET & RETRIEVE ) from a single class ?. Can we do that using generic APIViews? Thankx in advance

You can use ModelViewSet instead of RetrieveAPIView: django-rest-framework.org/api-guide/viewsets/#modelviewset — Harun Yilmaz
– Harun Yilmaz, Commented Jul 14, 2020 at 7:12
can we user generic API for get and retrieve? Instead of using modelviewset — Yoshi Hutchinson
– Yoshi Hutchinson, Commented Jul 14, 2020 at 10:28

Vishal Singh · Accepted Answer · 2020-07-14 07:18:48Z

1

If you only require GET and RETRIEVE operation then you can use ReadOnlyModelViewSet.

from rest_framework.viewsets import ReadOnlyModelViewSet

class StockView(ReadOnlyModelViewSet):
    serializer_class = StockSerializer
    lookup_field = 'slug'
    queryset = Stock.objects.all()

answered Jul 14, 2020 at 7:18

Vishal Singh

6,2522 gold badges19 silver badges34 bronze badges

Sign up to request clarification or add additional context in comments.

2 Comments

Yoshi Hutchinson Over a year ago

can we user generic API for get and retrieve? Instead of using modelviewset

Vishal Singh Over a year ago

@YoshiHutchinson you can use GenericAPIView but then you will have to write a lot of manual code.

Collectives™ on Stack Overflow

How to return list or retrieve object from single class in django rest?

1 Answer 1

2 Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

2 Comments

Your Answer

Sign up or log in

Post as a guest

Related