diff --git a/app/app.py b/app/app.py index e28824b..0fc7935 100644 --- a/app/app.py +++ b/app/app.py @@ -24,19 +24,23 @@ def register(): -@app.route('/login', methods=['POST']) +@app.route('/login', methods=['GET', 'POST']) def login(): - username = request.form['username'] - password = request.form['password'] - print(username,password) - # 验证用户名和密码... - if valid_login(username, password): - # 登录成功 - session['username'] = username - return jsonify(success=True, message="登录成功") + if request.method == 'GET': + return render_template('login.html') else: - # 登录失败 - return jsonify(success=False, message="无效的用户名或密码") + username = request.form['username'] + password = request.form['password'] + print(username, password) + # 验证用户名和密码... + if valid_login(username, password): + # 登录成功 + session['username'] = username + return jsonify(success=True, message="登录成功") + else: + # 登录失败 + return jsonify(success=False, message="无效的用户名或密码") + def valid_login(username, password): @@ -51,9 +55,15 @@ def forget_page(): def home(): if 'username' in session: return render_template('home.html') - else: - return redirect(url_for('/')) # 如果用户未登录,重定向到登录页面 + return redirect("login") + +@app.route('/logout') +def logout(): + # 清除session中的所有信息 + session.clear() + # 返回一个响应,或者重定向到登录页面 + return redirect('/login') if __name__ == '__main__': app.run(debug=True)