strepsiades
219 lines
8.5 KiB
Python
219 lines
8.5 KiB
Python
"""
|
||
LAR (移民征地) API Handler
|
||
|
||
职责：
|
||
- 更新移民征地数据（依赖 Project）
|
||
- 支持主数据、搬迁信息、投资信息三种更新
|
||
- 不支持创建和删除（数据随项目自动生成）
|
||
|
||
实现的接口：
|
||
PUT:
|
||
- /lar/main - 更新主数据
|
||
- /lar/resettlement - 更新搬迁信息
|
||
- /lar/investment - 更新投资信息
|
||
"""
|
||
|
||
from typing import List, Dict, Any
|
||
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 LarSyncNode
|
||
from schemas.project_extensions.lar import (
|
||
LarDetailSchema,
|
||
LarMainInfoUpdateSchema,
|
||
LarResettlementSchema,
|
||
LarInvestmentSchema
|
||
)
|
||
|
||
|
||
class LarApiHandler(BaseApiHandler):
|
||
"""移民征地 API Handler"""
|
||
|
||
def __init__(self):
|
||
super().__init__()
|
||
self._node_type = "lar"
|
||
self._node_class = LarSyncNode
|
||
self._schema = LarDetailSchema
|
||
|
||
async def load(self) -> List[SyncNode]:
|
||
"""
|
||
加载移民征地数据（从 project all_info 中提取）
|
||
|
||
注意：LAR 是可选的，只有特定项目类型（水电、抽蓄）才有
|
||
"""
|
||
from ..project.api_handler import ProjectApiHandler
|
||
|
||
# 获取已加载的项目
|
||
projects = self._collection.filter(node_type="project")
|
||
if not projects:
|
||
print("⚠️ No projects found, skipping LAR 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
|
||
)
|
||
|
||
# 提取 lar（注意：lar 是单个对象，不是列表）
|
||
lar_data = all_info.get("lar")
|
||
if lar_data:
|
||
try:
|
||
validated = self._schema.model_validate(lar_data)
|
||
node = self._create_node(validated.model_dump())
|
||
node.context["project_id"] = project_id
|
||
nodes.append(node)
|
||
print(f" Project {project_id}: 1 LAR")
|
||
except Exception as e:
|
||
print(f"⚠️ Skipping invalid LAR for project {project_id}: {e}")
|
||
continue
|
||
|
||
except Exception as e:
|
||
print(f"❌ Failed to load LAR for project {project_id}: {e}")
|
||
continue
|
||
|
||
return nodes
|
||
|
||
async def create_all(self, nodes: List[SyncNode]) -> List[TaskResult]:
|
||
"""不支持创建（LAR随项目自动生成）"""
|
||
return [
|
||
TaskResult.failed(node_id=node.node_id, error="LAR 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 = []
|
||
|
||
# 定义三种更新类型
|
||
update_configs = [
|
||
(LarMainInfoUpdateSchema, api_update_lar_main),
|
||
(LarResettlementSchema, api_update_lar_resettlement),
|
||
(LarInvestmentSchema, api_update_lar_investment),
|
||
]
|
||
|
||
for node in nodes:
|
||
node_data = node.get_data()
|
||
if not node_data:
|
||
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 {}
|
||
node_updated = False
|
||
|
||
for schema, api_func in update_configs:
|
||
# 获取过滤后的更新数据
|
||
update_data = self._filter_update_data(node_data, origin_data, schema)
|
||
if update_data:
|
||
# 验证并转换为 Pydantic model
|
||
try:
|
||
update_model = schema.model_validate(update_data)
|
||
except ValidationError as e:
|
||
results.append(TaskResult.failed(node_id=node.node_id, error=f"{ERROR_VALIDATION_FAILED}: {e}"))
|
||
break
|
||
|
||
push_id = self._generate_push_id()
|
||
update_model_data = update_model.model_dump()
|
||
biz_id = node.data_id or update_model_data.get("id")
|
||
if not biz_id:
|
||
results.append(TaskResult.failed(node_id=node.node_id, error="Missing biz_id for update"))
|
||
break
|
||
|
||
try:
|
||
await api_func(self.api_client, update_model, push_id, biz_id)
|
||
node_updated = True
|
||
except Exception as e:
|
||
results.append(TaskResult.failed(node_id=node.node_id, error=str(e)))
|
||
break
|
||
|
||
if node_updated:
|
||
results.append(TaskResult.success(node_id=node.node_id))
|
||
elif not any(r.node_id == node.node_id for r in results):
|
||
# 如果没有任何更新且没有报错，视为跳过（归一化后无差异）
|
||
reason = f"No physical diff found across all 3 LAR sub-schemas for {node.data_id} despite strategy update request (Normalization Discrepancy)."
|
||
print(f" [WARNING] [lar] {reason}")
|
||
results.append(TaskResult.skipped(node_id=node.node_id, reason=reason))
|
||
|
||
return results
|
||
|
||
async def delete_all(self, nodes: List[SyncNode]) -> List[TaskResult]:
|
||
"""不支持删除（LAR随项目存在）"""
|
||
return [
|
||
TaskResult.failed(node_id=node.node_id, error="LAR deletion not supported")
|
||
for node in nodes
|
||
]
|
||
|
||
async def poll_tasks(self, task_ids: List[str]) -> Dict[str, TaskResult]:
|
||
"""轮询异步任务状态"""
|
||
return await super().poll_tasks(task_ids)
|
||
|
||
def _create_node(self, data: Dict[str, Any]) -> SyncNode:
|
||
node = self._create_basic_node(data, depend_ids=[])
|
||
|
||
if data.get("project_id"):
|
||
node.context["project_id"] = data["project_id"]
|
||
node.context["lar_id"] = node.data_id
|
||
|
||
return node
|
||
|
||
|
||
# ========== 静态 API 函数 ==========
|
||
|
||
async def api_update_lar_main(client, data: LarMainInfoUpdateSchema, push_id: str, biz_id: str) -> None:
|
||
"""
|
||
PUT /lar/main - 更新主数据
|
||
|
||
Args:
|
||
client: API 客户端
|
||
data: 更新数据（Pydantic model）
|
||
push_id: 推送ID
|
||
biz_id: 业务ID
|
||
"""
|
||
request_data = {"push_id": push_id, "biz_id": biz_id, "data": data.model_dump(exclude_unset=True)}
|
||
response = await client.request(method="PUT", endpoint="/lar/main", data=request_data)
|
||
if response.get("code") != 200:
|
||
raise Exception(f"API error: {response.get('message', 'Unknown error')}")
|
||
|
||
|
||
async def api_update_lar_resettlement(client, data: LarResettlementSchema, push_id: str, biz_id: str) -> None:
|
||
"""
|
||
PUT /lar/resettlement - 更新搬迁信息
|
||
|
||
Args:
|
||
client: API 客户端
|
||
data: 更新数据（Pydantic model）
|
||
push_id: 推送ID
|
||
biz_id: 业务ID
|
||
"""
|
||
request_data = {"push_id": push_id, "biz_id": biz_id, "data": data.model_dump(exclude_unset=True)}
|
||
response = await client.request(method="PUT", endpoint="/lar/resettlement", data=request_data)
|
||
if response.get("code") != 200:
|
||
raise Exception(f"API error: {response.get('message', 'Unknown error')}")
|
||
|
||
|
||
async def api_update_lar_investment(client, data: LarInvestmentSchema, push_id: str, biz_id: str) -> None:
|
||
"""
|
||
PUT /lar/investment - 更新投资信息
|
||
|
||
Args:
|
||
client: API 客户端
|
||
data: 更新数据（Pydantic model）
|
||
push_id: 推送ID
|
||
biz_id: 业务ID
|
||
"""
|
||
request_data = {"push_id": push_id, "biz_id": biz_id, "data": data.model_dump(exclude_unset=True)}
|
||
response = await client.request(method="PUT", endpoint="/lar/investment", data=request_data)
|
||
if response.get("code") != 200:
|
||
raise Exception(f"API error: {response.get('message', 'Unknown error')}")
|