feat: use prefetch_related to get message media, to not generate new …

…queries
weni-ai · Oct 18, 2024 · e7c9a3f · e7c9a3f
1 parent 6ceabec
commit e7c9a3f
Showing 1 changed file with 4 additions and 0 deletions.
diff --git 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)