mirror of
https://github.com/fastapi-practices/fastapi-best-architecture.git
synced 2026-09-22 21:26:30 +00:00
40 lines
1.3 KiB
Python
40 lines
1.3 KiB
Python
import uuid
|
|
|
|
from fast_captcha import img_captcha
|
|
from fastapi import APIRouter, Depends
|
|
from pyrate_limiter import Duration, Rate
|
|
from starlette.concurrency import run_in_threadpool
|
|
|
|
from backend.app.admin.schema.captcha import GetCaptchaDetail
|
|
from backend.common.response.response_schema import ResponseSchemaModel, response_base
|
|
from backend.core.conf import settings
|
|
from backend.database.db import CurrentSession
|
|
from backend.database.redis import redis_client
|
|
from backend.utils.dynamic_config import load_login_config
|
|
from backend.utils.limiter import RateLimiter
|
|
|
|
router = APIRouter()
|
|
|
|
|
|
@router.get(
|
|
'/captcha',
|
|
summary='获取登录验证码',
|
|
dependencies=[Depends(RateLimiter(Rate(5, Duration.SECOND * 30)))],
|
|
)
|
|
async def get_captcha(db: CurrentSession) -> ResponseSchemaModel[GetCaptchaDetail]:
|
|
await load_login_config(db)
|
|
img, code = await run_in_threadpool(img_captcha, img_byte='base64')
|
|
captcha_uuid = str(uuid.uuid4())
|
|
await redis_client.set(
|
|
f'{settings.LOGIN_CAPTCHA_REDIS_PREFIX}:{captcha_uuid}',
|
|
code,
|
|
ex=settings.LOGIN_CAPTCHA_EXPIRE_SECONDS,
|
|
)
|
|
data = GetCaptchaDetail(
|
|
is_enabled=settings.LOGIN_CAPTCHA_ENABLED,
|
|
expire_seconds=settings.LOGIN_CAPTCHA_EXPIRE_SECONDS,
|
|
uuid=captcha_uuid,
|
|
image=img,
|
|
)
|
|
return response_base.success(data=data)
|