dts-dance 0.2.5__py3-none-any.whl → 0.2.8__py3-none-any.whl

{dts_dance-0.2.5.dist-info → dts_dance-0.2.8.dist-info}/METADATA

@@ -1,10 +1,10 @@
 Metadata-Version: 2.4
 Name: dts-dance
-Version: 0.2.5
+Version: 0.2.8
 Summary: dts dance lib
 Keywords: observation,tools
 Requires-Python: >=3.12
-Requires-Dist: boto3>=1.42.30
+Requires-Dist: boto3>=1.42.35
 Requires-Dist: loguru<0.8.0,>=0.7.3
 Requires-Dist: pyyaml<7.0.0,>=6.0.3
 Requires-Dist: requests-toolbelt==1.0.0

dts_dance-0.2.8.dist-info/RECORD

@@ -0,0 +1,16 @@
+dtsdance/__init__.py,sha256=Yl_jEZ5weYfcrklnDvwB4wSgCOvMBLRRgWx0gHs3qfM,49
+dtsdance/bytecloud.py,sha256=LZ-uw5NMmapyNpt89bMKhCgv4U2cGR-XjLzFFFEXGxM,5899
+dtsdance/ddutil.py,sha256=1ZUmW11nw8WW1NfAIRbjuKy8zeQVTFvoXl6QY3o4YJ0,1534
+dtsdance/dflow.py,sha256=v6tQv9O6-xjXhNuU14vys0lIxi4hKKv27nuBjLgRk6Y,7747
+dtsdance/dsyncer.py,sha256=PCatQIFXgi1l2qT60zNN-ALAqtv7_sg-D2oNMB_t8qw,9906
+dtsdance/feishu_base.py,sha256=dYYNnsCjGZYN2I6inR9FRQoUiDLB7j40rzydw50N38o,11825
+dtsdance/feishu_table.py,sha256=JleCjQJ-rcbC15v2UAc5BK-YioB2SMGsdGztQ9BaT4k,10779
+dtsdance/metrics_fe.py,sha256=hzIl5BJmuCrkqJOHELVzXm3YAqrPttbyVkKBglS4mgQ,18978
+dtsdance/s3.py,sha256=Bh-cwLksfO5PewNtIzE_Md3rRLDLI1DUVoOD7Pou5T8,1294
+dtsdance/spacex.py,sha256=W_YGt_Wo_Cy4eDhgXhNfskayPHcZ1kg2Mil2ghI_VLE,4791
+dtsdance/tcc_inner.py,sha256=QqVyhwP0kCdp1HFHFgtme9o0im-eQNn1HVGdLt3Xr8g,1627
+dtsdance/tcc_open.py,sha256=JIfVhL3yfm0ftS_vzONE2QNzfqcZuX1mPTvjovAtwmE,6893
+dtsdance/tce.py,sha256=QwqPimZu0GKeP--sdbMP9h1e22nGuZL2FHqBUykXDOw,11563
+dts_dance-0.2.8.dist-info/METADATA,sha256=YjipvJ24oD--koIvidnlXrhll5UzI5Efpl-Zog_kQZc,826
+dts_dance-0.2.8.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+dts_dance-0.2.8.dist-info/RECORD,,

dtsdance/bytecloud.py

@@ -145,18 +145,23 @@ class ByteCloudClient:
             site_config = self.get_site_config(site)
             return self._acquire_jwt_token(site_config.endpoint, site_config.svc_secret)
 
-    def build_request_headers(self, site: str) -> dict[str, str]:
+    def build_request_headers(self, site: str, vregion: str | None = None) -> dict[str, str]:
         """
         构建请求头
 
         Args:
             site: 站点名称
-
+            vregion: 默认为 None
         Returns:
             dict[str, str]: 请求头字典
         """
         jwt_token = self.get_jwt_token(site)
