131 lines
4.8 KiB
Python
131 lines
4.8 KiB
Python
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 = await UserService.query_item(session, row_dict={"id": body.user_id})
|
|
hotel_cls: HotelModel = await 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,
|
|
"approve_from": "book_hotel",
|
|
}
|
|
# 马上插入审批单信息
|
|
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
|