From d12ed9ae22e4e540288cf933283c48b5923d212c Mon Sep 17 00:00:00 2001 From: Ivlev Denis Date: Tue, 20 Mar 2018 09:27:18 +0300 Subject: [PATCH] Add is_deactivated filter for comments --- api/v1/views.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/api/v1/views.py b/api/v1/views.py index 8aaf84b8..909195d4 100644 --- a/api/v1/views.py +++ b/api/v1/views.py @@ -346,6 +346,16 @@ class CommentViewSet(ExtendedModelViewSet): serializer_class = CommentSerializer permission_classes = (IsAdmin,) + def get_queryset(self): + queryset = self.queryset + is_deactivated = self.request.query_params.get('is_deactivated', '0') + if is_deactivated == '0': + return queryset + elif is_deactivated == '1': + return queryset.filter(deactivated_at__isnull=True) + elif is_deactivated == '2': + return queryset.filter(deactivated_at__isnull=False) + class AuthorRequestViewSet(ExtendedModelViewSet): queryset = AuthorRequest.objects.all()