def list(self, request, *args, **kwargs):
try:
if Room.objects.get(id=kwargs['id']).members.filter(
id=request.user.id):
self.queryset = Message.objects.filter(room=kwargs['id'])
return ListAPIView.list(self, request, *args, **kwargs)
return Response('Permission denied',
status=status.HTTP_403_FORBIDDEN)
except ObjectDoesNotExist:
return Response({'detail': 'Not found.'},
status=status.HTTP_404_NOT_FOUND)
def list(self, request, *args, **kwargs):
try:
# Verify that the user is a member of the room
if (Room.objects.get(id=kwargs["id"]).members.filter(
id=request.user.id).exists()):
# Retrieve the messages for the room
self.queryset = Message.objects.filter(room=kwargs["id"])
return ListAPIView.list(self, request, *args, **kwargs)
return Response("Permission denied",
status=status.HTTP_403_FORBIDDEN)
# Room was not found
except ObjectDoesNotExist:
return Response({"detail": "Not found."},
status=status.HTTP_404_NOT_FOUND)
def list(self, request, *args, **kwargs):
return ListAPIView.list(self, request, *args, **kwargs)
def list(self, request, *args, **kwargs):
return ListAPIView.list(self, request, *args, **kwargs)
