This commit is contained in:
王思川 2022-11-08 14:10:41 +08:00
parent 807b2b2056
commit 046fcadc28
1 changed files with 11 additions and 10 deletions

View File

@ -1,21 +1,22 @@
from fastapi import APIRouter, Depends, HTTPException
from sqlalchemy.orm import Session
from . import Crud, Models, Schemas
from Utils.AccessControl import AccessUtil
from Utils.AccessControl.AccessUtil import rbac
from Utils.DataBase.SqlAlchemyUtils import get_db, engine
Models.Base.metadata.create_all(bind=engine)
router = APIRouter(
prefix="/api/index_store",
tags=['指标']
tags=['指标'],
dependencies=[Depends(rbac)]
)
# 新建指标
# @router.post("/create", summary='新建指标', response_model=Schemas.Indicator, dependencies=[Depends(AccessUtil.ac_admin)])
@router.post("/create", summary='新建指标', response_model=Schemas.Indicator)
def create_indicator(body: Schemas.IndicatorCreate, db: Session = Depends(get_db)):
data = Crud.get_indicator_by_ename(db, ename=body.ename)
@ -33,14 +34,14 @@ def create_indicator(body: Schemas.IndicatorCreate, db: Session = Depends(get_db
# 删除指标
@router.post("/delete/{iid}", summary='删除指标', dependencies=[Depends(AccessUtil.rbac)])
@router.post("/delete/{iid}", summary='删除指标')
def delete_indicator(iid: str, db: Session = Depends(get_db)):
Crud.delete_indicator(db=db, iid=iid)
return {"info": "Success"}
# 编辑指标
@router.post("/edit/{iid}", summary='编辑指标', response_model=Schemas.Indicator, dependencies=[Depends(AccessUtil.rbac)])
@router.post("/edit/{iid}", summary='编辑指标', response_model=Schemas.Indicator)
def edit_indicator(iid: str, body: Schemas.IndicatorEdit, db: Session = Depends(get_db)):
db_indicator = Crud.get_indicator_by_iid(db=db, iid=iid)
if not db_indicator:
@ -49,7 +50,7 @@ def edit_indicator(iid: str, body: Schemas.IndicatorEdit, db: Session = Depends(
# 查看指标
@router.get("/view/{iid}", summary='查看指标', response_model=Schemas.Indicator, dependencies=[Depends(AccessUtil.rbac)])
@router.get("/view/{iid}", summary='查看指标', response_model=Schemas.Indicator)
def read_indicator(iid: str, db: Session = Depends(get_db)):
data = Crud.get_indicator_by_iid(db, iid=iid)
if data is None:
@ -58,27 +59,27 @@ def read_indicator(iid: str, db: Session = Depends(get_db)):
# 查询指标描述
@router.post("/describe/{iid}", summary='查看指标描述', dependencies=[Depends(AccessUtil.rbac)])
@router.post("/describe/{iid}", summary='查看指标描述')
def indicators_description(iid: str, db: Session = Depends(get_db)):
indicators = Crud.get_indicator_description_by_iid(db, iid=iid)
return indicators
# 查询指标
@router.post("/search", summary='查询指标', dependencies=[Depends(AccessUtil.rbac)])
@router.post("/search", summary='查询指标')
def read_indicators(body: Schemas.IndicatorSearch, db: Session = Depends(get_db)):
indicators, total = Crud.search_indicators(db, body=body, page=body.page, pagesize=body.pagesize)
return {"items": indicators, "total": total}
# 新建参数
@router.post("/param/create/{iid}", summary='新建参数', response_model=Schemas.Parameter, dependencies=[Depends(AccessUtil.rbac)])
@router.post("/param/create/{iid}", summary='新建参数', response_model=Schemas.Parameter)
def create_index_param(iid: str, body: Schemas.ParameterEdit, db: Session = Depends(get_db)):
return Crud.create_indicator_parameter(db=db, body=body, _iid=iid)
# 删除参数
@router.post("/param/delete/{pid}", summary='删除参数', dependencies=[Depends(AccessUtil.rbac)])
@router.post("/param/delete/{pid}", summary='删除参数')
def delete_parameters_for_indicator(pid: str, db: Session = Depends(get_db)):
data = Crud.get_parameter_by_pid(db=db, pid=pid)
if not data: