优化验证码存储

app/api/users/views.py

 import pyotp
+import socket
 from typing import List
 from fastapi import Depends, APIRouter, Request
 from datetime import timedelta
@@ -25,7 +26,9 @@ def img_code():
 @router.post("/login")
 def login(form_data: UserLoginForm, db: Session = Depends(get_db)):
     user_info = authenticate_user(db=db, form_data=form_data)
-    code = red.get('VerifyCode')
+    # 函数 gethostname() 返回当前正在执行 Python 的系统主机名
+    res = socket.gethostbyname(socket.gethostname())
+    code = red.get(res + '-VerifyCode')
     if not code:
         return HttpResultResponse(code=500, msg="验证码已过期")
     if form_data.verify.lower() != code.lower():

libs/img_code.py

@@ -3,6 +3,7 @@ import os.path
 import random
 import string
 import requests
+import socket
 from PIL import Image, ImageFont, ImageDraw
 from six import BytesIO
 
@@ -61,7 +62,8 @@ class imageCode():
         buf_str = buf.getvalue()
         img = b"data:image/png;base64," + base64.b64encode(buf_str)
         session.headers['verify'] = code
-        red.set('VerifyCode', code, 100)
+        res = socket.gethostbyname(socket.gethostname())
+        red.set(res + '-VerifyCode', code, 100)
         return img.decode('utf-8')