1234567891011121314151617 |
- from rest_framework import viewsets
- from organizations.models import Organization
- from .serializers.serializers import (
- OrganizationSerializer,
- OrganizationDetailSerializer,
- )
- class OrganizationViewSet(viewsets.ModelViewSet):
- queryset = Organization.objects.all()
- serializer_class = OrganizationSerializer
- lookup_field = "slug"
- def get_serializer_class(self):
- if self.action in ["retrieve"]:
- return OrganizationDetailSerializer
- return super().get_serializer_class()
|