From 0679198d0f28c5ceac60ee18b98ebbeee5306bce Mon Sep 17 00:00:00 2001 From: ahhui Date: Fri, 4 Aug 2023 18:26:44 +0800 Subject: [PATCH] =?UTF-8?q?perf:=20=20=E4=BC=98=E5=8C=96=E5=88=97=E6=9D=83?= =?UTF-8?q?=E9=99=90=E6=96=B0=E5=A2=9E=E4=BB=A3=E7=A0=81?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- backend/dvadmin/system/views/column.py | 20 +++++--------------- 1 file changed, 5 insertions(+), 15 deletions(-) diff --git a/backend/dvadmin/system/views/column.py b/backend/dvadmin/system/views/column.py index 18da1ce..5b9024c 100644 --- a/backend/dvadmin/system/views/column.py +++ b/backend/dvadmin/system/views/column.py @@ -44,26 +44,16 @@ class ColumnViewSet(CustomModelViewSet): def create(self, request, *args, **kwargs): payload = request.data - if not Role.objects.filter(pk=payload.get('role')).exists(): - return ErrorResponse(msg='角色不存在') - model = None - for app in get_custom_app_models(): - equal = False - for model in app: - if payload.get('model') == model['model']: - equal = True - model = model - break - if equal: + for model in get_custom_app_models(payload.get('app')): + if payload.get('model') == model['model']: break else: return ErrorResponse(msg='模型表不存在') + if Columns.objects.filter(app=model['app'], model=model['model'], field_name=payload.get('field_name')).exists(): return ErrorResponse(msg='‘%s’ 字段权限已有，不可重复创建' % payload.get('title')) - serializer = self.get_serializer(data=request.data, request=request) - serializer.is_valid(raise_exception=True) - self.perform_create(serializer) - return DetailResponse(data=serializer.data, msg="新增成功") + + return super().create(request, *args, **kwargs) @action(methods=['GET'], detail=False, permission_classes=[IsAuthenticated]) def get_models(self, request):