zcdsj
/
factory
5
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

feat: base 增加get_queryset_custom

This commit is contained in:
caoqianming 2024-06-03 08:36:02 +08:00
parent e695543932
commit 62b012b690
1 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
apps/utils/viewsets.py
View File

@ -83,14 +83,19 @@ class CustomGenericViewSet(MyLoggingMixin, GenericViewSet):
if not hasattr(self, 'filterset_fields'): if not hasattr(self, 'filterset_fields'):
self.filterset_fields = self.select_related_fields self.filterset_fields = self.select_related_fields
def filter_custom(self, queryset): def get_queryset_custom(self, queryset):
""" """
自定义过滤方法可复写 自定义过滤方法可复写
""" """
if self.action in ["list", "retrieve", "create", "update", "partial_update", "destroy"]:
return queryset
elif hasattr(self, f'get_queryset_{self.action}'):
return getattr(self, f'get_queryset_{self.action}')(queryset)
return queryset return queryset
def filter_queryset(self, queryset): def filter_queryset(self, queryset):
queryset = super().filter_queryset(queryset) queryset = super().filter_queryset(queryset)
# 用于性能优化
if self.select_related_fields: if self.select_related_fields:
queryset = queryset.select_related(*self.select_related_fields) queryset = queryset.select_related(*self.select_related_fields)
if self.prefetch_related_fields: if self.prefetch_related_fields:
@ -108,7 +113,7 @@ class CustomGenericViewSet(MyLoggingMixin, GenericViewSet):
def get_queryset(self): def get_queryset(self):
queryset = super().get_queryset() queryset = super().get_queryset()
queryset = self.filter_custom(queryset) queryset = self.get_queryset_custom(queryset)
if self.data_filter: if self.data_filter:
user = self.request.user user = self.request.user
if user.is_superuser: if user.is_superuser:
@ -194,8 +199,7 @@ class CustomModelViewSet(BulkCreateModelMixin, BulkUpdateModelMixin, CustomListM
for k, v in self.perms_map.items(): for k, v in self.perms_map.items():
if v not in ALL_PERMS and v != '*': if v not in ALL_PERMS and v != '*':
ALL_PERMS.append(v) ALL_PERMS.append(v)
@swagger_auto_schema(request_body=ComplexSerializer, responses={200: {}}) @swagger_auto_schema(request_body=ComplexSerializer, responses={200: {}})
@action(methods=['post'], detail=False, perms_map={'post': '*'}) @action(methods=['post'], detail=False, perms_map={'post': '*'})
def cquery(self, request): def cquery(self, request):