diff --git a/api/views.py b/api/views.py index f7301a3..cdf1baa 100755 --- a/api/views.py +++ b/api/views.py @@ -33,15 +33,15 @@ class MessageViewSet(ModelViewSet): queryset = Message.objects.all() serializer_class = MessageSerializer filter_class = MessageFilterSet - - def get_queryset(self): - queryset = Message.objects.all() - search_param = self.request.query_params.get('operand', None) - # queryset = queryset.filter(Q(sender=self.request.user.pk) | Q(recipent=self.request.user.pk)) - # import code; code.interact(local=dict(globals(), **locals())) - if search_param: - queryset = queryset.filter(Q(sender=self.request.user.pk) | Q(recipent=self.request.user.pk)) - return queryset + + def get_queryset(self): + queryset = Message.objects.all() + search_param = self.request.query_params.get('operand', None) + # queryset = queryset.filter(Q(sender=self.request.user.pk) | Q(recipent=self.request.user.pk)) + # import code; code.interact(local=dict(globals(), **locals())) + if search_param: + queryset = queryset.filter(Q(sender=self.request.user.pk) | Q(recipent=self.request.user.pk)) + return queryset