feat: 审批接口，需要根据类型来判断是调用报销单审批流程还是酒店预定流程

2025-08-24 22:50:21 +08:00
parent 9801179183
commit a624a92c47
1 changed files with 129 additions and 0 deletions
@@ -0,0 +1,129 @@
+from fastapi import FastAPI
+from langchain_core.runnables import RunnableConfig
+from langgraph.constants import START
+from langgraph.graph import StateGraph
+from pydantic import BaseModel
+
+from app.model.ApproveModel import ApproveModel, ApproveService
+from app.model.HotelModel import HotelService, HotelModel
+from app.model.OrderModel import OrderService
+from app.model.ReimburseModel import ReimburseService
+from app.model.ReimburseOtherModel import ReimburseOtherService
+from app.model.UserModel import UserServiceModel, UserService
+from app.utils.db_utils import AsyncSessionDep
+from app.utils.postgres_checkpointer import AsyncPostgresSaverDep
+from app.workflow.approve import create_approve_graph, ApproveGraphSchema
+
+
+def add_hotel_route(app: FastAPI):
+  @app.post('/book_hotel')
+  async def book_hotel(
+    body: BookHotelSchema,
+    session: AsyncSessionDep,
+    checkpointer: AsyncPostgresSaverDep,
+  ):
+    user_cls: UserServiceModel = UserService.query_item(session, row_dict={"id": body.user_id})
+    hotel_cls: HotelModel = HotelService.query_item(session, row_dict={"id": body.hotel_id})
+
+    # 新建审批单信息
+    new_approve_dict = {
+      "title": f"{user_cls.full_name} - 酒店预定审批",
+      "description": f"酒店预定信息：{hotel_cls.title}, {hotel_cls.description}, {hotel_cls.discount_price}",
+      "status": "approving",
+      "amount": hotel_cls.discount_price,
+      "logs": "[]",
+      "user_id": body.user_id,  # 先把审批人的id设置为申请人
+      "proj_id": body.proj_id,
+    }
+    # 马上插入审批单信息
+    insert_approve_cls: ApproveModel = await ApproveService.item_insert(session=session, row_dict=new_approve_dict)
+
+    book_hotel_graph = create_book_hotel_graph(session=session, checkpointer=checkpointer)
+
+    # 审批通过之后，要创建的报销单信息
+    input_reimburse_dict = {
+      "title": f"{user_cls.full_name} - 酒店预定",
+      "remarks": f"酒店预定信息：{hotel_cls.title}, {hotel_cls.description}, {hotel_cls.discount_price}",
+      "proj_id": body.proj_id,
+      "approve_id": insert_approve_cls.id,
+      "user_id": body.user_id,
+    }
+    input_reimburse_other_dict = {
+      "title": f"{user_cls.full_name} - 酒店预定",
+      "type": "酒店宾馆",
+      "amount": str(hotel_cls.discount_price),
+      "recipy_type": "电子普通发票",
+      "reimburse_id": "",
+    }
+
+    graph_state = await book_hotel_graph.ainvoke(
+      {
+        "input_user_id": body.user_id,
+        "input_approve_id": insert_approve_cls.id,
+        "input_amount": str(insert_approve_cls.amount),
+        "input_reimburse_dict": input_reimburse_dict,
+        "input_hotel_dict": hotel_cls.model_dump(),
+        "input_reimburse_other_dict": input_reimburse_other_dict,
+      },
+      config={"configurable": {"thread_id": insert_approve_cls.id}}
+    )
+
+    return {"message": "已经提交审批单，审批通过之后将自动创建订单！", "graph_state": graph_state}
+
+
+class BookHotelSchema(BaseModel):
+  hotel_id: str
+  user_id: str
+  proj_id: str
+
+
+def create_book_hotel_graph(
+  checkpointer: AsyncPostgresSaverDep,
+  session: AsyncSessionDep,
+):
+  approve_graph = create_approve_graph(checkpointer, session)
+
+  builder = StateGraph(BookHotelGraphSchema)
+
+  async def node(state: BookHotelGraphSchema, config: RunnableConfig):
+    print('\n\n', ":::::::::::node start::::::::::::", '\n\n')
+    print(state)
+    print('\n\n', ":::::::::::node end::::::::::::", '\n\n')
+
+    graph_state = await approve_graph.ainvoke(state, config=config)
+    approve_flag = graph_state.get('approve_flag')
+
+    if approve_flag:
+      # 酒店预定，先走审批流程，审批通过之后：
+      # 1. 创建报销单
+      # 2. 将审批单与报销单关联
+      # 3. 创建订单
+
+      # 新建报销单信息
+      input_reimburse_dict = state.get('input_reimburse_dict')
+      insert_reimburse_cls = await ReimburseService.item_insert(session=session, row_dict=input_reimburse_dict)
+
+      # 新建报销单其他费用信息
+      input_reimburse_other_dict = state.get('input_reimburse_other_dict')
+      input_reimburse_other_dict['reimburse_id'] = insert_reimburse_cls.id
+      await ReimburseOtherService.item_insert(session=session, row_dict=input_reimburse_other_dict)
+
+      # 新建订单信息
+      insert_order_dict = {
+        "name": state.get('input_hotel_dict').get('title'),
+        "price": state.get('input_hotel_dict').get('discount_price'),
+      }
+      await OrderService.item_insert(session=session, row_dict=insert_order_dict)
+
+    return {}
+
+  builder.add_node("node", node)
+  builder.add_edge(START, "node")
+
+  return builder.compile(checkpointer=checkpointer)
+
+
+class BookHotelGraphSchema(ApproveGraphSchema):
+  input_hotel_dict: dict
+  input_reimburse_dict: dict
+  input_reimburse_other_dict: dict