def test_approve_user_(self, task_mail, configure_mail_function): default_role = Role(name='default', description='n/a') new_role = Role(name='new role', description='n/a') db.session.add(default_role) db.session.add(new_role) user_1 = User( username='******', email='*****@*****.**', password='******', first_name='Test', last_name='User', institution='TEST CI', ) user_1.roles.append(default_role) db.session.add(user_1) db.session.commit() with self.app_context: approve_user(user_1.id, default_role.id, new_role.id, {}) configure_mail_function.assert_called() task_mail.send.assert_called() self.assertTrue(user_1.is_approved, 'Failed to approve user') self.assertEqual(user_1.roles, [new_role], 'Failed to set Role correctly')
def create_role(request): Role.objects.bulk_create([ Role(name="admin"), Role(name="manager"), Role(name="user"), ]) # return redirect("to_user_role") return redirect("to_user_role")
def create(self, validated_data): user = User(email=validated_data['email'], username=validated_data['username'], first_name=validated_data['first_name'], last_name=validated_data['last_name']) user.set_password(validated_data['password']) user.save() role = Role(user_id=user.id, is_admin=False) role.save() return user
def create(request): if request.method == 'POST': form = UserRegisterForm(request.POST) if form.is_valid(): form.save() messages.success(request, 'User registration by admin is successfully!') role = Role(user_id=User.objects.latest('id').id, role=request.POST.get('role')) role.save() return redirect('setting-index') return render(request, 'user/add.html', {'title': 'New User'})
def handle(self, *args, **kwargs): Role.objects.all().delete() User.objects.all().delete() username = '******' user = User.objects.create_user(username=username, password='******', email='*****@*****.**', first_name='Admin', last_name='Example') user.is_superuser = False user.is_staff = True user.save() role = Role(user_id=user.id, is_admin=True) role.save() for i in range(5): fake = Faker() name = fake.first_name() user = User.objects.create_user(username=name.lower(), password='******', email=fake.email(), first_name=name, last_name=fake.last_name()) user.is_superuser = False user.is_staff = False user.save() role = Role(user_id=user.id, is_admin=False) role.save() self.stdout.write('Added new user')
def post(self, request): ret = dict(result=False) role = Role() role_form = RoleForm(request.POST, instance=role) if role_form.is_valid(): role_form.save() ret['result'] = True return JsonResponse(ret)
def register(request): if request.method == 'POST': form = RegisterForm(request.POST, request.FILES) if form.is_valid(): # 创建 User 对象,并修改密码 user = form.save(commit=False) user.password = make_password(user.password) user.save() Role.add_perm(user.id,'user') # 写入 session 数据 request.session['uid'] = user.id request.session['nickname'] = user.nickname return redirect('/user/info/') else: return render(request, 'register.html', {'error': form.errors}) return render(request, 'register.html')
def handle(self, *args, **kwargs): normal_role = Role.objects.filter(name='normal').first() if not normal_role: normal_role = Role(name='normal') normal_role.save() normal_role.permissions = NORMAL_USER_PERMISSIONS normal_role.save()
def run(self, **kwargs): #check if admin exists a = Role.objects.filter(name='admin').first() if a == None: Role(name='admin').save() u = prompt('Admin Email?', default='*****@*****.**') p = prompt('Admin Password (min 6 characters)?', default='enferno') CreateUserCommand().run(email=u, password=p, active=1) AddRoleCommand().run(user_identifier=u, role_name='admin') else: print 'Seems like an Admin is already installed'
def create_role(request): if len(request.POST['firstname']) < 5: return JsonResponse({'failure': 'Name of Role error'}) try: if 0 < int(request.POST['permission']) < 7: pass else: return JsonResponse({'failure': 'You must select permission'}) except: return JsonResponse({'failure': 'You must select permission'}) if request.POST['description'] == '': return JsonResponse({'failure': 'Description can not be blank'}) new_role = Role(Name=request.POST['firstname'], Permission=request.POST['permission'], Description=request.POST['description']) new_role.save() if new_role: return JsonResponse({'success': 'Create successfully a new Role'}) else: return JsonResponse({'failure': 'Create un-successfully a new Role'})
def run(self, **kwargs): db.create_all() # check if admin exists a = Role.query.filter_by(name='admin').first() if a is None: db.session.add(Role(name='admin')) db.session.commit() u = prompt('Admin Email?', default='*****@*****.**') p = prompt('Admin Password (min 6 characters)?', default='enferno') CreateUserCommand().run(email=u, password=p, active=1) AddRoleCommand().run(user_identifier=u, role_name='admin') else: print 'Seems like an Admin is already installed'
def setUp(self): super().setUp() # Dropping and creating tables on signup test\ # to ensure retest success db.drop_all() db.create_all() Role.insert_roles() Permission.insert_permissions() User.create_first_user() # create default accounts for tests for role in self.user_data: if role == "user": continue user = {"email": self.user_data[role]["email"]} user = User(**user) user.password = self.user_data[role]["password"] user.user_roles.append( UserRole(role=eval(f"Role.get_{role}_role()"), user=user)) db.session.add(user) db.session.commit()
def install(): """Install a default admin user and add an admin role to it. """ #check if admin exists from user.models import Role a = Role.objects.filter(name='admin').first() if a == None: r = Role(name='admin').save() u = click.prompt('Admin Email?', default='*****@*****.**') p = click.prompt('Admin Password (min 6 characters)?', default='enferno') CreateUserCommand().run(email=u, password=p, active=1) AddRoleCommand().run(user_identifier=u, role_name='admin') else: print 'Seems like an Admin is already installed'
def install(): """ Install a default admin user and add an admin role to it. """ #check if admin exists a = Role.query.filter(Role.name == 'admin').first() if a is None: r = Role(name='admin') db.session.add(r) db.session.commit() m = click.prompt('Admin Email?', default='admin@localhost') n = click.prompt('Admin Name ?', default='arduinow') p = click.prompt('Admin Password (min 6 characters)?', default='aarduinow') u = User(email=m, username=n, password=p, active=1) u.roles.append(r) db.session.add(u) db.session.commit() else: print('Seems like an Admin is already installed')