添加重置密码

redis/test_user_redis.py

@@ -14,8 +14,8 @@ SECERT_KEY = "8U2LL1"
 MY_OPENAI_API_KEY = os.environ.get('MY_OPENAI_API_KEY')
 
 app = Flask(__name__)
-redis = Redis(host='192.168.3.229', port=6379, password='lizhen-redis')
-# redis = Redis(host='10.254.13.87', port=6379)
+# 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)
 
 
@@ -57,6 +57,9 @@ def register():
     password = request.json.get('password')
     verification_code = request.json.get('verification_code')
 
+    if is_email_registered(email):
+        return jsonify({'code': 400, 'message': '邮箱已被注册'})
+
     # 检查验证码是否匹配
     stored_code = redis.get(email)
     if stored_code is None or verification_code != stored_code.decode('utf-8'):
@@ -78,6 +81,9 @@ def register():
     }
     redis.hset('users', username, json.dumps(user_data))
 
+    # 清除验证码
+    redis.delete(email)
+
     return jsonify({
         'code': 0,
         'message': 'Registration successful'
@@ -123,6 +129,26 @@ def protected():
     return jsonify({'code': 401, 'message': 'Unauthorized'})
 
 
+@app.route('/resetPassword', methods=['POST'])
+def reset_password():
+    email = request.json.get('email')
+    verification_code = request.json.get('verification_code')
+    new_password = request.json.get('new_password')
+    update_password(email, new_password)
+
+    if not is_email_registered(email):
+        return jsonify({'code': 400, 'message': '邮箱未注册'})
+
+    # 检查验证码是否匹配
+    stored_code = redis.get(email)
+    if stored_code is None or verification_code != stored_code.decode('utf-8'):
+        return jsonify({'code': 400, 'message': 'Invalid verification code'})
+
+    update_password(email, new_password)
+    redis.delete(email)
+    return jsonify({'code': 0, 'message': '密码已更新'})
+
+
 # 用户注销
 @app.route('/logout', methods=['POST'])
 def logout():
@@ -402,5 +428,27 @@ def exceeded_chat_limit(user_id, package, model):
     return False
 
 
+# 检查邮箱是否已注册
+def is_email_registered(email):
+    # 遍历所有用户数据
+    for username in redis.hkeys('users'):
+        user_data = json.loads(redis.hget('users', username).decode('utf-8'))
+        if user_data['email'] == email:
+            return True
+    return False
+
+
+# 更新用户密码
+def update_password(email, new_password):
+    # 遍历所有用户数据
+    for username in redis.hkeys('users'):
+        user_data = json.loads(redis.hget('users', username).decode('utf-8'))
+        if user_data['email'] == email:
+            user_data['password'] = new_password
+            redis.hset('users', username, json.dumps(user_data))
+            return True
+    return False
+
+
 if __name__ == '__main__':
     app.run(debug=True)