From d2080e318982480bc6e1f194691ad62581db56a5 Mon Sep 17 00:00:00 2001
From: wangsiyuan <2392948297@qq.com>
Date: Sat, 23 Dec 2023 16:05:48 +0800
Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=96=B0=20app.py?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 app/app.py | 43 +++++++++++++++++++++++++++++++++++++++----
 1 file changed, 39 insertions(+), 4 deletions(-)

diff --git a/app/app.py b/app/app.py
index f32501e..610f96f 100644
--- a/app/app.py
+++ b/app/app.py
@@ -1,11 +1,46 @@
-from flask import Flask, render_template
+from flask import Flask, request, redirect, url_for, render_template, session, jsonify
 
 app = Flask(__name__)
-
+app.secret_key = 'sUNiJ7QPulxrbmZD'
 
 @app.route('/')
+def index():
+    # 如果用户已登录，则重定向到主页；否则，重定向到登录页面
+    if 'username' in session:
+        return redirect(url_for('home'))
+    else:
+        return render_template('login.html')
+@app.route('/register', methods=['GET','POST'])
+def register():
+    return render_template('register.html')
+
+
+@app.route('/login', methods=['POST'])
 def login():
-    return render_template('login.html')
+    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):
+    # 这里应该是验证用户名和密码的逻辑，比如查询数据库等等
+    # 假设用户名是admin且密码是secret
+    return username == 'admin' and password == '1'
+
+@app.route('/home')
+def home():
+    if 'username' in session:
+        return '你已登录，欢迎, %s!' % session['username']
+    else:
+        return redirect(url_for('login'))  # 如果用户未登录，重定向到登录页面
 
 if __name__ == '__main__':
-    app.run()
+    app.run(debug=True)