def tsignin():
form = UserForm()
warn = ''
if form.validate_on_submit():
warn = Methods(form.name.data).add_user(form.pwd.data)
return render_template('todo/tsignin.html', form=form, warn=warn)
def tlogin():
form = UserForm() # 表单对象
warn = ''
if form.validate_on_submit(): # 表单提交
user_data = User.objects(user_name=form.name.data).first() # 数据库用户认证
if user_data:
if form.pwd.data == user_data.user_pwd:
login_user(user_data)
return redirect(url_for('test.todo'))
else:
warn = '密码错误'
else:
warn = '用户不存在'
return render_template('todo/tlogin.html',
title="Sign In",
form=form,
warn=warn)
def user_modify():
form = UserForm(request.form)
if 'id' in request.args:
user = db.session.query(User).get(request.args.get('id'))
form = UserForm(obj=user)
if request.method == "GET":
return render_template('user_modify.html', form=form)
elif request.method == "POST":
if form.validate_on_submit():
if form.id.data:
user = db.session.query(User).get(form.id.data)
user.name = form.name.data
user.alias = form.alias.data
user.email = form.email.data
user.role_id = form.role_id.data
if form.delete_date.data:
user.delete_date = form.delete_date.data
else:
user.delete_date = None
db.session.commit()
flash("user modified")
else:
newuser = User(name=form.name.data,
alias=form.alias.data,
email=form.email.data,
password=None,
role_id=form.role_id.data,
avatar=None,
delete_date=None)
db.session.add(newuser)
db.session.commit()
flash("user added")
return redirect(url_for('user_page'))
else:
flash("wrong data")
return render_template('user_modify.html', form=form)
