user-wsc/AppUser/Router/DepartmentRouter.py

45 lines
1.4 KiB
Python

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