mirror of
https://github.com/fastapi-practices/fastapi_best_architecture.git
synced 2025-08-19 07:21:31 +08:00
a22aaa0a3e
* Attempt to serialize performance optimization * Add casbin service functions return type * update comments
71 lines
2.7 KiB
Python
71 lines
2.7 KiB
Python
#!/usr/bin/env python3
|
|
# -*- coding: utf-8 -*-
|
|
from typing import Annotated
|
|
|
|
from fastapi import APIRouter, Depends, Query, Request
|
|
from fastapi.security import OAuth2PasswordRequestForm
|
|
from fastapi_limiter.depends import RateLimiter
|
|
from starlette.background import BackgroundTasks
|
|
|
|
from backend.app.common.jwt import DependsJwtAuth
|
|
from backend.app.common.response.response_schema import ResponseModel, response_base
|
|
from backend.app.schemas.token import GetLoginToken, GetNewToken, GetSwaggerToken
|
|
from backend.app.schemas.user import AuthLogin
|
|
from backend.app.services.auth_service import AuthService
|
|
|
|
router = APIRouter()
|
|
|
|
|
|
@router.post(
|
|
'/swagger_login',
|
|
summary='swagger 表单登录',
|
|
description='form 格式登录，用于 swagger 文档调试以及获取 JWT Auth',
|
|
deprecated=True,
|
|
)
|
|
async def swagger_user_login(form_data: OAuth2PasswordRequestForm = Depends()) -> GetSwaggerToken:
|
|
token, user = await AuthService().swagger_login(form_data=form_data)
|
|
return GetSwaggerToken(access_token=token, user=user) # type: ignore
|
|
|
|
|
|
@router.post(
|
|
'/login',
|
|
summary='用户登录',
|
|
description='json 格式登录, 仅支持在第三方api工具调试, 例如: postman',
|
|
dependencies=[Depends(RateLimiter(times=5, minutes=1))],
|
|
)
|
|
async def user_login(request: Request, obj: AuthLogin, background_tasks: BackgroundTasks) -> ResponseModel:
|
|
access_token, refresh_token, access_expire, refresh_expire, user = await AuthService().login(
|
|
request=request, obj=obj, background_tasks=background_tasks
|
|
)
|
|
data = GetLoginToken(
|
|
access_token=access_token,
|
|
refresh_token=refresh_token,
|
|
access_token_expire_time=access_expire,
|
|
refresh_token_expire_time=refresh_expire,
|
|
user=user, # type: ignore
|
|
)
|
|
return await response_base.success(data=data)
|
|
|
|
|
|
@router.post('/new_token', summary='创建新 token', dependencies=[DependsJwtAuth])
|
|
async def create_new_token(request: Request, refresh_token: Annotated[str, Query(...)]) -> ResponseModel:
|
|
(
|
|
new_access_token,
|
|
new_refresh_token,
|
|
new_access_token_expire_time,
|
|
new_refresh_token_expire_time,
|
|
) = await AuthService.new_token(request=request, refresh_token=refresh_token)
|
|
data = GetNewToken(
|
|
access_token=new_access_token,
|
|
access_token_expire_time=new_access_token_expire_time,
|
|
refresh_token=new_refresh_token,
|
|
refresh_token_expire_time=new_refresh_token_expire_time,
|
|
)
|
|
return await response_base.success(data=data)
|
|
|
|
|
|
@router.post('/logout', summary='用户登出', dependencies=[DependsJwtAuth])
|
|
async def user_logout(request: Request) -> ResponseModel:
|
|
await AuthService.logout(request=request)
|
|
return await response_base.success()
|