Upgrade verification code

2022-01-14 13:23:41 +08:00
parent 0cd0c414ee
commit 2acc3cce67
4 changed files with 123 additions and 16 deletions
@@ -4,7 +4,7 @@ from io import BytesIO
 from flask import session, make_response, current_app
 from flask_login import current_user

-from applications.common.utils.gen_captcha import gen_captcha
+from applications.common.utils.gen_captcha import vieCode
 from applications.schemas import PowerOutSchema


@@ -61,7 +61,7 @@ def make_menu_tree():

 # 生成验证码
 def get_captcha():
-    code, image = gen_captcha()
+    image, code = vieCode().GetCodeImage()
    out = BytesIO()
    session["code"] = code
    image.save(out, 'png')