-        return {"Content-Type": "application/json", "x-jwt-token": jwt_token}
+        headers = {"Content-Type": "application/json", "x-jwt-token": jwt_token}
+        if vregion:
+            headers["x-bcgw-vregion"] = vregion
+            headers["get-svc"] = "1"  # response header 中显示实际请求的机房
+
+        return headers
 
     def get_site_config(self, site: str) -> SiteConfig:
         """

dtsdance/ddutil.py

@@ -1,6 +1,7 @@
 from typing import Any, Optional
 from loguru import logger
 import requests
+import json
 
 
 def make_request(method: str, url: str, headers: dict[str, str], payload: Optional[dict] = None) -> dict[str, Any]:
@@ -37,3 +38,15 @@ def make_request(method: str, url: str, headers: dict[str, str], payload: Option
             error_msg += f", response.text: {response.text}"
         logger.warning(error_msg)
         raise
+
+
+def output_curl(url: str, headers: dict[str, str], payload: dict) -> str:
+    """
+    生成等效的 curl 命令字符串
+    """
+    curl_headers = " \\\n    ".join([f"-H '{k}: {v}'" for k, v in headers.items()])
+    curl_payload = json.dumps(payload, ensure_ascii=False)
+    curl_cmd = f"""curl -X POST '{url}' \\
+    {curl_headers} \\
+    -d '{curl_payload}'"""
+    return curl_cmd

dtsdance/dflow.py

@@ -1,8 +1,8 @@
 import collections
-from typing import Any, cast
+from typing import Any
 from loguru import logger
 
-from dtsdance.ddutil import make_request
+from dtsdance.ddutil import make_request, output_curl
 from .bytecloud import ByteCloudClient
 
 
@@ -19,12 +19,6 @@ class DFlowClient:
     def __init__(self, bytecloud_client: ByteCloudClient) -> None:
         self.bytecloud_client = bytecloud_client
 
-    def build_request_headers(self, site: str, vregion: str | None) -> dict[str, str]:
-        headers = self.bytecloud_client.build_request_headers(site)
-        if vregion:
-            headers["x-bcgw-vregion"] = vregion
-        return headers
-
     def get_task_info(self, site: str, task_id: str, vregion: str | None = None) -> dict[str, Any]:
         """
         获取 DFlow 任务信息
@@ -43,7 +37,7 @@ class DFlowClient:
         # 构建请求数据
         payload = {"id": int(task_id)}
 
-        response_data = make_request("POST", url, self.build_request_headers(site, vregion), payload)
+        response_data = make_request("POST", url, self.bytecloud_client.build_request_headers(site, vregion), payload)
 
         message = response_data.get("message")
         # logger.debug(f"get_task_info {site} {task_id}, message: {message}")
@@ -52,8 +46,7 @@ class DFlowClient:
             raise TaskNotFound(f"获取 DFlow 任务信息失败，站点: {site}, 任务 ID: {task_id} 不存在")
 
         try:
