功能变化: 优化权限范围

pull/84/head
猿小天 2022-12-30 14:33:06 +08:00
parent f2e474bf12
commit 0a3516369e
1 changed files with 65 additions and 30 deletions

View File

@ -116,8 +116,8 @@ class RoleViewSet(CustomModelViewSet):
@action(methods=['GET'], detail=False, permission_classes=[IsAuthenticated]) @action(methods=['GET'], detail=False, permission_classes=[IsAuthenticated])
def data_scope(self, request): def data_scope(self, request):
is_superuser = request.user.is_superuser is_superuser = request.user.is_superuser
is_admin = Role.objects.filter(users__id=request.user.id,admin=True) role_queryset = Role.objects.filter(users__id=request.user.id).values_list('data_range', flat=True)
if is_superuser or is_admin: if is_superuser:
data = [ data = [
{ {
"value": 0, "value": 0,
@ -141,18 +141,53 @@ class RoleViewSet(CustomModelViewSet):
} }
] ]
else: else:
data = [ data = []
{ data_range_list = list(set(role_queryset))
for item in data_range_list:
if item == 0:
data = [{
"value": 0, "value": 0,
"label": '仅本人数据权限' "label": '仅本人数据权限'
}, }]
{ elif item == 1:
data = [{
"value": 0,
"label": '仅本人数据权限'
}, {
"value": 1, "value": 1,
"label": '本部门及以下数据权限' "label": '本部门及以下数据权限'
}, },
{ {
"value": 2, "value": 2,
"label": '本部门数据权限' "label": '本部门数据权限'
} }]
] elif item == 2:
data = [{
"value": 0,
"label": '仅本人数据权限'
},
{
"value": 2,
"label": '本部门数据权限'
}]
elif item == 3:
data = [{
"value": 0,
"label": '仅本人数据权限'
},
{
"value": 3,
"label": '全部数据权限'
}, ]
elif item == 4:
data = [{
"value": 0,
"label": '仅本人数据权限'
},
{
"value": 4,
"label": '自定义数据权限'
}]
else:
data = []
return DetailResponse(data=data) return DetailResponse(data=data)