tapdata-sdk 0.1.0__py3-none-any.whl → 0.2.0__py3-none-any.whl

tapdata_sdk/client.py

@@ -14,7 +14,7 @@ from .exceptions import (
     TapdataError,
     TapdataTimeoutError,
 )
-from .models import Connection, Task, TaskDetail, TaskLog
+from .models import Connection, Task, TaskDetail, TaskLog, TaskRelation
 from .utils import rc4_encrypt, gen_sign, build_filter
 from .enums import ConnectionType, DatabaseType, Status, LogLevel
 
@@ -363,6 +363,28 @@ class TaskClient:
         """
         resp = self.client._request("GET", f"/api/Task/{task_id}")
         return TaskDetail.from_dict(resp["data"])
+
+    def get_table_relation(self, task_id: str) -> TaskRelation:
+        """
+        Get Task Table Name Relation
+
+        Args:
+            task_id: Unique identifier for the task
+
+        Returns:
+            TaskRelation
+        """
+        task_detail = self.get(task_id)
+
+        relation = TaskRelation.from_dict(task_detail.to_dict())
+
+        if relation.source_connection_id:
+            relation.source_conn = self.client.connections.get(relation.source_connection_id)
+
+        if relation.target_connection_id:
+            relation.target_conn = self.client.connections.get(relation.target_connection_id)
+
+        return relation
     
     def list_running(self) -> List[Task]:
         """Get all running tasks"""

tapdata_sdk/models.py

@@ -1,6 +1,6 @@
 """Data model definitions"""
 from dataclasses import dataclass
-from typing import Optional, List
+from typing import Optional, List, Dict
 
 
 @dataclass
@@ -124,6 +124,55 @@ class TaskDetail:
             "nodes": self.nodes
         }
 
+@dataclass
+class TaskRelation:
+    """
+    Task connection relationship mapping model
+    """
+    source_connection_id: Optional[str] = None
+    target_connection_id: Optional[str] = None
+    table_name_relation: Optional[Dict[str, str]] = None
+    source_conn: Optional[Connection] = None
+    target_conn: Optional[Connection] = None
+
+    @classmethod
+    def from_dict(cls, data: dict) -> "TaskRelation":
+        """
+        Create TaskRelation from task detail dictionary
+
+        Args:
+            data: Raw task data from API (containing 'dag' and 'nodes')
+        """
+        nodes = data.get("nodes", [])
+
+        if len(nodes) < 2:
+            return cls()
+
+        source_node = nodes[0]
+        target_node = nodes[-1]
+
+        relations = {}
+        for obj in target_node.get("syncObjects", []):
+            if "tableNameRelation" in obj:
+                relations.update(obj["tableNameRelation"])
+
+        return cls(
+            source_connection_id=source_node.get("connectionId"),
+            target_connection_id=target_node.get("connectionId"),
+            table_name_relation=relations
+        )
+
+    def to_dict(self) -> dict:
+        """
+        Convert to dictionary including nested connection details
+        """
+        return {
+            "source_connection_id": self.source_connection_id,
+            "target_connection_id": self.target_connection_id,
+            "table_name_relation": self.table_name_relation,
+            "source": self.source_conn.to_dict() if self.source_conn else None,
+            "target": self.target_conn.to_dict() if self.target_conn else None
+        }
 
 @dataclass
 class TaskLog:

{tapdata_sdk-0.1.0.dist-info → tapdata_sdk-0.2.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tapdata_sdk
-Version: 0.1.0
+Version: 0.2.0
 Summary: A Python client library for interacting with Tapdata API
 Author-email: doubled <309294891@qq.com>
 License: MIT
@@ -52,14 +52,14 @@ A Python client library for interacting with Tapdata API.
 ## Installation
 
 ```bash
-pip install tapdata_sdk
+pip install tapdata-sdk
 ```
 
 Or install from source:
 
 ```bash
 git clone https://github.com/lddlww/tapdata_sdk.git
-cd tapdata-sdk
+cd tapdata_sdk
 pip install -e .
 ```
 
@@ -85,6 +85,10 @@ for conn in connections:
 tasks = client.tasks.list()
 for task in tasks:
     print(f"{task.name}: {task.status}")
+
+# Get Table Relation
+table_relation = client.tasks.get_table_relation(task_id='xxx')
+print(f"{table_relation.table_name_relation}")
 ```
 
 ### Connection Management
@@ -318,7 +322,7 @@ isort tapdata_sdk/
 
 ## Changelog
 
-### v0.2.0 (2024-01-29)
+### v0.2.0 (2026-01-29)
 - ✨ Refactored code architecture with modular design
 - 📦 Added data model classes (Connection, Task, TaskLog)
 - 🎯 Improved enum types using Python Enum

tapdata_sdk-0.2.0.dist-info/RECORD

@@ -0,0 +1,11 @@
+tapdata_sdk/__init__.py,sha256=DCbgmv3TOkTHR69LFEJ3Y9Oxm9WfUBQUhyqUMyySSVk,1332
+tapdata_sdk/client.py,sha256=SJ6pTHJUmpkBHT1YnijHUzkzCs0miQsoAocjIWBmkL4,14543
+tapdata_sdk/enums.py,sha256=ywCPyswMdbISvuII9Lv7QSmwHUbfqsQB3mK_UPV7C9A,1160
+tapdata_sdk/exceptions.py,sha256=Edm1P6Aghb7wJhUOubYU9JuxcQrfI_WGn3wnX4ukH7Y,627
+tapdata_sdk/models.py,sha256=3WMU3KPYaZorM8f5BE_IG3bQiTjYeFhE0oc8ZsgAr7A,6217
+tapdata_sdk/utils.py,sha256=ojPuW9YwW7dkb0nLJ46_IgMbOzor7DtppYDBuN4nCiY,2682
+tapdata_sdk-0.2.0.dist-info/licenses/LICENSE,sha256=g3faROod99RM6X9tAOfJ_7LsPxI1sOISYpIy2w9VexQ,1064
+tapdata_sdk-0.2.0.dist-info/METADATA,sha256=VS9wa67Rn4anYjVabz5L4q9wgHZnQ_F3Y13k6tnYxoU,8066
+tapdata_sdk-0.2.0.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+tapdata_sdk-0.2.0.dist-info/top_level.txt,sha256=YNIwm3uNhULMB32qsLcj-zAoL5yYMKWJbXnstNsIYNk,12
+tapdata_sdk-0.2.0.dist-info/RECORD,,

tapdata_sdk-0.1.0.dist-info/RECORD

@@ -1,11 +0,0 @@
-tapdata_sdk/__init__.py,sha256=DCbgmv3TOkTHR69LFEJ3Y9Oxm9WfUBQUhyqUMyySSVk,1332
-tapdata_sdk/client.py,sha256=yx6W2iVnE_3b8ybguRevr3Oid2dUs73XPLZyu-KVyjw,13888
-tapdata_sdk/enums.py,sha256=ywCPyswMdbISvuII9Lv7QSmwHUbfqsQB3mK_UPV7C9A,1160
-tapdata_sdk/exceptions.py,sha256=Edm1P6Aghb7wJhUOubYU9JuxcQrfI_WGn3wnX4ukH7Y,627
-tapdata_sdk/models.py,sha256=elJxe-6IzSCOpSLCQroUGec4dBgjyf2PU7LHo9O_yQ8,4601
-tapdata_sdk/utils.py,sha256=ojPuW9YwW7dkb0nLJ46_IgMbOzor7DtppYDBuN4nCiY,2682
-tapdata_sdk-0.1.0.dist-info/licenses/LICENSE,sha256=g3faROod99RM6X9tAOfJ_7LsPxI1sOISYpIy2w9VexQ,1064
-tapdata_sdk-0.1.0.dist-info/METADATA,sha256=d477QW0uSb5KbHfEO99E_02ZaVTPak9M0dp-JDhDR6k,7933
-tapdata_sdk-0.1.0.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-tapdata_sdk-0.1.0.dist-info/top_level.txt,sha256=YNIwm3uNhULMB32qsLcj-zAoL5yYMKWJbXnstNsIYNk,12
-tapdata_sdk-0.1.0.dist-info/RECORD,,