-            data = cast(dict, response_data.get("data", {}))
-            task = cast(dict, data.get("task", {}))
+            task = response_data.get("data", {}).get("task", {})
             # 提取核心信息
             filtered_data = {
                 "task_id": task.get("id", ""),
@@ -85,7 +78,7 @@ class DFlowClient:
         # 构建请求数据
         payload = {"dflow_id": int(dflow_id)}
 
-        response_data = make_request("POST", url, self.build_request_headers(site, vregion), payload)
+        response_data = make_request("POST", url, self.bytecloud_client.build_request_headers(site, vregion), payload)
 
         message = response_data.get("message", "")
         # logger.debug(f"get_dflow_info {site} {dflow_id}, message: {message}")
@@ -94,8 +87,7 @@ class DFlowClient:
             raise DFlowNotFound(f"获取 DFlow 进程信息失败，站点: {site}, 进程 ID: {dflow_id} 不存在")
 
         try:
-            data = cast(dict, response_data.get("data", {}))
-            dflow = cast(dict, data.get("dflow", {}))
+            dflow = response_data.get("data", {}).get("dflow", {})
             # 提取核心信息
             filtered_data = {
                 "dflow_id": dflow.get("id", ""),
@@ -125,12 +117,13 @@ class DFlowClient:
         site_info = self.bytecloud_client.get_site_config(site)
         return f"{site_info.endpoint}/bytedts/datasync/detail/{task_id}"
 
-    def init_resources(self, site: str, ctrl_env: str) -> bool:
+    def init_resources(self, site: str, vregion: str, ctrl_env: str) -> bool:
         """
         初始化 CTRL 环境资源
 
         Args:
             site: 站点名称
+            vregion: 虚拟区域
             ctrl_env: 控制环境
 
         Returns:
@@ -143,21 +136,21 @@ class DFlowClient:
         # 构建请求数据
         payload = {"ctrl_env": ctrl_env}
 
-        response_data = make_request("POST", url, self.build_request_headers(site, None), payload)
+        response_data = make_request("POST", url, self.bytecloud_client.build_request_headers(site, vregion), payload)
 
         message = response_data.get("message")
         logger.info(f"int_resources {site} {ctrl_env}, message: {message}")
 
         return message == "ok"
 
-    def list_resources(self, site: str, ctrl_env: str) -> list[str]:
+    def list_resources(self, site: str, vregion: str, ctrl_env: str) -> list[str]:
         """
         列举 CTRL 环境资源列表，连同 agent 列表
 
         Args:
             site: 站点名称
+            vregion: 虚拟区域
             ctrl_env: 控制环境
-
         Returns:
             list[str]: CTRL 环境资源列表
         """
@@ -166,9 +159,12 @@ class DFlowClient:
         url = f"{site_info.endpoint}/api/v1/bytedts/api/bytedts/v3/DescribeResources"
 
         # 构建请求数据
+        headers = self.bytecloud_client.build_request_headers(site, vregion)
         payload = {"offset": 0, "limit": 10, "ctrl_env": ctrl_env}
+        response_data = make_request("POST", url, headers, payload)
 
-        response_data = make_request("POST", url, self.build_request_headers(site, None), payload)
+        # curl_cmd = output_curl(url, headers, payload)
+        # print(f"Equivalent curl:\n{curl_cmd}")
 
         try:
             data = response_data.get("data", {})
@@ -181,7 +177,7 @@ class DFlowClient:
         """获取所有可用的控制环境列表，返回以domain为键，ctrl_env列表为值的字典"""
         site_info = self.bytecloud_client.get_site_config(site)
         url = f"{site_info.endpoint}/api/v1/bytedts/api/bytedts/v3/DescribeRegions"
-        resp_json = make_request("POST", url, self.build_request_headers(site, None), {})
+        resp_json = make_request("POST", url, self.bytecloud_client.build_request_headers(site, None), {})
         if resp_json["code"] != 0:
             raise Exception(resp_json["message"])
 

dtsdance/dsyncer.py

@@ -21,7 +21,7 @@ class DSyncerClient:
         self.envs = envs
         self.bytecloud_client = bytecloud_client
 
-    def _build_headers(self, env: str, secret_api: bool = False) -> dict[str, str]:
+    def _build_headers(self, site: str, secret_api: bool = False) -> dict[str, str]:
         """
         构建请求头
 
@@ -32,11 +32,11 @@ class DSyncerClient:
         Returns:
             dict[str, str]: 请求头字典
         """
-        jwt_token = self.bytecloud_client.get_jwt_token(env)
-        headers = {"X-Jwt-Token": jwt_token, "x-bcgw-vregion": env}
+        jwt_token = self.bytecloud_client.get_jwt_token(site)
+        headers = {"X-Jwt-Token": jwt_token}
 
         if secret_api:
-            token = self.envs[env].token
+            token = self.envs[site].token
             headers["Authorization"] = f"Token {token}"
 
         return headers

dtsdance/spacex.py

@@ -40,11 +40,11 @@ class SpaceXClient:
             logger.warning(f"do quest queryServerMeta exception: {e}")
             raise
 
-    def register_resource(self, site: str, data_raw: dict[str, Any]) -> bool:
+    def register_resource(self, site: str, payload: dict[str, Any]) -> bool:
         site_info = self.bytecloud_client.get_site_config(site)
         url = f"{site_info.endpoint_bytedts_spacex}/resource/v1/registerResource"
         try:
-            response = requests.post(url, json=data_raw, headers=self.bytecloud_client.build_request_headers(site))
+            response = requests.post(url, json=payload, headers=self.bytecloud_client.build_request_headers(site))
             # print(f"response status code: {response.status_code}, response text: {response.text}")
 
             response.raise_for_status()
@@ -56,11 +56,11 @@ class SpaceXClient:
             logger.warning(f"do quest registerResource exception: {e}")
             raise
 
-    def list_resources(self, site: str, data_raw: dict[str, Any]) -> list[str]:
+    def list_resources(self, site: str, payload: dict[str, Any]) -> list[str]:
         site_info = self.bytecloud_client.get_site_config(site)
         url = f"{site_info.endpoint_bytedts_spacex}/resource/v1/listResource"
         try:
-            response = requests.post(url, json=data_raw, headers=self.bytecloud_client.build_request_headers(site))
+            response = requests.post(url, json=payload, headers=self.bytecloud_client.build_request_headers(site))
             # print(f"response status code: {response.status_code}, response text: {response.text}")
 
             response.raise_for_status()
@@ -78,7 +78,7 @@ class SpaceXClient:
     def register_gateway(self, site: str, gateway_info: GatewayInfo) -> bool:
         site_info = self.bytecloud_client.get_site_config(site)
         url = f"{site_info.endpoint_bytedts_spacex}/bytedts/v1/registryGateway"
-        data_raw = {
+        payload = {
             "region": gateway_info.mgr_name,
             "server_region": gateway_info.mgr_name,
             "cluster_region": gateway_info.ctrl_name,
@@ -97,7 +97,7 @@ class SpaceXClient:
             "unified_tao_service": 1,
         }
         try:
-            response = requests.post(url, json=data_raw, headers=self.bytecloud_client.build_request_headers(site))
+            response = requests.post(url, json=payload, headers=self.bytecloud_client.build_request_headers(site))
             response.raise_for_status()
 
             result = response.json()
@@ -107,11 +107,11 @@ class SpaceXClient:
             logger.warning(f"do quest registryGateway exception: {e}")
             raise
 
-    def gateway_operation(self, site: str, operation: str, data_raw: dict[str, Any]) -> dict[str, Any]:
+    def gateway_operation(self, site: str, operation: str, payload: dict[str, Any]) -> dict[str, Any]:
         site_info = self.bytecloud_client.get_site_config(site)
         url = f"{site_info.endpoint_bytedts_spacex}/bytedts/v1/{operation}"
         try:
-            response = requests.post(url, json=data_raw, headers=self.bytecloud_client.build_request_headers(site))
+            response = requests.post(url, json=payload, headers=self.bytecloud_client.build_request_headers(site))
             # print(f"response status code: {response.status_code}, response text: {response.text}")
 
             response.raise_for_status()

dtsdance/tcc_inner.py

@@ -27,7 +27,7 @@ class TCCInnerClient:
         site_info = self.bytecloud_client.get_site_config(site)
         url = f"{site_info.endpoint}/api/v3/tcc/bcc/config/list_v2"
 
-        data_raw = {
+        payload = {
             "ns_name": ns_name,
             "region": region,
             "dir_path": dir,
@@ -40,7 +40,7 @@ class TCCInnerClient:
         }
 
         try:
-            response = requests.post(url, headers=self.bytecloud_client.build_request_headers(site), json=data_raw, timeout=30)
+            response = requests.post(url, headers=self.bytecloud_client.build_request_headers(site), json=payload, timeout=30)
             response.raise_for_status()
             result = response.json()
 

dtsdance/tce.py

@@ -0,0 +1,289 @@
+from typing import Any
+import requests
+
+from .bytecloud import ByteCloudClient
+
+
+class TCEError(Exception):
+    pass
+
+
+class TCEErrorServiceExist(Exception):
+    pass
+
+
+class TCEErrorClusterExist(Exception):
+    pass
+
+
+class TCEClient:
+    """Client for TCE deployment API."""
+
+    def __init__(self, bytecloud_client: ByteCloudClient) -> None:
+        """
+        Initialize TCE client.
+
+        Args:
+            base_url: TCE API base URL
+            jwt_token: JWT token for authentication (can be fetched automatically)
+        """
+        self.bytecloud_client = bytecloud_client
+
+    def create_service(self, site: str, ctrl_name: str, bytetree_parent_id: int) -> int | None:
+        site_info = self.bytecloud_client.get_site_config(site)
+        url = f"{site_info.endpoint}/api/v1/tce/open-apis/v1/deployment/service/create/"
+
+        service_info = {
+            "meta": {"name": f"bytedts.cm.{ctrl_name}", "psm": f"bytedts.cm.{ctrl_name}", "env": "prod"},
+            "auth": {
+                "galaxy_info": {"parent_id": bytetree_parent_id},
+                "approval_controls": ["reset"],
+                "iam_info": {"owners": [{"username": site_info.svc_account}, {"username": "mahang"}]},
+                "owner_name": site_info.svc_account,
+            },
+            "extra_features": {
+                "service_token": {
+                    "tech_stack": {"language": "go", "framework": "custom", "platform_types_v_2": {"tce": ["http"]}},
+                    "deploy": {"type": "stateless_compute", "scale": {"enable": True}, "descale": {"enable": True}},
+                    "level": "P2",
+                    "purpose": "convention",
+                    "contain_user_personal_info": False,
+                    "data_sensitivity_level": "L2",
+                }
+            },
+            "build": {
+                "dependency_mode": "default",
+                "base_image": "debian.bookworm.tce_service:latest",
+                "install_package_info": [],
+                "scm_repo_info": [
+                    {"name": "bytedts/dflow/dtscm", "remote_id": 387073, "main_repo": True, "path": "dts-cm"},
+                    {"name": "toutiao/load", "remote_id": 667, "path": "toutiao/load"},
+                    {"name": "toutiao/runtime", "remote_id": 631, "path": "toutiao/runtime"},
+                ],
+                "language": "go",
+                "framework_support": "none",
+                "tech_stack": "custom",
+            },
+            "runtime": {
+                "umount_ss_conf": False,
+                "new_ports": [
+                    {"intent": "primary", "port": 8088, "is_primary": True, "type": "tcp"},
+                    {"intent": "other", "port": 3033, "is_primary": False, "type": "tcp"},
+                ],
+                "is_runtime": False,
+                "app_path": "dts-cm/bootstrap.sh",
+                "enable_iast": False,
+            },
+        }
+
+        payload = {"service_info": service_info}
+
+        try:
+            response = requests.post(url, json=payload, headers=self.bytecloud_client.build_request_headers(site), timeout=60)
+            if response.status_code != 200:
+                if "already exists" in response.text:
+                    raise TCEErrorServiceExist("TCE cluster with the same name already exists")
+
+                raise TCEError(f"TCE API request failed with status {response.status_code}\n" f"Response: {response.text}")
+
+            result = response.json()
+            if result.get("code", 0) == -1:
+                raise TCEError(f"TCE API error: {result.get('error')}")
+
+            return result.get("data", {}).get("service", 0)
+
+        except requests.RequestException as e:
+            raise TCEError(f"Failed to create TCE cluster: {str(e)}") from e
+
+    def create_cluster(self, site: str, cluster_info: dict[str, Any], service: int) -> int | None:
+        """
+        Create TCE cluster.
+
+        Args:
+            cluster_info: Cluster configuration (meta, resource, runtime)
+            service: Service ID
+
+        Returns:
+            The created cluster ID, or None if creation failed
+
+        Raises:
+            TCEError: If API call fails
+        """
+        site_info = self.bytecloud_client.get_site_config(site)
+        url = f"{site_info.endpoint}/api/v1/tce/open-apis/v1/deployment/cluster/create/"
+        payload = {"cluster_info": cluster_info, "service": service}
+
+        try:
+            response = requests.post(url, json=payload, headers=self.bytecloud_client.build_request_headers(site), timeout=60)
+            if response.status_code != 200:
+                if "a same named TCE cluster already exists" in response.text:
+                    raise TCEErrorClusterExist("TCE cluster with the same name already exists")
+
+                raise TCEError(f"TCE API request failed with status {response.status_code}\n" f"Response: {response.text}")
+
+            result = response.json()
+
+            # Check for errors in response
+            if not isinstance(result, dict):
+                raise TCEError(f"Unexpected response format: {result}")
+
+            # TCE API may return error information in different formats
+            if result.get("code", 0) == -1:
+                raise TCEError(f"TCE API error: {result.get('error')}")
+
+            data = result.get("data", {})
+            create_success = data.get("deployment_type", "") == "create" and data.get("status", "") == "pending"
+            if not create_success:
+                raise TCEError(f"TCE cluster creation failed: {result}")
+
+            cluster_id = data.get("cluster_info", {}).get("meta", {}).get("id", 0)
+            # Ensure cluster_id is an integer
+            return int(cluster_id) if cluster_id else 0
+
+        except requests.RequestException as e:
+            raise TCEError(f"Failed to create TCE cluster: {str(e)}") from e
+
+    def delete_cluster(self, site: str, cluster_id: int, service_id: int, cluster_name: str) -> int | None:
+        """
+        Delete TCE cluster (reset/destroy).
+
+        Args:
+            site: Site identifier
+            cluster_id: Cluster ID to delete
+            service_id: Service ID
+            cluster_name: Cluster name
+
+        Returns:
+            Response data
+
+        Raises:
+            TCEError: If API call fails
+        """
+        site_info = self.bytecloud_client.get_site_config(site)
+        url = f"{site_info.endpoint}/api/v1/tce/open-apis/v1/deployment/cluster/reset/"
+        payload = {"cluster": cluster_id, "service": service_id}
+
+        # print(f"Deleting TCE cluster {cluster_id} for service {service_id}")
+        try:
+            response = requests.post(url, json=payload, headers=self.bytecloud_client.build_request_headers(site), timeout=60)
+            if response.status_code != 200:
+                raise TCEError(f"TCE API request failed with status {response.status_code}\n" f"Response: {response.text}")
+
+            result = response.json()
+            if result.get("code", 0) == -1:
+                error_msg = result.get("error", "")
+                if "do not exist" in error_msg:
+                    print(f"cluster {cluster_name} not exist, maybe already deleted")
+                    return None
+
+                raise TCEError(f"TCE API error: {error_msg}")
+
+            data = result.get("data", {})
+            # 查找 "人工确认" step 的 id
+            steps = data.get("steps", [])
+            for step in steps:
+                if step.get("step_type") == "ConfirmStep":
+                    return step.get("id")
+
+            return None
+
+        except requests.RequestException as e:
+            raise TCEError(f"Failed to delete TCE cluster: {cluster_name}, msg: {str(e)}") from e
+
+    def get_cluster_info(self, site: str, cluster_id: int) -> dict[str, Any]:
+        site_info = self.bytecloud_client.get_site_config(site)
+        url = f"{site_info.endpoint}/api/v1/tce/open-apis/v1/clusters/{cluster_id}/"
+
+        try:
+            response = requests.get(url, headers=self.bytecloud_client.build_request_headers(site), timeout=60)
+            if response.status_code != 200:
+                raise TCEError(f"TCE API request failed with status {response.status_code}\n" f"Response: {response.text}")
+
+            # print(f"get_cluster_info response.text: {response.text}")
+            result = response.json()
+            if result.get("code", 0) == -1:
+                raise TCEError(f"TCE API error: {result.get('error')}")
+
+            return result.get("data", {})
+
+        except requests.RequestException as e:
+            raise TCEError(f"Failed to create TCE cluster: {str(e)}") from e
+
+    def is_cluster_running(self, site: str, cluster_id: int | None) -> bool:
+        if cluster_id is None:
+            return False
+
+        cluster_info = self.get_cluster_info(site, cluster_id)
+        return cluster_info.get("meta", {}).get("status", "unknown") == "running"
+
+    def restart_cluster(self, site: str, service_id: int, cluster_id: int) -> list[int]:
+        site_info = self.bytecloud_client.get_site_config(site)
+        url = f"{site_info.endpoint}/api/v1/tce/open-apis/v1/deployment/cluster/pod_exec/"
+
+        payload = {
+            "service": service_id,
+            "cluster": cluster_id,
+            "exec_command": "rebuild",
+            "exec_params": {
+                "surge_percent": 25,
+                "ready_seconds": 10,
+                "timeout": 120,
+                "tolerance_num": 3,
+                "failed_policy": "remain",
+                "need_check": True,
+            },
+            "exec_type": "rebuild",
+            "parallel_mode": True,
+        }
+
+        try:
+            response = requests.post(url, json=payload, headers=self.bytecloud_client.build_request_headers(site), timeout=60)
+            if response.status_code != 200:
+                raise TCEError(f"TCE API request failed with status {response.status_code}\n" f"Response: {response.text}")
+
+            result = response.json()
+            if result.get("code", 0) == -1:
+                raise TCEError(f"TCE API error: {result.get('error')}")
+
+            steps = result.get("data", {}).get("steps", [])
+            step_ids = [step.get("id", 0) for step in steps]
+            return step_ids
+
+        except requests.RequestException as e:
+            raise TCEError(f"Failed to create TCE cluster: {str(e)}") from e
+
+    def confirm_step(self, site: str, step_id: int, action: str = "confirm") -> bool:
+        """
+        Confirm a step in TCE cluster deployment.
+
+        Args:
+            step_id: Step ID
+
+        Returns:
+            Response data
+
+        Raises:
+            TCEError: If API call fails
+        """
+        site_info = self.bytecloud_client.get_site_config(site)
+        url = f"{site_info.endpoint}/api/v1/tce/open-apis/v1/deployment_steps/{step_id}/actions/"
+        payload = {"action": action}
+
+        try:
+            response = requests.post(url, json=payload, headers=self.bytecloud_client.build_request_headers(site), timeout=60)
+            # print(f"Confirm step response: {response.text}")
+
+            response.raise_for_status()
+            result = response.json()
+
+            # Check for errors in response
+            if not isinstance(result, dict):
+                raise TCEError(f"Unexpected response format: {result}")
+
+            if result.get("code", 0) == -1:
+                raise TCEError(f"TCE API error: {result.get('error')}")
+
+            return True
+
+        except requests.RequestException as e:
+            raise TCEError(f"Failed to confirm ticket, msg: {str(e)}") from e

dts_dance-0.2.5.dist-info/RECORD

@@ -1,15 +0,0 @@
-dtsdance/__init__.py,sha256=Yl_jEZ5weYfcrklnDvwB4wSgCOvMBLRRgWx0gHs3qfM,49
-dtsdance/bytecloud.py,sha256=s1suo-YexMH6gUHYwC6P5k6AeyOm7Q_acYh_gHCDVjI,5654
-dtsdance/ddutil.py,sha256=aZ-fG_e2ZkHs8sxZsudf6bVCnp_OI2Z4WjbxOfPoJRQ,1134
-dtsdance/dflow.py,sha256=lEn_gd0pUOqEhuvs0hSQICR8LKOk3wP4DBbeyHZ-A44,7766
-dtsdance/dsyncer.py,sha256=gK2rRiO3041P3efbmXgvLPeTgKlJowuPR7CCz_uyoSA,9926
-dtsdance/feishu_base.py,sha256=dYYNnsCjGZYN2I6inR9FRQoUiDLB7j40rzydw50N38o,11825
-dtsdance/feishu_table.py,sha256=JleCjQJ-rcbC15v2UAc5BK-YioB2SMGsdGztQ9BaT4k,10779
-dtsdance/metrics_fe.py,sha256=hzIl5BJmuCrkqJOHELVzXm3YAqrPttbyVkKBglS4mgQ,18978
-dtsdance/s3.py,sha256=Bh-cwLksfO5PewNtIzE_Md3rRLDLI1DUVoOD7Pou5T8,1294
-dtsdance/spacex.py,sha256=oVnHXTj-cuq2vcsUJq33_aaq0TQp7A52kldvspKueI4,4799
-dtsdance/tcc_inner.py,sha256=Z0Wr-z2YPwAti6BO_fsX0VPT5pCbfuCBLQ37UKMFL7U,1629
-dtsdance/tcc_open.py,sha256=JIfVhL3yfm0ftS_vzONE2QNzfqcZuX1mPTvjovAtwmE,6893
-dts_dance-0.2.5.dist-info/METADATA,sha256=-skiMz-CuBlzpt7BBDJpOEXrmsj8np_xKnG-6K_HPao,826
-dts_dance-0.2.5.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-dts_dance-0.2.5.dist-info/RECORD,,