user-wsc/AppUser/Router/DepartmentRouter.py

from fastapi import APIRouter, Depends, HTTPException
from sqlalchemy.orm import Session
from sqlalchemy.exc import IntegrityError

from AppUser.Crud import DepartmentCrud

from Utils.AccessControl import AccessUtil
from Utils.DataBase.SqlAlchemyUtils import get_db

router = APIRouter(
    tags=["部门"],
    prefix="/api/user/department",
    dependencies=[Depends(AccessUtil.rbac)]
)


@router.post("/create", summary="新建部门")
def func01(name: str, db: Session = Depends(get_db)):
    dp_obj = DepartmentCrud.get_department_by_name(db=db, name=name)
    if dp_obj:
        raise HTTPException(status_code=202, detail="Department Already Existed")

    DepartmentCrud.create_department(db=db, name=name)
    return {"info": "Success"}


@router.post("/delete", summary="删除部门")
def func02(dp_id: str, db: Session = Depends(get_db)):
    dp_obj = DepartmentCrud.get_department_by_id(db=db, dp_id=dp_id)
    if not dp_obj:
        raise HTTPException(status_code=404, detail="Department Not Found")

    try:
        DepartmentCrud.delete_department_by_id(db=db, dp_id=dp_id)
    except IntegrityError:
        raise HTTPException(status_code=202, detail="Department Is Being Used")

    return {"info": "Success"}


@router.get("/list", summary="部门列表")
def func03(db: Session = Depends(get_db)):
    role_list = DepartmentCrud.get_department_list(db=db)
    return role_list