mirror of
https://github.com/fastapi-practices/fastapi_best_architecture.git
synced 2025-08-15 03:52:54 +08:00
44befcc79f
* Add query users by department ID * fix dept parent id schema field * fix dept get all select * update the test sql
110 lines
4.3 KiB
Python
110 lines
4.3 KiB
Python
#!/usr/bin/env python3
|
|
# -*- coding: utf-8 -*-
|
|
from typing import Annotated
|
|
|
|
from fastapi import APIRouter, Query, Request
|
|
|
|
from backend.app.common.casbin_rbac import DependsRBAC
|
|
from backend.app.common.jwt import DependsJwtAuth
|
|
from backend.app.common.pagination import paging_data, PageDepends
|
|
from backend.app.common.response.response_schema import response_base
|
|
from backend.app.database.db_mysql import CurrentSession
|
|
from backend.app.schemas.user import CreateUser, GetAllUserInfo, ResetPassword, UpdateUser, Avatar, GetCurrentUserInfo
|
|
from backend.app.services.user_service import UserService
|
|
from backend.app.utils.serializers import select_to_json
|
|
|
|
router = APIRouter()
|
|
|
|
|
|
@router.post('/register', summary='用户注册')
|
|
async def user_register(obj: CreateUser):
|
|
await UserService.register(obj=obj)
|
|
return await response_base.success()
|
|
|
|
|
|
@router.post('/password/reset', summary='密码重置', dependencies=[DependsJwtAuth])
|
|
async def password_reset(request: Request, obj: ResetPassword):
|
|
count = await UserService.pwd_reset(request=request, obj=obj)
|
|
if count > 0:
|
|
return await response_base.success()
|
|
return await response_base.fail()
|
|
|
|
|
|
@router.get('/me', summary='获取当前用户信息', dependencies=[DependsJwtAuth])
|
|
async def current_userinfo(request: Request):
|
|
data = GetCurrentUserInfo(**select_to_json(request.user))
|
|
return await response_base.success(data=data, exclude={'password'})
|
|
|
|
|
|
@router.get('/{username}', summary='查看用户信息', dependencies=[DependsJwtAuth])
|
|
async def get_user(username: str):
|
|
current_user = await UserService.get_userinfo(username=username)
|
|
data = GetAllUserInfo(**select_to_json(current_user))
|
|
return await response_base.success(data=data)
|
|
|
|
|
|
@router.put('/{username}', summary='更新用户信息', dependencies=[DependsJwtAuth])
|
|
async def update_userinfo(request: Request, username: str, obj: UpdateUser):
|
|
count = await UserService.update(request=request, username=username, obj=obj)
|
|
if count > 0:
|
|
return await response_base.success()
|
|
return await response_base.fail()
|
|
|
|
|
|
@router.put('/{username}/avatar', summary='更新头像', dependencies=[DependsJwtAuth])
|
|
async def update_avatar(request: Request, username: str, avatar: Avatar):
|
|
count = await UserService.update_avatar(request=request, username=username, avatar=avatar)
|
|
if count > 0:
|
|
return await response_base.success()
|
|
return await response_base.fail()
|
|
|
|
|
|
@router.get('', summary='（模糊条件）分页获取所有用户', dependencies=[DependsJwtAuth, PageDepends])
|
|
async def get_all_users(
|
|
db: CurrentSession,
|
|
dept: Annotated[int | None, Query()] = None,
|
|
username: Annotated[str | None, Query()] = None,
|
|
phone: Annotated[str | None, Query()] = None,
|
|
status: Annotated[int | None, Query()] = None,
|
|
):
|
|
user_select = await UserService.get_select(dept=dept, username=username, phone=phone, status=status)
|
|
page_data = await paging_data(db, user_select, GetAllUserInfo)
|
|
return await response_base.success(data=page_data)
|
|
|
|
|
|
@router.post('/{pk}/super', summary='修改用户超级权限', dependencies=[DependsRBAC])
|
|
async def super_set(request: Request, pk: int):
|
|
count = await UserService.update_permission(request=request, pk=pk)
|
|
if count > 0:
|
|
return await response_base.success()
|
|
return await response_base.fail()
|
|
|
|
|
|
@router.post('/{pk}/action', summary='修改用户状态', dependencies=[DependsRBAC])
|
|
async def status_set(request: Request, pk: int):
|
|
count = await UserService.update_status(request=request, pk=pk)
|
|
if count > 0:
|
|
return await response_base.success()
|
|
return await response_base.fail()
|
|
|
|
|
|
@router.post('/{pk}/multi', summary='修改用户多点登录状态', dependencies=[DependsRBAC])
|
|
async def multi_set(request: Request, pk: int):
|
|
count = await UserService.update_multi_login(request=request, pk=pk)
|
|
if count > 0:
|
|
return await response_base.success()
|
|
return await response_base.fail()
|
|
|
|
|
|
@router.delete(
|
|
path='/{username}',
|
|
summary='用户注销',
|
|
description='用户注销 != 用户登出，注销之后用户将从数据库删除',
|
|
dependencies=[DependsRBAC],
|
|
)
|
|
async def delete_user(request: Request, username: str):
|
|
count = await UserService.delete(request=request, username=username)
|
|
if count > 0:
|
|
return await response_base.success()
|
|
return await response_base.fail()
|