strepsiades
158 lines
6.0 KiB
Python
158 lines
6.0 KiB
Python
"""
|
|
Preparation (里程碑) API Handler
|
|
|
|
职责：
|
|
- 更新项目里程碑数据
|
|
- 不支持创建和删除（里程碑随项目自动生成）
|
|
|
|
实现的接口：
|
|
PUT:
|
|
- /preparation - 更新里程碑数据
|
|
"""
|
|
|
|
from typing import List, Dict
|
|
from ...datasource.api.handler import BaseApiHandler
|
|
from ...datasource.task_result import TaskResult
|
|
from ...common.sync_node import SyncNode
|
|
from ...datasource.api.errors import ERROR_VALIDATION_FAILED
|
|
from .sync_node import PreparationSyncNode
|
|
from schemas.project_extensions.preparation import PreparationDetail, PostPreparation
|
|
|
|
|
|
class PreparationApiHandler(BaseApiHandler):
|
|
"""里程碑 API Handler"""
|
|
|
|
def __init__(self):
|
|
super().__init__(node_class=PreparationSyncNode, schema=PreparationDetail)
|
|
self.update_schemas = [PostPreparation]
|
|
|
|
async def load(self) -> List[SyncNode]:
|
|
"""从 API 加载里程碑数据（需要项目上下文）"""
|
|
from ..project.api_handler import ProjectApiHandler
|
|
|
|
# 获取要加载的项目列表
|
|
projects = self._collection.filter(node_type="project")
|
|
if not projects:
|
|
print(" No projects found, skipping preparation load")
|
|
return []
|
|
|
|
nodes = []
|
|
for project in projects:
|
|
project_id = project.data_id
|
|
|
|
try:
|
|
# 从 project context 或缓存获取 all_info
|
|
all_info = project.context.get("all_info")
|
|
if not all_info:
|
|
all_info = await ProjectApiHandler.get_all_info(
|
|
self.api_client,
|
|
project_id
|
|
)
|
|
|
|
# 提取 preparation_list
|
|
preparation_list = all_info.get("preparation_list", [])
|
|
print(f" Project {project_id}: {len(preparation_list)} preparations")
|
|
|
|
for prep_data in preparation_list:
|
|
# 验证数据
|
|
try:
|
|
validated = self._schema.model_validate(prep_data)
|
|
node = self._create_node(validated.model_dump())
|
|
nodes.append(node)
|
|
except Exception as e:
|
|
print(f" Skipping invalid preparation: {e}")
|
|
continue
|
|
|
|
except Exception as e:
|
|
print(f" Failed to load preparations for project {project_id}: {e}")
|
|
continue
|
|
|
|
return nodes
|
|
|
|
async def create_all(self, nodes: List[SyncNode]) -> List[TaskResult]:
|
|
"""不支持创建（里程碑随项目自动生成）"""
|
|
return [
|
|
TaskResult.failed(node_id=node.node_id, error="Preparation creation not supported (auto-generated with project)")
|
|
for node in nodes
|
|
]
|
|
|
|
async def update_all(self, nodes: List[SyncNode]) -> List[TaskResult]:
|
|
"""批量更新里程碑数据"""
|
|
from pydantic import ValidationError
|
|
|
|
results = []
|
|
|
|
for node in nodes:
|
|
node_data = node.get_data()
|
|
if node_data is None:
|
|
error_msg = node.error or "Node data is missing or invalid"
|
|
results.append(TaskResult.failed(node_id=node.node_id, error=f"Cannot update: {error_msg}"))
|
|
continue
|
|
|
|
origin_data = node.get_origin_data() or {}
|
|
force_steps_update = self._steps_changed(node)
|
|
update_data = self.build_update_request_data(
|
|
node_data,
|
|
origin_data,
|
|
PostPreparation,
|
|
force_full_schema=force_steps_update,
|
|
)
|
|
if not update_data:
|
|
reason = f"No physical diff found for preparation {node.data_id} despite strategy update request (Normalization Discrepancy)."
|
|
print(f" [WARNING] [preparation] {reason}")
|
|
results.append(TaskResult.skipped(node_id=node.node_id, reason=reason))
|
|
continue
|
|
|
|
# 验证并转换为 Pydantic model
|
|
try:
|
|
update_model = PostPreparation.model_validate(update_data)
|
|
except Exception as e:
|
|
results.append(TaskResult.failed(node_id=node.node_id, error=f"{ERROR_VALIDATION_FAILED}: {e}"))
|
|
continue
|
|
|
|
push_id = self._generate_push_id()
|
|
biz_id = node.data_id or update_model.id
|
|
if not biz_id:
|
|
results.append(TaskResult.failed(node_id=node.node_id, error="Missing biz_id for update"))
|
|
continue
|
|
|
|
try:
|
|
await api_update_preparation(
|
|
self.api_client,
|
|
update_model,
|
|
push_id,
|
|
biz_id,
|
|
steps=self._get_request_steps(node),
|
|
)
|
|
self._mark_approval_snapshot_synced(node)
|
|
results.append(TaskResult.success(node_id=node.node_id))
|
|
except Exception as e:
|
|
results.append(TaskResult.failed(node_id=node.node_id, error=str(e)))
|
|
|
|
return results
|
|
|
|
async def delete_all(self, nodes: List[SyncNode]) -> List[TaskResult]:
|
|
"""不支持删除"""
|
|
return [
|
|
TaskResult.failed(node_id=node.node_id, error="Preparation deletion not supported")
|
|
for node in nodes
|
|
]
|
|
|
|
# ========== 静态 API 函数 ==========
|
|
|
|
async def api_update_preparation(
|
|
client,
|
|
data: PostPreparation,
|
|
push_id: str,
|
|
biz_id: str,
|
|
*,
|
|
steps: List[Dict[str, Any]] | None = None,
|
|
) -> None:
|
|
"""
|
|
PUT /preparation_nodes - 更新里程碑数据
|
|
"""
|
|
request_data = {"push_id": push_id, "biz_id": biz_id, "data": data.model_dump(exclude_unset=True), "steps": steps or []}
|
|
response = await client.request(method="PUT", endpoint="/preparation_nodes", data=request_data)
|
|
if response.get("code") != 200:
|
|
raise Exception(f"API error: {response.get('message', 'Unknown error')}")
|