From e7c9a3fde1110628c37ee1cecc8e3012af039fca Mon Sep 17 00:00:00 2001 From: helllllllder Date: Fri, 18 Oct 2024 11:14:58 -0300 Subject: [PATCH] feat: use prefetch_related to get message media, to not generate new queries --- chats/apps/api/v1/msgs/viewsets.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/chats/apps/api/v1/msgs/viewsets.py b/chats/apps/api/v1/msgs/viewsets.py index 6f064ffb..682bbca8 100644 --- a/chats/apps/api/v1/msgs/viewsets.py +++ b/chats/apps/api/v1/msgs/viewsets.py @@ -36,6 +36,10 @@ def get_paginated_response(self, data): qs = super().get_paginated_response(data) return qs + def get_queryset(self): + qs = super().get_queryset() + return qs if self.action != "list" else qs.prefetch_related("medias") + def create(self, request, *args, **kwargs): # TODO USE THE REQUEST.USER TO SET THE USER IN THE MESSAGE return super().create(request, *args, **kwargs)