From 2f84e3cc4bff981590ab37bd95da64bd72c1f2e0 Mon Sep 17 00:00:00 2001 From: zhaoqingang <zhaoqg0118@163.com> Date: 星期一, 16 十二月 2024 09:40:00 +0800 Subject: [PATCH] 12-16 --- app/api/dialog.py | 41 +++++++++++++++++++++++++++++++++++++---- 1 files changed, 37 insertions(+), 4 deletions(-) diff --git a/app/api/dialog.py b/app/api/dialog.py index c11b4fd..28bd1df 100644 --- a/app/api/dialog.py +++ b/app/api/dialog.py @@ -2,15 +2,48 @@ from fastapi import APIRouter, Depends from app.api import Response, get_current_user, ResponseList -from app.models import RoleParameter, GroupModel, KnowledgeModel +from app.models.dialog_model import dialogData, dialogDataUpdate from app.models.base_model import get_db from app.models.user_model import UserModel -from app.service.dialog import get_dialog_list +from app.service.dialog import get_dialog_list, create_dialog_service, update_dialog_status_service dialog_router = APIRouter() @dialog_router.get("/list", response_model=Response) -async def dialog_list(current_user: UserModel = Depends(get_current_user), +async def dialog_list(current: int, + pageSize: int, + keyword: str = "", + label: int = 0, + status: str ="", + current_user: UserModel = Depends(get_current_user), db=Depends(get_db)): - return Response(code=200, msg="", data=await get_dialog_list(db, current_user.id)) + if current and not pageSize: + return ResponseList(code=400, msg="缂哄皯鍙傛暟") + return Response(code=200, msg="", data=await get_dialog_list(db, current_user.id, keyword, label, status, pageSize, current)) + + +@dialog_router.post("/create", response_model=Response) +async def create_dialog_api(dialog: dialogData, current_user: UserModel = Depends(get_current_user), db=Depends(get_db)): + is_create = await create_dialog_service(db, dialog.id, dialog.name, dialog.description, dialog.icon, dialog.dialogType, dialog.mode,current_user.id) + if not is_create: + return Response(code=500, msg="role create failure", data={}) + return Response(code=200, msg="role create success", data={}) + + +@dialog_router.get("/update", response_model=Response) +async def change_dialog_data(dialog: dialogDataUpdate, current_user: UserModel = Depends(get_current_user), db=Depends(get_db)): + # is_create = await create_dialog_service(db, dialog.id, dialog.name, dialog.description, dialog.icon, dialog.dialogType) + # if not is_create: + # return Response(code=500, msg="role create failure", data={}) + return Response(code=200, msg="role create success", data={}) + + +@dialog_router.put("/status", response_model=Response) +async def change_dialog_status(dialog: dialogDataUpdate, current_user: UserModel = Depends(get_current_user), db=Depends(get_db)): + if dialog.status not in ["0", "1"]: + return Response(code=400, msg="invalid parameter value", data={}) + is_create = await update_dialog_status_service(db, dialog.id, dialog.status) + if not is_create: + return Response(code=500, msg="dialog update failure", data={}) + return Response(code=200, msg="dialog update success", data={}) \ No newline at end of file -- Gitblit v1.8.0