first commit
This commit is contained in:
strepsiades
@@ -0,0 +1,7 @@
|
||||
"""Contract domain"""
|
||||
from .sync_node import ContractSyncNode
|
||||
from .sync_strategy import ContractSyncStrategy
|
||||
from .jsonl_handler import ContractJsonlHandler
|
||||
|
||||
__all__ = ["ContractSyncNode", "ContractSyncStrategy", "ContractJsonlHandler"]
|
||||
|
||||
@@ -0,0 +1,329 @@
|
||||
"""
|
||||
Contract API Handler
|
||||
|
||||
职责：
|
||||
- 加载合同列表（依赖 Project）
|
||||
- 合同创建（POST /contract/create，异步）
|
||||
- 合同更新（PUT /contract/update，同步）
|
||||
- 合同删除（DELETE /contract/delete，同步）
|
||||
"""
|
||||
|
||||
from typing import List, Dict, Any, Optional
|
||||
|
||||
from ...datasource.api.handler import BaseApiHandler
|
||||
from ...datasource.task_result import TaskResult
|
||||
from ...common.sync_node import SyncNode
|
||||
from ...datasource.api.client import ApiClient
|
||||
from .sync_node import ContractSyncNode
|
||||
from schemas.contract.contract import ContractResponse, ContractCreate, ContractUpdate
|
||||
|
||||
|
||||
# ========== 静态 API 函数 ==========
|
||||
|
||||
async def api_get_contract_list(client: ApiClient, project_id: str) -> List[ContractResponse]:
|
||||
"""GET /contract/list - 获取项目合同列表"""
|
||||
response = await client.request(
|
||||
method="GET",
|
||||
endpoint="/contract/list",
|
||||
params={"project_id": project_id}
|
||||
)
|
||||
if response.get("code") != 200:
|
||||
raise Exception(f"Failed to get contract list: {response.get('message', 'Unknown error')}")
|
||||
|
||||
# V2 API 返回格式：{code: 200, data: [...]}
|
||||
items = response.get("data", [])
|
||||
|
||||
return [ContractResponse.model_validate(item) for item in items]
|
||||
|
||||
|
||||
async def api_get_contract(client: ApiClient, contract_id: str) -> ContractResponse:
|
||||
"""GET /contract - 获取合同详情"""
|
||||
response = await client.request(
|
||||
method="GET",
|
||||
endpoint="/contract",
|
||||
params={"contract_id": contract_id}
|
||||
)
|
||||
if response.get("code") != 200:
|
||||
raise Exception(f"Failed to get contract: {response.get('message', 'Unknown error')}")
|
||||
return ContractResponse.model_validate(response["data"])
|
||||
|
||||
|
||||
async def api_create_contract(client: ApiClient, data: Dict[str, Any], push_id: str) -> Dict[str, Any]:
|
||||
"""POST /contract/create - 创建合同（异步）"""
|
||||
validated_data = ContractCreate.model_validate(data)
|
||||
request_data = {"push_id": push_id, "data": validated_data.model_dump(exclude_none=True)}
|
||||
response = await client.request("POST", "/contract/create", data=request_data)
|
||||
if response.get("code") != 200:
|
||||
raise Exception(f"Failed to create contract: {response.get('message', 'Unknown error')}")
|
||||
return response
|
||||
|
||||
|
||||
async def api_update_contract(client: ApiClient, data_id: str, data: Dict[str, Any], push_id: str) -> Dict[str, Any]:
|
||||
"""PUT /contract/update - 更新合同（同步）"""
|
||||
validated_data = ContractUpdate.model_validate(data)
|
||||
request_data = {"push_id": push_id, "biz_id": data_id, "data": validated_data.model_dump(exclude_none=True)}
|
||||
response = await client.request("PUT", "/contract/update", data=request_data)
|
||||
if response.get("code") != 200:
|
||||
raise Exception(f"Failed to update contract: {response.get('message', 'Unknown error')}")
|
||||
return response
|
||||
|
||||
|
||||
async def api_delete_contract(client: ApiClient, data_id: str, push_id: str) -> Dict[str, Any]:
|
||||
"""DELETE /contract/delete - 删除合同（同步）"""
|
||||
request_data = {"push_id": push_id, "biz_id": data_id}
|
||||
response = await client.request("DELETE", "/contract/delete", data=request_data)
|
||||
if response.get("code") != 200:
|
||||
raise Exception(f"Failed to delete contract: {response.get('message', 'Unknown error')}")
|
||||
return response
|
||||
|
||||
|
||||
# ========== Handler ==========
|
||||
|
||||
class ContractApiHandler(BaseApiHandler):
|
||||
"""合同 API Handler"""
|
||||
|
||||
def __init__(self):
|
||||
super().__init__()
|
||||
self._node_type = "contract"
|
||||
self._node_class = ContractSyncNode
|
||||
self._schema = ContractResponse
|
||||
self.update_schemas = [ContractCreate, ContractUpdate]
|
||||
|
||||
# ========== Handler 接口实现 ==========
|
||||
|
||||
async def load(self) -> List[SyncNode]:
|
||||
"""
|
||||
加载合同列表
|
||||
|
||||
依赖 Project：从 collection 获取项目列表，然后加载这些项目的合同
|
||||
|
||||
Returns:
|
||||
合同数据列表
|
||||
"""
|
||||
try:
|
||||
# 从 collection 获取项目列表
|
||||
projects = self._collection.filter(node_type="project")
|
||||
project_ids = [p.data_id for p in projects if p.data_id]
|
||||
|
||||
if not project_ids:
|
||||
print("⚠️ No projects found, skipping contract load")
|
||||
return []
|
||||
|
||||
# 为每个项目加载合同
|
||||
all_nodes = []
|
||||
for project_id in project_ids:
|
||||
try:
|
||||
contracts = await api_get_contract_list(self.api_client, project_id)
|
||||
|
||||
# 已验证的 Pydantic model，转换为 dict 并创建 node
|
||||
for contract in contracts:
|
||||
contract_data = contract.model_dump()
|
||||
all_nodes.append(self._create_node(contract_data))
|
||||
|
||||
except Exception as e:
|
||||
print(f"❌ Failed to load contracts for project {project_id}: {e}")
|
||||
continue
|
||||
|
||||
return all_nodes
|
||||
|
||||
except Exception as e:
|
||||
print(f"❌ Failed to load contracts: {e}")
|
||||
return []
|
||||
|
||||
async def create_all(self, nodes: List[SyncNode]) -> List[TaskResult]:
|
||||
"""批量创建合同"""
|
||||
results = []
|
||||
for node in nodes:
|
||||
result = await self._create_single(node)
|
||||
results.append(result)
|
||||
return results
|
||||
|
||||
async def update_all(self, nodes: List[SyncNode]) -> List[TaskResult]:
|
||||
"""批量更新合同"""
|
||||
results = []
|
||||
for node in nodes:
|
||||
result = await self._update_single(node)
|
||||
results.append(result)
|
||||
return results
|
||||
|
||||
async def delete_all(self, nodes: List[SyncNode]) -> List[TaskResult]:
|
||||
"""批量删除合同"""
|
||||
results = []
|
||||
for node in nodes:
|
||||
result = await self._delete_single(node)
|
||||
results.append(result)
|
||||
return results
|
||||
|
||||
async def poll_tasks(self, task_ids: List[str]) -> Dict[str, TaskResult]:
|
||||
"""轮询异步任务状态"""
|
||||
return await super().poll_tasks(task_ids)
|
||||
|
||||
async def _create_single(self, node: SyncNode) -> TaskResult:
|
||||
"""
|
||||
创建合同（异步）
|
||||
|
||||
Args:
|
||||
node: 同步节点
|
||||
|
||||
Returns:
|
||||
TaskResult (IN_PROGRESS with push_id)
|
||||
"""
|
||||
push_id: str | None = None
|
||||
try:
|
||||
# 获取数据
|
||||
data = node.get_data()
|
||||
if not data:
|
||||
return TaskResult.failed(node_id=node.node_id, error="Node data is None or validation failed")
|
||||
|
||||
# 从 context 获取 project_id
|
||||
project_id = node.context.get("project_id")
|
||||
if not project_id:
|
||||
project_id = data.get("project_id")
|
||||
|
||||
if not project_id:
|
||||
return TaskResult.failed(
|
||||
node_id=node.node_id,
|
||||
error="project_id not found in context or data"
|
||||
)
|
||||
|
||||
# 构建请求数据（确保包含 project_id）
|
||||
request_data = {
|
||||
"project_id": project_id,
|
||||
**data
|
||||
}
|
||||
|
||||
# 调用 API（直接调用顶层函数）
|
||||
push_id = self._generate_push_id()
|
||||
response = await api_create_contract(self.api_client, request_data, push_id)
|
||||
if self.poll_mode == "sync":
|
||||
created_id = self._extract_created_id_from_response(response)
|
||||
if not created_id:
|
||||
return TaskResult.failed(
|
||||
node_id=node.node_id,
|
||||
error="Missing created_id in create response"
|
||||
)
|
||||
return TaskResult.success(node_id=node.node_id, data_id=created_id)
|
||||
else:
|
||||
return TaskResult.in_progress(node_id=node.node_id, task_id=push_id)
|
||||
|
||||
except Exception as e:
|
||||
return TaskResult.failed(
|
||||
node_id=node.node_id,
|
||||
error=f"Create failed: {str(e)}",
|
||||
push_id=push_id,
|
||||
)
|
||||
|
||||
async def _update_single(self, node: SyncNode) -> TaskResult:
|
||||
"""
|
||||
更新合同（同步）
|
||||
|
||||
自动选择需要更新的字段（基于 update_schema）
|
||||
|
||||
Args:
|
||||
node: 同步节点
|
||||
|
||||
Returns:
|
||||
TaskResult (SUCCESS or FAILED)
|
||||
"""
|
||||
try:
|
||||
data = node.get_data()
|
||||
if not data:
|
||||
return TaskResult.failed(node_id=node.node_id, error="Missing data for update")
|
||||
|
||||
# 过滤出需要更新的字段（排除空值、未变化的字段）
|
||||
origin_data = node.get_origin_data() or {}
|
||||
update_data = self._filter_update_data(
|
||||
new_data=data,
|
||||
original_data=origin_data,
|
||||
schema=ContractUpdate,
|
||||
)
|
||||
|
||||
# 如果没有需要更新的字段，降级 action 并返回跳过
|
||||
if not update_data:
|
||||
return TaskResult.skipped(node_id=node.node_id, reason="No updatable fields changed")
|
||||
|
||||
# Schema 验证
|
||||
error = self._validate_schema(update_data, ContractUpdate)
|
||||
if error:
|
||||
return TaskResult.failed(
|
||||
node_id=node.node_id,
|
||||
error=f"Schema validation failed: {error}"
|
||||
)
|
||||
|
||||
# 生成 push_id 和 biz_id
|
||||
push_id = self._generate_push_id()
|
||||
biz_id = node.data_id or update_data.get("id")
|
||||
if not biz_id:
|
||||
return TaskResult.failed(node_id=node.node_id, error="Missing biz_id for update")
|
||||
|
||||
# 调用 API
|
||||
response = await api_update_contract(self.api_client, biz_id, update_data, push_id)
|
||||
|
||||
# 检查响应状态（更新接口是同步的）
|
||||
if response.get("code") == 200:
|
||||
return TaskResult.success(node_id=node.node_id)
|
||||
else:
|
||||
return TaskResult.failed(
|
||||
node_id=node.node_id,
|
||||
error=f"API returned error: {response.get('message', 'Unknown error')}"
|
||||
)
|
||||
|
||||
except Exception as e:
|
||||
return TaskResult.failed(
|
||||
node_id=node.node_id,
|
||||
error=f"Update failed: {str(e)}"
|
||||
)
|
||||
|
||||
async def _delete_single(self, node: SyncNode) -> TaskResult:
|
||||
"""
|
||||
删除合同（同步）
|
||||
|
||||
Args:
|
||||
node: 同步节点
|
||||
|
||||
Returns:
|
||||
TaskResult (SUCCESS or FAILED)
|
||||
"""
|
||||
if not self.api_client:
|
||||
return TaskResult.failed(
|
||||
node_id=node.node_id,
|
||||
error="api_client not initialized"
|
||||
)
|
||||
|
||||
try:
|
||||
# 生成 push_id 和 biz_id
|
||||
push_id = self._generate_push_id()
|
||||
biz_id = node.data_id
|
||||
if not biz_id:
|
||||
return TaskResult.failed(node_id=node.node_id, error="Missing biz_id for delete")
|
||||
|
||||
# 调用 API
|
||||
response = await api_delete_contract(self.api_client, biz_id, push_id)
|
||||
|
||||
# 检查响应状态
|
||||
if response.get("code") == 200:
|
||||
return TaskResult.success(node_id=node.node_id)
|
||||
else:
|
||||
return TaskResult.failed(
|
||||
node_id=node.node_id,
|
||||
error=f"API returned error: {response.get('message', 'Unknown error')}"
|
||||
)
|
||||
|
||||
except Exception as e:
|
||||
return TaskResult.failed(
|
||||
node_id=node.node_id,
|
||||
error=f"Delete failed: {str(e)}"
|
||||
)
|
||||
|
||||
# ========== 辅助方法 ==========
|
||||
|
||||
def _create_node(self, data: Dict[str, Any]) -> SyncNode:
|
||||
"""创建合同同步节点（设置 project_id 到 context）"""
|
||||
node = self._create_basic_node(data, depend_ids=[])
|
||||
|
||||
# 设置 project_id 到 context（供子节点使用）
|
||||
if data.get("project_id"):
|
||||
node.context["project_id"] = data["project_id"]
|
||||
|
||||
return node
|
||||
|
||||
@@ -0,0 +1,14 @@
|
||||
"""
|
||||
Contract JSONL Handler
|
||||
"""
|
||||
from ...datasource import BaseJsonlHandler, JsonlDataSource
|
||||
from .sync_node import ContractSyncNode
|
||||
from schemas.contract.contract import ContractResponse
|
||||
|
||||
|
||||
class ContractJsonlHandler(BaseJsonlHandler):
|
||||
"""合同 JSONL Handler"""
|
||||
def __init__(self, datasource: JsonlDataSource):
|
||||
super().__init__(datasource, "contract", ContractSyncNode, ContractResponse)
|
||||
|
||||
|
||||
@@ -0,0 +1,17 @@
|
||||
"""Contract domain registration"""
|
||||
from ...common.registry import DomainRegistry
|
||||
from schemas.contract.contract import ContractResponse
|
||||
from .sync_node import ContractSyncNode
|
||||
from .sync_strategy import ContractSyncStrategy
|
||||
from .jsonl_handler import ContractJsonlHandler
|
||||
from .api_handler import ContractApiHandler
|
||||
|
||||
# 注册 contract domain
|
||||
DomainRegistry.register(
|
||||
node_type="contract",
|
||||
schema=ContractResponse,
|
||||
node_class=ContractSyncNode,
|
||||
strategy_class=ContractSyncStrategy,
|
||||
jsonl_handler_class=ContractJsonlHandler,
|
||||
api_handler_class=ContractApiHandler,
|
||||
)
|
||||
@@ -0,0 +1,11 @@
|
||||
from ...common.sync_node import SyncNode
|
||||
from schemas.contract.contract import ContractResponse
|
||||
|
||||
|
||||
class ContractSyncNode(SyncNode[ContractResponse]):
|
||||
"""合同同步节点"""
|
||||
node_type = "contract"
|
||||
schema = ContractResponse
|
||||
|
||||
def __init__(self, **kwargs):
|
||||
super().__init__(**kwargs)
|
||||
@@ -0,0 +1,30 @@
|
||||
from ...sync_system.strategy import DefaultSyncStrategy
|
||||
from ...sync_system.config import StrategyConfig, OrphanAction, UpdateDirection
|
||||
from schemas.contract.contract import ContractResponse
|
||||
|
||||
|
||||
class ContractSyncStrategy(DefaultSyncStrategy[ContractResponse]):
|
||||
"""
|
||||
Contract 同步策略
|
||||
|
||||
业务规则：
|
||||
1. 使用 (project_id, code) 作为业务唯一键
|
||||
2. 依赖于 project 和 company
|
||||
3. 支持自动绑定
|
||||
4. 默认配置为日常推送模式（本地创建推送到远程，不拉取远程孤儿）
|
||||
|
||||
schema 已在类定义中设置，禁止在初始化时传入其他 schema。
|
||||
"""
|
||||
|
||||
# 类变量：schema 固定为 ContractResponse
|
||||
schema = ContractResponse
|
||||
|
||||
# 类变量：默认配置
|
||||
default_config = StrategyConfig(
|
||||
auto_bind=True,
|
||||
auto_bind_fields=["project_id","company_id", "code"],
|
||||
depend_fields={"project_id": "project", "company_id": "supplier"},
|
||||
local_orphan_action=OrphanAction.CREATE_REMOTE,
|
||||
remote_orphan_action=OrphanAction.NONE,
|
||||
update_direction=UpdateDirection.PUSH,
|
||||
)
|
||||
Reference in New Issue
Block a user