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 add_user():
userform = UserForm(request.form, csrf_enabled= False)
if userform.validate_on_submit() and request.method == 'POST':
user = User(username= userform.username.data, email= userform.email.data, address= userform.address.data)
user.set_password(userform.password.data)
db.session.add(user)
db.session.commit()
# flash('User Added!')
return redirect(url_for('admin.users'))
return render_template('admin/add_user.html', form= userform, action_url=url_for('admin.add_user'), btn_label="Add User")
def user_add():
form = UserForm()
if form.validate_on_submit():
data = form.data
user = User(name=data["name"], number=data["number"])
db.session.add(user)
db.session.commit()
flash("添加员工成功！", "ok")
return redirect(url_for("admin.user_add"))
return render_template("admin/user_add.html", form=form)
def test_email_validation(self):
user = User(email='*****@*****.**', name='Foo Bar', password='******')
db.session.add(user)
db.session.commit()
form = UserForm(DummyPostData(email='*****@*****.**', name='Foo Baz', password='******'))
self.assertFalse(form.validate())
form = UserForm(DummyPostData(email='*****@*****.**', name='Foo Baz', password='******'), obj=user)
self.assertTrue(form.validate())
def user_view(id):
form = UserForm()
user = User.query.get_or_404(int(id))
form.id = user.id
form.name = user.name
form.email = user.email
form.phone = user.phone
form.face = user.face
form.addtime = user.addtime
form.uuid = user.uuid
form.info = user.info
return render_template("admin/user_view.html", form=form)
def edit_user(user_id=None):
if user_id:
user = User.query.get(user_id)
else:
user = User()
form = UserForm(request.form, user)
if form.validate_on_submit():
form.populate_obj(user)
user.save()
return redirect(url_for('admin.show_users'))
print(form.errors)
return render_template('admin/edit-user.html', form=form, user=user)
def edit_user(id):
user = User.query.filter_by(id = id).one()
userform = UserForm(request.form, csrf_enabled=False, obj= user)
if userform.validate_on_submit() and request.method == 'POST':
user.username = userform.username.data
user.email = userform.email.data
user.address = userform.address.data
user.set_password(userform.password.data)
db.session.merge(user)
db.session.commit()
# flash('User Updated!')
return redirect(url_for('admin.users'))
return render_template('admin/add_user.html', form=userform, action_url="/admin/user/edit/"+str(id), btn_label="Update User")
def admin_user_add():
if request.method == "GET":
return render_template('admin_user_add.html')
else:
forms = UserForm(request.form)
if forms.validate():
datas = forms.data
datas['reg_time'] = datetime.strptime(datas['reg_time'],
'%Y-%m-%d %H:%M:%S.%f')
datas = {**datas, 'is_super': 0}
add_common(db, User, **datas)
data = {"msg": "添加成功", "status": "200"}
else:
data = {"msg": "表单验证失败", "status": "202"}
return jsonify(data)
def edit_user():
user = User.query.filter_by(id= current_user.id).one()
form = UserForm(csrf_enabled= False, obj= user)
if form.validate_on_submit() and request.method == 'POST':
user.username = form.username.data
user.set_password(form.username.data)
user.address = form.address.data
user.email = form.email.data
db.session.merge(user)
db.session.commit()
# flash('User profile updated!')
return redirect(url_for('main.index'))
return render_template('main/edit_user.html',user= current_user, form= form, action_url=url_for('main.edit_user'), btn_label="Update User Profile")
def edit_profil():
form = UserForm(obj=current_user)
if request.method == 'POST':
if form.validate_on_submit:
if form.photo.data and form.photo.data != current_user.photo:
image = uploadImage(form.photo.data, 'upload/users/')
current_user.photo = image
current_user.name = form.name.data
current_user.email = form.email.data
current_user.phone = form.phone.data
current_user.job = form.job.data
current_user.facebook = form.facebook.data
current_user.twitter = form.twitter.data
current_user.github = form.github.data
current_user.linkedin = form.linkedin.data
current_user.bio = form.bio.data
current_user.location = form.location.data
repository.save(current_user)
flash('Compte mis à jour avec succès', 'success')
return redirect(url_for('admin.profil'))
else:
flash('Les champs du formulaire ne sont pas bien remplis', 'error')
else:
return redirect(url_for('admin.profil'))
def updateuser(id=None):
form = UserForm()
user = User.query.get_or_404(id)
if form.validate_on_submit():
if form.validate_on_submit():
data = form.data
u_count = User.query.filter_by(name=data["name"]).count()
if u_count == 1:
flash("用户名已存在", "err")
return redirect(url_for("admin.updateuser", id=id))
user.name = data["name"]
user.pwd = generate_password_hash(data["pwd"])
db.session.add(user)
db.session.commit()
flash("修改用户成功！", "ok")
return redirect(url_for("admin.updateuser", id=id))
return render_template("admin/updateuser.html", form=form, u=user, id=id)
def profil():
form = UserForm(obj=current_user)
passwordForm = PasswordForm()
activities = Activity.query.order_by(Activity.created_at.desc())
colors = ['success', 'info', 'danger', 'warning', 'primary']
return render_template('admin/profils/profil.html',
form=form,
activities=activities,
passwordForm=passwordForm,
random=random,
colors=colors)
def new_user():
"""管理后台创建新用户"""
form = UserForm()
if current_user.role_id == 3: # 普通用户无权创建用户
return abort(403)
if request.method == "POST":
password = form.password.data
password2 = form.password2.data
email = form.email.data
user_kind = form.user_kind.data
username = form.username.data
if password2 == password and email and user_kind and username:
new_user = User(username=username,
password=password,
email=email,
role_id=int(user_kind))
db.session.add(new_user)
db.session.commit()
return "OK"
return render_template("admin/create_user.html", form=form)
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})
def create_user():
form = UserForm()
if form.validate_on_submit():
return form
else:
return 'fail'
