genkid123

Untitled

Oct 22nd, 2020
75
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.84 KB | None | 0 0
  1. class DialogRoomSerializer(ModelSerializer):
  2. class Meta:
  3. model = [Room, Dialogue]
  4. fields = "__all__"
  5.  
  6. @action(methods=["get"], detail=False, url_path="dialog_room", url_name="dialog_and_room")
  7. def get_dialog_room(self, request):
  8. current_user = self.request.user
  9. dialog = Dialogue.objects.filter(Q(employee=current_user) | Q(student=current_user))
  10. room = Room.objects.filter(Q(user=current_user) | Q(members=current_user))
  11. dialog_room = list(chain(dialog, room))
  12. page = self.paginate_queryset(dialog_room)
  13. if page:
  14. serializer = DialogRoomSerializer(page, many=True)
  15. return self.get_paginated_response(serializer.data)
  16. serializer = DialogRoomSerializer(page, many=True)
  17. return Response(data=serializer.data, status=status.HTTP_200_OK)
Add Comment
Please, Sign In to add comment