65 lines
2.9 KiB
Python
65 lines
2.9 KiB
Python
|
# -*- coding: utf-8 -*-
|
|||
|
|
"""业务层 · 任务状态机（抢单 / 投标 / 交付 / 中标 / 验收）。
|
|||
|
|
|
|||
|
|
业务层只处理业务逻辑；不感知 HTTP 之外的框架细节。依赖基础设施层
|
|||
|
|
Repository（经 Database 门面），不反向依赖接口层。
|
|||
|
|
"""
|
|||
|
|
from __future__ import annotations
|
|||
|
|
|
|||
|
|
from fastapi import HTTPException
|
|||
|
|
|
|||
|
|
from ..infrastructure.repositories import Database
|
|||
|
|
|
|||
|
|
|
|||
|
|
class TaskService:
|
|||
|
|
"""统一任务状态流转：grab/bid/deliver/win/review。"""
|
|||
|
|
|
|||
|
|
def __init__(self, db: Database):
|
|||
|
|
self.db = db
|
|||
|
|
|
|||
|
|
async def grab(self, task_id: str, actor: dict) -> dict:
|
|||
|
|
"""抢单：仅 published + grab 模式可抢，抢后置 in_progress。"""
|
|||
|
|
task = await self.db.tasks.get(task_id)
|
|||
|
|
if task is None or task["status"] != "published" or task["mode"] != "grab":
|
|||
|
|
raise HTTPException(status_code=400, detail="任务不可抢单")
|
|||
|
|
return await self.db.tasks.set_status(task_id, "in_progress")
|
|||
|
|
|
|||
|
|
async def bid(self, task_id: str, actor: dict, quote: int, plan: str) -> dict:
|
|||
|
|
"""投标：仅 published + bid 模式可投。"""
|
|||
|
|
task = await self.db.tasks.get(task_id)
|
|||
|
|
if task is None or task["status"] != "published" or task["mode"] != "bid":
|
|||
|
|
raise HTTPException(status_code=400, detail="任务不可投标")
|
|||
|
|
return await self.db.bids.create(
|
|||
|
|
task_id, actor["id"], actor.get("nickname") or actor["username"],
|
|||
|
|
quote, plan,
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
async def deliver(self, task_id: str, actor: dict) -> dict:
|
|||
|
|
"""交付：任务存在且处于进行中（in_progress）才可交付 → delivered。
|
|||
|
|
|
|||
|
|
补齐原实现「无任何状态预检」的缺陷，防随意交付。
|
|||
|
|
"""
|
|||
|
|
task = await self.db.tasks.get(task_id)
|
|||
|
|
if task is None:
|
|||
|
|
raise HTTPException(status_code=404, detail="Task not found")
|
|||
|
|
if task["status"] != "in_progress":
|
|||
|
|
raise HTTPException(status_code=400, detail="任务未在进行中，无法交付")
|
|||
|
|
return await self.db.tasks.set_status(task_id, "delivered")
|
|||
|
|
|
|||
|
|
async def enterprise_submit(self, task_id: str) -> dict:
|
|||
|
|
"""企业提交审核：pending 任务 → published。"""
|
|||
|
|
return await self.db.tasks.set_status(task_id, "published")
|
|||
|
|
|
|||
|
|
async def win_bid(self, task_id: str, bid_id: str) -> dict:
|
|||
|
|
"""企业评标中标：bid → win，task → in_progress（双状态联动）。"""
|
|||
|
|
bid = await self.db.bids.get(bid_id)
|
|||
|
|
if bid is None or bid["task_id"] != task_id:
|
|||
|
|
raise HTTPException(status_code=404, detail="竞标不存在")
|
|||
|
|
await self.db.bids.set_status(bid_id, "win")
|
|||
|
|
return await self.db.tasks.set_status(task_id, "in_progress")
|
|||
|
|
|
|||
|
|
async def review(self, task_id: str, accept: bool) -> dict:
|
|||
|
|
"""企业验收：accept → completed；reject → in_progress。"""
|
|||
|
|
target = "completed" if accept else "in_progress"
|
|||
|
|
return await self.db.tasks.set_status(task_id, target)
|