生成验证码、注册、登录

redis/test_user_redis.py

@@ -1,38 +1,91 @@
-import redis
+import json
+import random
+import string
 from flask import Flask, request, jsonify
+from redis import Redis
 
 app = Flask(__name__)
-db = redis.Redis(host='localhost', port=6379, db=0)
+redis = Redis(host='10.254.13.87', 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():
+    # 从请求中获取邮箱地址
+    email = request.json.get('email')
+
+    # 生成验证码
+    verification_code = generate_verification_code()
+
+    # 发送验证码到用户邮箱
+    send_verification_code(email, verification_code)
+
+    # 保存验证码到Redis，并设置过期时间（例如，5分钟）
+    redis.setex(email, 300, verification_code)
+
+    return jsonify({'message': 'Verification code sent'})
+
+# 用户注册
 
 
 @app.route('/register', methods=['POST'])
 def register():
-    data = request.get_json()
-    username = data['username']
-    password = data['password']
+    # 从请求中获取注册信息
+    email = request.json.get('email')
+    username = request.json.get('username')
+    password = request.json.get('password')
+    verification_code = request.json.get('verification_code')
 
-    # 检查用户名是否已存在
-    if db.exists(username):
+    # 检查验证码是否匹配
+    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
 
-    # 在 Redis 中保存用户名和密码
-    db.hset(username, mapping={'password': password})
+    # 保存用户信息到Redis
+    user_data = {
+        'email': email,
+        'password': password
+    }
+    redis.hset('users', username, json.dumps(user_data))
+
+    return jsonify({'message': 'Registration successful'})
 
-    return jsonify({'message': 'User registered successfully'}), 201
+# 用户登录
 
 
 @app.route('/login', methods=['POST'])
 def login():
-    data = request.get_json()
-    username = data['username']
-    password = data['password']
-
-    # 获取保存在 Redis 中的密码
-    saved_password = db.hget(username, 'password')
+    # 从请求中获取登录信息
+    username = request.json.get('username')
+    password = request.json.get('password')
 
     # 检查用户名和密码是否匹配
-    if saved_password is None or saved_password.decode() != password:
-        return jsonify({'message': 'Invalid username or password'}), 401
+    user_data = redis.hget('users', username)
+    if not user_data:
+        return jsonify({'message': 'Invalid username'}), 400
+
+    user_data = user_data.decode('utf-8')
+    if password != eval(user_data)['password']:
+        return jsonify({'message': 'Invalid password'}), 400
 
     return jsonify({'message': 'Login successful'})