refactor: 重构权限管理

1.更新字段管理
This commit is contained in:
猿小天
2023-11-20 17:19:20 +08:00
parent 94ad6b1bae
commit 4ac8ed7627
6 changed files with 25 additions and 40 deletions

View File

@@ -15,16 +15,14 @@ class FieldPermissionMixin:
获取字段权限
"""
finded = False
for app in get_custom_app_models():
for model in app:
if model['object'] is self.serializer_class.Meta.model:
finded = True
break
for model in get_custom_app_models():
if model['object'] is self.serializer_class.Meta.model:
finded = True
break
if finded:
break
if finded is False:
return []
print(266666,model)
user = request.user
if user.is_superuser==1:
data = MenuField.objects.filter( model=model['model']).values('field_name')

View File

@@ -141,6 +141,7 @@ def get_custom_app_models(app_name=None):
return get_model_from_app(app_name)
res = []
for app in settings.CUSTOM_APPS:
# print(222,get_model_from_app(app))
res.append(get_model_from_app(app))
all_models = get_model_from_app(app)
for model in all_models:
res.append(model)
return res

View File

@@ -82,12 +82,9 @@ class CustomModelViewSet(ModelViewSet, ImportSerializerMixin, ExportSerializerMi
def get_menu_field(self, serializer_class):
"""获取字段权限"""
finded = False
for app in get_custom_app_models():
for model in app:
if model['object'] is serializer_class.Meta.model:
finded = True
break
if finded:
for model in get_custom_app_models():
if model['object'] is serializer_class.Meta.model:
finded = True
break
if finded is False:
return []