def new_user():
form = UserForm()
if form.validate_on_submit():
form.save()
flash('Created user.', 'success')
return form.redirect(url_for('admin.index'))
return render_template('admin/user_form.html', title='New user', form=form)
def test_save(self):
form = UserForm(DummyPostData(email='*****@*****.**', name='Foo Bar', password='******'))
form.save()
user = User.query.first()
self.assertEqual(user.email, '*****@*****.**')
self.assertEqual(user.name, 'Foo Bar')
self.assertNotEqual(user.password, 'foobar')
def form (request, id=False):
if id:
instance = get_object_or_404(User, id=id, company=request.user.company)
msg = _("User successfully added")
else:
instance = User()
msg = _("New user successfully added")
#Save and set to active, require valid form
if request.method == 'POST':
form = UserForm(request.POST, instance=instance)
if form.is_valid():
o = form.save(commit=False)
new = False
if not o.id:
new = True
o.save()
form.save_m2m()
#Sets company to current_user company
if not o.get_company():
o.set_company()
if new:
#send new generated password to the new user
generate_new_password_for_user(o)
#Add the new user to allemployee group of the company
if Core.current_user().get_company_allemployeesgroup():
Core.current_user().get_company_allemployeesgroup().add_member(o)
request.message_success(msg)
#Redirects after save for direct editing
return redirect(overview)
else:
form = UserForm(instance=instance)
return render(request, "admin/users/form.html",
{'title': _("User"), 'userCard': instance, 'form': form})
