权限代码修改

This commit is contained in:
曹前明 2022-06-07 09:55:11 +08:00
parent 97ea46bed5
commit 48b7267bb1
2 changed files with 16 additions and 13 deletions

View File

@ -198,7 +198,7 @@ class QActionViewSet(PageOrNot, ListModelMixin, DestroyModelMixin, RetrieveModel
return Response()
@action(methods=['post'], detail=False,
perms_map = {'post':'ability_create'}, serializer_class=QActionACreateSerializer)
perms_map = {'post':'qaction_create'}, serializer_class=QActionACreateSerializer)
@transaction.atomic
def ability_create(self, request, *args, **kwargs):
"""
@ -213,7 +213,7 @@ class QActionViewSet(PageOrNot, ListModelMixin, DestroyModelMixin, RetrieveModel
return Response()
@action(methods=['post'], detail=False,
perms_map = {'post':'ability_nochange'}, serializer_class=QActionNoChangeSerializer)
perms_map = {'post':'qaction_create'}, serializer_class=QActionNoChangeSerializer)
@transaction.atomic
def ability_nochange(self, request, *args, **kwargs):
"""
@ -228,7 +228,7 @@ class QActionViewSet(PageOrNot, ListModelMixin, DestroyModelMixin, RetrieveModel
return Response()
@action(methods=['post'], detail=False,
perms_map = {'post':'quali_nochange'}, serializer_class=QActionNoChangeSerializer)
perms_map = {'post':'qaction_create'}, serializer_class=QActionNoChangeSerializer)
@transaction.atomic
def quali_nochange(self, request, *args, **kwargs):
"""

View File

@ -378,25 +378,28 @@ class UserViewSet(PageOrNot, ModelViewSet):
i = 3
role1 = Role.objects.get(name='能力查询')
role2 = Role.objects.get(name='资质能力报送')
role3 = Role.objects.get(name='资料分享和视频培训')
role3 = Role.objects.get(name='资料分享')
role4 = Role.objects.get(name='视频培训')
while sheet['b'+str(i)].value:
name = sheet['b'+str(i)].value
email = sheet['e'+str(i)].value
if not User.objects.filter(username=email).exists():
dept = Organization.objects.get(name=sheet['j'+str(i)].value)
user = User.objects.create(name=name,
username=email,
password=make_password(''.join(random.sample(string.ascii_letters + string.digits, 8))),
dept=dept)
if sheet['f'+str(i)].value:
user.roles.add(role1)
if sheet['g'+str(i)].value:
user.roles.add(role2)
if sheet['h'+str(i)].value:
user.roles.add(role3)
print(str(i)+name+'已导入')
else:
print(str(i)+name+'已存在')
user = User.objects.get(username=email)
dept = Organization.objects.get(name=sheet['j'+str(i)].value)
if sheet['f'+str(i)].value:
user.roles.add(role1)
if sheet['g'+str(i)].value:
user.roles.add(role2)
if sheet['h'+str(i)].value:
user.roles.add(role3)
if sheet['i'+str(i)].value:
user.roles.add(role4)
print(str(i)+name+'已导入')
i = i + 1
return Response()