更新 app.py

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 login():
+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():
+    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)