insistence
7 months ago
2 changed files with 26 additions and 13 deletions
@ -1,28 +1,41 @@ |
|||
import uuid |
|||
from datetime import timedelta |
|||
from fastapi import APIRouter, Request |
|||
from config.env import RedisInitKeyConfig |
|||
from module_admin.service.captcha_service import * |
|||
from module_admin.entity.vo.login_vo import CaptchaCode |
|||
from utils.response_util import * |
|||
from utils.log_util import * |
|||
from datetime import timedelta |
|||
from module_admin.service.captcha_service import CaptchaService |
|||
from utils.response_util import ResponseUtil |
|||
from utils.log_util import logger |
|||
|
|||
|
|||
captchaController = APIRouter() |
|||
|
|||
|
|||
@captchaController.get("/captchaImage") |
|||
@captchaController.get('/captchaImage') |
|||
async def get_captcha_image(request: Request): |
|||
captcha_enabled = True if await request.app.state.redis.get(f"{RedisInitKeyConfig.SYS_CONFIG.get('key')}:sys.account.captchaEnabled") == 'true' else False |
|||
register_enabled = True if await request.app.state.redis.get( |
|||
f"{RedisInitKeyConfig.SYS_CONFIG.get('key')}:sys.account.registerUser") == 'true' else False |
|||
captcha_enabled = ( |
|||
True |
|||
if await request.app.state.redis.get(f"{RedisInitKeyConfig.SYS_CONFIG.get('key')}:sys.account.captchaEnabled") |
|||
== 'true' |
|||
else False |
|||
) |
|||
register_enabled = ( |
|||
True |
|||
if await request.app.state.redis.get(f"{RedisInitKeyConfig.SYS_CONFIG.get('key')}:sys.account.registerUser") |
|||
== 'true' |
|||
else False |
|||
) |
|||
session_id = str(uuid.uuid4()) |
|||
captcha_result = await CaptchaService.create_captcha_image_service() |
|||
image = captcha_result[0] |
|||
computed_result = captcha_result[1] |
|||
await request.app.state.redis.set(f"{RedisInitKeyConfig.CAPTCHA_CODES.get('key')}:{session_id}", computed_result, ex=timedelta(minutes=2)) |
|||
await request.app.state.redis.set( |
|||
f"{RedisInitKeyConfig.CAPTCHA_CODES.get('key')}:{session_id}", computed_result, ex=timedelta(minutes=2) |
|||
) |
|||
logger.info(f'编号为{session_id}的会话获取图片验证码成功') |
|||
|
|||
return ResponseUtil.success( |
|||
model_content=CaptchaCode(captchaEnabled=captcha_enabled, registerEnabled=register_enabled, img=image, uuid=session_id) |
|||
model_content=CaptchaCode( |
|||
captchaEnabled=captcha_enabled, registerEnabled=register_enabled, img=image, uuid=session_id |
|||
) |
|||
) |
|||
|
Loading…
Reference in new issue