添加用户登录态校验

redis/secret_key.py

@@ -0,0 +1,4 @@
+import secrets
+
+secret_key = secrets.token_hex(16)  # 生成一个长度为16字节的随机十六进制字符串
+print(secret_key)

redis/test_user_redis.py

@@ -1,29 +1,29 @@
 import json
 import random
 import string
-from flask import Flask, request, jsonify
+from flask import Flask, request, jsonify, session
 from redis import Redis
 
 app = Flask(__name__)
-redis = Redis(host='10.254.13.87', port=6379)
-
-# 生成验证码
+app.secret_key = '333888'
+redis = Redis(host='192.168.3.229', port=6379, password='lizhen-redis')
+# redis = Redis(host='10.254.13.87', port=6379)
+# redis = Redis(host='localhost', port=6379)
 
 
+# 生成验证码
 def generate_verification_code():
     # code = ''.join(random.choices(string.ascii_uppercase + string.digits, k=6))
     code = ''.join(random.choices(string.digits, k=6))
     return code
 
-# 发送验证码到用户邮箱（这里只是模拟发送过程）
-
 
+# 发送验证码到用户邮箱（这里只是模拟发送过程）
 def send_verification_code(email, code):
     print(f'Sending verification code {code} to {email}...')
 
-# 用户请求发送验证码
-
 
+# 用户请求发送验证码
 @app.route('/send_verification_code', methods=['POST'])
 def send_verification_code_endpoint():
     # 从请求中获取邮箱地址
@@ -40,9 +40,8 @@ def send_verification_code_endpoint():
 
     return jsonify({'message': 'Verification code sent'})
 
-# 用户注册
-
 
+# 用户注册
 @app.route('/register', methods=['POST'])
 def register():
     # 从请求中获取注册信息
@@ -55,7 +54,7 @@ def register():
     stored_code = redis.get(email)
     if stored_code is None or verification_code != stored_code.decode('utf-8'):
         return jsonify({'message': 'Invalid verification code'}), 400
-    
+
     # 检查用户名是否已被注册
     if redis.hexists('users', username):
         return jsonify({'message': 'Username already exists'}), 400
@@ -69,9 +68,8 @@ def register():
 
     return jsonify({'message': 'Registration successful'})
 
-# 用户登录
-
 
+# 用户登录
 @app.route('/login', methods=['POST'])
 def login():
     # 从请求中获取登录信息
@@ -87,8 +85,32 @@ def login():
     if password != eval(user_data)['password']:
         return jsonify({'message': 'Invalid password'}), 400
 
+    # 登录验证通过，将用户信息存储到会话中
+    session['username'] = request.json.get('username')
+
     return jsonify({'message': 'Login successful'})
 
 
+# 需要验证登录状态的接口
+@app.route('/protected', methods=['GET'])
+def protected():
+    # 检查会话中的用户信息
+    if 'username' in session:
+        username = session['username']
+        # 其他业务逻辑...
+        return jsonify({'message': f'Hello, {username}! This is a protected endpoint.'})
+    
+    # 如果用户未登录，则返回未授权的响应
+    return jsonify({'message': 'Unauthorized'}), 401
+
+
+# 用户注销
+@app.route('/logout', methods=['POST'])
+def logout():
+    # 清除会话中的用户信息
+    session.pop('username', None)
+    return jsonify({'message': 'Logout successful'})
+
+
 if __name__ == '__main__':
     app.run(debug=True)