from django.contrib.admin.utils import lookup_field from rest_framework.generics import ListAPIView , RetrieveAPIView from ..models import Transmitters , TransmitterOpinion from ..serializers import TransmitterSerializer , TransmitterDetailSerializer from utils.pagination import NoPagination class TransmitterView(ListAPIView): queryset = Transmitters.objects.all() serializer_class = TransmitterSerializer pagination_class = NoPagination class TransmitterDetailView(RetrieveAPIView): serializer_class = TransmitterDetailSerializer lookup_field = 'id' lookup_url_kwarg = 'transmitters_id' def get_queryset(self): input = self.kwargs['transmitters_id'] return Transmitters.objects.filter(id=input)