From 7202c82ea18c149e4a290ef00ac1df3e60885316 Mon Sep 17 00:00:00 2001 From: caoqianming Date: Fri, 3 Nov 2023 09:07:10 +0800 Subject: [PATCH] =?UTF-8?q?feat:=20base=20select=5Frel=E5=92=8Cprefetch=5F?= =?UTF-8?q?rel=E6=94=BE=E4=BA=8Efilter=E4=B9=8B=E5=90=8E?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- apps/utils/viewsets.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/apps/utils/viewsets.py b/apps/utils/viewsets.py index 9fc0ce84..869187c4 100755 --- a/apps/utils/viewsets.py +++ b/apps/utils/viewsets.py @@ -94,12 +94,16 @@ class CustomGenericViewSet(MyLoggingMixin, GenericViewSet): """ return queryset - def get_queryset(self): - queryset = super().get_queryset() + def filter_queryset(self, queryset): + queryset = super().filter_queryset(queryset) if self.select_related_fields: queryset = queryset.select_related(*self.select_related_fields) if self.prefetch_related_fields: queryset = queryset.prefetch_related(*self.prefetch_related_fields) + return queryset + + def get_queryset(self): + queryset = super().get_queryset() queryset = self.filter_custom(queryset) if self.data_filter: user = self.request.user