views.py 522 B

1234567891011121314151617
  1. from rest_framework import viewsets
  2. from organizations.models import Organization
  3. from .serializers.serializers import (
  4. OrganizationSerializer,
  5. OrganizationDetailSerializer,
  6. )
  7. class OrganizationViewSet(viewsets.ModelViewSet):
  8. queryset = Organization.objects.all()
  9. serializer_class = OrganizationSerializer
  10. lookup_field = "slug"
  11. def get_serializer_class(self):
  12. if self.action in ["retrieve"]:
  13. return OrganizationDetailSerializer
  14. return super().get_serializer_class()