权限代码修改

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

View File

@ -378,25 +378,28 @@ class UserViewSet(PageOrNot, ModelViewSet):
i = 3 i = 3
role1 = Role.objects.get(name='能力查询') role1 = Role.objects.get(name='能力查询')
role2 = 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: while sheet['b'+str(i)].value:
name = sheet['b'+str(i)].value name = sheet['b'+str(i)].value
email = sheet['e'+str(i)].value email = sheet['e'+str(i)].value
if not User.objects.filter(username=email).exists(): if not User.objects.filter(username=email).exists():
dept = Organization.objects.get(name=sheet['j'+str(i)].value)
user = User.objects.create(name=name, user = User.objects.create(name=name,
username=email, username=email,
password=make_password(''.join(random.sample(string.ascii_letters + string.digits, 8))), password=make_password(''.join(random.sample(string.ascii_letters + string.digits, 8))),
dept=dept) 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: 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 i = i + 1
return Response() return Response()