You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
86 lines
3.1 KiB
86 lines
3.1 KiB
from django.contrib.admin.utils import lookup_field
|
|
from rest_framework.generics import ListAPIView , RetrieveAPIView
|
|
from rest_framework.response import Response
|
|
from ..models import Transmitters , TransmitterOpinion
|
|
from ..serializers import TransmitterSerializer , TransmitterDetailSerializer, TransmitterSyncSerializer
|
|
from ..docs import transmitter_list_swagger, transmitter_detail_swagger, transmitter_sync_swagger
|
|
from utils.pagination import NoPagination
|
|
|
|
class TransmitterView(ListAPIView):
|
|
queryset = Transmitters.objects.all()
|
|
serializer_class = TransmitterSerializer
|
|
|
|
@transmitter_list_swagger
|
|
def get(self, request, *args, **kwargs):
|
|
return super().get(request, *args, **kwargs)
|
|
|
|
def get_queryset(self):
|
|
queryset = Transmitters.objects.all()
|
|
|
|
# Filter by status (reliability)
|
|
status_filter = self.request.query_params.get('status', None)
|
|
if status_filter:
|
|
queryset = queryset.filter(reliability=status_filter)
|
|
|
|
# Filter by madhhab
|
|
madhhab_filter = self.request.query_params.get('madhhab', None)
|
|
if madhhab_filter:
|
|
queryset = queryset.filter(madhhab=madhhab_filter)
|
|
|
|
# Filter by generation (narrator layer)
|
|
# generation_filter = self.request.query_params.get('generation', None)
|
|
# if generation_filter:
|
|
# try:
|
|
# generation_num = int(generation_filter)
|
|
# # Filter transmitters who have hadis transmissions in the specified generation/layer
|
|
# queryset = queryset.filter(
|
|
# hadi=generation_num
|
|
# ).distinct()
|
|
# except (ValueError, TypeError):
|
|
# pass # Invalid generation number, ignore filter
|
|
|
|
return queryset
|
|
|
|
class TransmitterDetailView(RetrieveAPIView):
|
|
serializer_class = TransmitterDetailSerializer
|
|
lookup_field = 'id'
|
|
lookup_url_kwarg = 'transmitters_id'
|
|
|
|
@transmitter_detail_swagger
|
|
def get(self, request, *args, **kwargs):
|
|
return super().get(request, *args, **kwargs)
|
|
|
|
def get_queryset(self):
|
|
input = self.kwargs['transmitters_id']
|
|
return Transmitters.objects.filter(id=input)
|
|
|
|
|
|
class TransmitterSyncView(ListAPIView):
|
|
"""
|
|
API view to sync all transmitter data for offline mode
|
|
Returns all transmitters with biographical data and scholarly opinions
|
|
"""
|
|
serializer_class = TransmitterSyncSerializer
|
|
pagination_class = NoPagination
|
|
|
|
@transmitter_sync_swagger
|
|
|
|
def get_queryset(self):
|
|
return Transmitters.objects.prefetch_related('opinions').order_by('id')
|
|
|
|
def list(self, request, *args, **kwargs):
|
|
queryset = self.get_queryset()
|
|
serializer = self.get_serializer(queryset, many=True, context={'request': request})
|
|
|
|
# Group transmitters by ID for easy lookup
|
|
grouped_data = {}
|
|
for transmitter_data in serializer.data:
|
|
transmitter_id = str(transmitter_data['id'])
|
|
grouped_data[transmitter_id] = transmitter_data
|
|
|
|
response_data = {
|
|
'count': queryset.count(),
|
|
'results': grouped_data
|
|
}
|
|
|
|
return Response(response_data)
|