from decimal import Decimal
from typing import List

from fastapi import FastAPI

from app.model.ApproveModel import ApproveService, ApproveModel
from app.model.ProjectModel import ProjectModel, ProjectService
from app.model.ReimburseModel import ReimburseService
from app.utils.db_utils import AsyncSessionDep


def add_approve_route(app: FastAPI):
  # 提交报销单接口
  @app.post('/submit_reimburse')
  async def submit_reimburse(reimburse: dict, session: AsyncSessionDep):
    reimburse_id = reimburse.get('id')
    reimburse_cls = await ReimburseService.query_item(
      session=session,
      row_dict={"id": reimburse_id}
    )

    return

  @app.post('/query_spent')
  async def query_spent(session: AsyncSessionDep, proj_id_list: List[str]):
    result = {item: Decimal(0) for item in proj_id_list}
    approve_cls_list = (await session.execute(ApproveService.select_cls().where(ApproveModel.proj_id.in_(proj_id_list)))).scalars().all()
    for approve_item in approve_cls_list:
      if approve_item.status == 'approved':
        result[approve_item.proj_id] += Decimal(approve_item.amount)

    project_cls_list = (await session.execute(ProjectService.select_cls().where(ProjectModel.id.in_(proj_id_list)))).scalars().all()
    project_dict_list = [item.model_dump() for item in project_cls_list]
    for dict_item in project_dict_list:
      # 已花费金额
      dict_item['spent'] = result[dict_item['id']]
      # 剩余金额
      dict_item['balance'] = Decimal(dict_item['budget']) - Decimal(dict_item['spent'])

    return project_dict_list