diff --git a/apps/hadis/views_admin.py b/apps/hadis/views_admin.py index 9ae331c..2288899 100644 --- a/apps/hadis/views_admin.py +++ b/apps/hadis/views_admin.py @@ -49,7 +49,11 @@ class AdminHadisCategoryViewSet(ReadOnlyModelViewSet): pagination_class = None def get_queryset(self): - return HadisCategory.objects.all().order_by("order", "id") + queryset = HadisCategory.objects.all() + source_type = self.request.query_params.get("source_type") + if source_type: + queryset = queryset.filter(source_type=source_type) + return queryset.order_by("order", "id") class DovodiHadisSectViewSet(ModelViewSet): @@ -288,6 +292,10 @@ class AdminHadisViewSet(ModelViewSet): if category_id and category_id != "all": queryset = queryset.filter(category_id=category_id) + source_type = self.request.query_params.get("source_type") + if source_type: + queryset = queryset.filter(category__source_type=source_type) + status_filter = self.request.query_params.get("status") if status_filter in {"true", "false"}: queryset = queryset.filter(status=status_filter == "true")