From a624a92c473af9b9b1bca905cc8a68cd42bc7735 Mon Sep 17 00:00:00 2001
From: martsforever <martsforever@aliyun.com>
Date: Sun, 24 Aug 2025 22:50:21 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E5=AE=A1=E6=89=B9=E6=8E=A5=E5=8F=A3?=
 =?UTF-8?q?=EF=BC=8C=E9=9C=80=E8=A6=81=E6=A0=B9=E6=8D=AE=E7=B1=BB=E5=9E=8B?=
 =?UTF-8?q?=E6=9D=A5=E5=88=A4=E6=96=AD=E6=98=AF=E8=B0=83=E7=94=A8=E6=8A=A5?=
 =?UTF-8?q?=E9=94=80=E5=8D=95=E5=AE=A1=E6=89=B9=E6=B5=81=E7=A8=8B=E8=BF=98?=
 =?UTF-8?q?=E6=98=AF=E9=85=92=E5=BA=97=E9=A2=84=E5=AE=9A=E6=B5=81=E7=A8=8B?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 app/controller/add_hotel_route.py | 129 ++++++++++++++++++++++++++++++
 1 file changed, 129 insertions(+)
 create mode 100644 app/controller/add_hotel_route.py

diff --git a/app/controller/add_hotel_route.py b/app/controller/add_hotel_route.py
new file mode 100644
index 0000000..1568ba3
--- /dev/null
+++ b/app/controller/add_hotel_route.py
@@ -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