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)