rag-gateway/app/api/label.py

# coding:utf-8

from fastapi import APIRouter, Depends
from app.api import Response, get_current_user
from app.models.base_model import get_db
from app.models.label_model import LabelData, LabelModel, SignLabelData
from app.models.user_model import UserModel
from app.service.label import create_label_service, label_list_service, delete_role_service, sign_label_service

label_router = APIRouter()

@label_router.get("/list", response_model=Response)
async def get_label_list(keyword="", labelType=1, objectId="",current_user: UserModel = Depends(get_current_user),
                          db=Depends(get_db)):

    return Response(code=200, msg="", data=await label_list_service(db, keyword, labelType, objectId))


@label_router.post("/add_label", response_model=Response)
async def add_label_api(label: LabelData, current_user: UserModel = Depends(get_current_user), db=Depends(get_db)):
    if not label.labelName:
        return Response(code=400, msg="The labelName cannot be empty!")
    db_role = db.query(LabelModel).filter(LabelModel.name == label.labelName).first()
    if db_role:
        return Response(code=400, msg="label already created")
    is_create = await create_label_service(db, label.labelName, label.labelType, current_user.id)
    if not is_create:
        return Response(code=500, msg="label create failure", data={})
    return Response(code=200, msg="label create successfully", data={"roleName": label.labelName})


@label_router.delete("/delete_label", response_model=Response)
async def delete_label_api(labelId: int, current_user: UserModel = Depends(get_current_user), db=Depends(get_db)):
    is_delete = await delete_role_service(db, labelId)
    if not is_delete:
        return Response(code=500, msg="label delete failure", data={})
    return Response(code=200, msg="label delete successfully", data={})


@label_router.post("/sign_label", response_model=Response)
async def sign_label_api(sign: SignLabelData, current_user: UserModel = Depends(get_current_user), db=Depends(get_db)):
    is_add = await sign_label_service(db, sign.objectId, sign.labelIdList)
    if not is_add:
        return Response(code=500, msg="label add failure", data={})
    return Response(code=200, msg="label sign add successfully", data={})