def change_key(): form = ChangeKeyForm() if request.method == 'POST': if form.validate_on_submit(): key = Key.get_by_id(1) key.key = generate_password_hash(form.ckey.data) key.save() flash('<script>alert("密码修改成功")</script>') return redirect(url_for('home')) else: flash('<script>alert("密码修改失败,请确保两次输入一致")</script>') return render_template('change_key.html', form=form)
def login(): if request.method == 'GET' and session.get('logged_in'): return redirect_back('home') form = KeyForm() if request.method == 'POST' and form.validate_on_submit(): key = form.key.data if check_password_hash(Key.get_by_id(1).key, key): session['logged_in'] = True flash('<script>alert("登陆认证成功")</script>') return redirect(url_for('home')) else: flash('<script>alert("密码验证失败")</script>') return render_template('login.html', form=form)