优化sync_system代码质量

sync_state_machine/datasource/api/handler.py

@@ -20,6 +20,7 @@ from typing import ClassVar, Dict, List, Any, Optional, TYPE_CHECKING, TypeVar
 from pydantic import BaseModel, ValidationError
 
 from ...common.type_safety import get_pydantic_model_fields
+from ...logging import get_logger
 from ..handler import BaseNodeHandler
 from ..task_result import TaskResult
 from ..handler import ValidationResult  # Import ValidationResult
@@ -30,6 +31,7 @@ if TYPE_CHECKING:
     from .client import ApiClient
 
 T = TypeVar("T", bound=BaseModel)
+logger = get_logger(__name__)
 
 
 class BaseApiHandler(BaseNodeHandler[T]):
@@ -67,6 +69,16 @@ class BaseApiHandler(BaseNodeHandler[T]):
         """设置 Collection（由 DataSource 调用）"""
         self._collection = collection
 
+    def ensure_api_client(self) -> 'ApiClient':
+        if self.api_client is None:
+            raise RuntimeError("API client not set. Call set_api_client() before invoking API handler methods.")
+        return self.api_client
+
+    def ensure_collection(self) -> 'DataCollection':
+        if self._collection is None:
+            raise RuntimeError("Collection not set. Call set_collection() before invoking handler methods.")
+        return self._collection
+
     @property
     def node_type(self) -> str:
         """返回节点类型"""
@@ -109,8 +121,7 @@ class BaseApiHandler(BaseNodeHandler[T]):
     def _create_basic_node(self, data: Dict[str, Any], depend_ids: Optional[List[str]] = None) -> "SyncNode":
         """创建或复用基础节点（默认使用 data['id'] 或 data['_id']）"""
         data_id = str(data.get("id") or data.get("_id") or "")
-        if self._collection is None:
-            raise RuntimeError("Collection not set. Call set_collection() before creating nodes.")
+        self.ensure_collection()
 
         if data_id:
             existing_node = self._collection.get_by_data_id(self.node_type, data_id)
@@ -329,10 +340,10 @@ class BaseApiHandler(BaseNodeHandler[T]):
 
         if diff_fields:
             node_info = f"[{node_id}]" if node_id else ""
-            print(f"  [DIFF] {self.node_type}{node_info} schema={schema.__name__}")
+            logger.info("  [DIFF] %s%s schema=%s", self.node_type, node_info, schema.__name__)
             for k, v in diff_fields.items():
                 orig_v = origin_dict.get(k)
-                print(f"    • {k}: {orig_v!r} → {v!r}")
+                logger.info("    • %s: %r → %r", k, orig_v, v)
 
         return diff_fields if diff_fields else None