dts-dance 0.1.7__py3-none-any.whl → 0.1.9__py3-none-any.whl

{dts_dance-0.1.7.dist-info → dts_dance-0.1.9.dist-info}/METADATA

@@ -1,12 +1,13 @@
 Metadata-Version: 2.4
 Name: dts-dance
-Version: 0.1.7
+Version: 0.1.9
 Summary: dts dance lib
 Keywords: observation,tools
 Requires-Python: >=3.12
-Requires-Dist: loguru==0.7.3
-Requires-Dist: pyyaml==6.0.3
-Requires-Dist: requests==2.32.5
+Requires-Dist: boto3<2.0.0,>=1.42.27
+Requires-Dist: loguru<0.8.0,>=0.7.3
+Requires-Dist: pyyaml<7.0.0,>=6.0.3
+Requires-Dist: requests<3.0.0,>=2.32.5
 Description-Content-Type: text/markdown
 
 # dts-dance

{dts_dance-0.1.7.dist-info → dts_dance-0.1.9.dist-info}/RECORD

@@ -1,13 +1,13 @@
 dtsdance/__init__.py,sha256=Yl_jEZ5weYfcrklnDvwB4wSgCOvMBLRRgWx0gHs3qfM,49
 dtsdance/bytecloud.py,sha256=joixJEQxX9k3VqFDNrccTVSI2WVdxdZ1JyuWFVF9zjo,4998
-dtsdance/dflow.py,sha256=pLhx9G3YTiCrz8VcycyTMedREShWN4RsxDmaRTwRhdA,3649
+dtsdance/dflow.py,sha256=gow-zCX5lbQqqW2MRWvWynej7pZ_TQ5NmOb__j2aQT4,5497
 dtsdance/dsyncer.py,sha256=1OateZLGekYP_EF9_oAHqsj4sdQm6ahIoRllEgT9Y3I,11142
 dtsdance/feishu_base.py,sha256=2j4ZM4PFqJ-9EhC6DQ1OmAg--3VBGZyyRuxyjL0j6OU,3733
 dtsdance/feishu_table.py,sha256=ZUeoKrM4nmm5hFhc3vWOVYeLP390orm-284Od92G4iQ,8424
 dtsdance/metrics_fe.py,sha256=uPdbjGaaYBOv-rK7lKUx6aVT5Sj6ZkW9m06NyqMrPf0,18990
-dtsdance/s3.py,sha256=EMY5uDnqRUuHnZQBkquLdQjLR5I51LuCpCPNsDz0Z0w,1340
+dtsdance/s3.py,sha256=Bh-cwLksfO5PewNtIzE_Md3rRLDLI1DUVoOD7Pou5T8,1294
 dtsdance/spacex.py,sha256=wgbuwDTLXopJnLn2puX-9MfeVXvi_nQ3C9uN8McoJf4,2113
 dtsdance/tcc.py,sha256=M_0cOVYyvUgjnC1uKWdz3YgRplafQUqX39gX_YEg8eE,6563
-dts_dance-0.1.7.dist-info/METADATA,sha256=u1kzvSXzkE4j1olIgqAz3uXYSy8ZU2xIAcMt2sOUliQ,735
-dts_dance-0.1.7.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-dts_dance-0.1.7.dist-info/RECORD,,
+dts_dance-0.1.9.dist-info/METADATA,sha256=IbvejpKLh-EM12cf9XtlwCVkUbD5jjvRc_Pz7PA_5b8,793
+dts_dance-0.1.9.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+dts_dance-0.1.9.dist-info/RECORD,,

dtsdance/dflow.py

@@ -3,6 +3,7 @@ from loguru import logger
 from .bytecloud import ByteCloudHelper
 import requests
 
+
 class DFlowHelper:
 
     def __init__(self, bytecloud_helper: ByteCloudHelper) -> None:
@@ -112,3 +113,61 @@ class DFlowHelper:
         # 根据环境生成对应的 scope 参数
         env_info = self.bytecloud_helper.get_env_info(env)
         return f"{env_info.endpoint}/bytedts/datasync/detail/{task_id}?scope={env}"
+
+    def init_resources(self, env: str, ctrl_env: str) -> bool:
+        """
+        初始化 CTRL 环境资源
+
+        Args:
+            env: 环境名称
+            ctrl_env: 控制环境
+
+        Returns:
+            bool: CTRL 环境资源初始化结果
+        """
+        # 构建 API URL
+        env_info = self.bytecloud_helper.get_env_info(env)
+        url = f"{env_info.endpoint}/api/v1/bytedts/api/bytedts/v3/InitSystemResource"
+
+        # 准备请求头
+        headers = self._build_headers(env)
+
+        # 构建请求数据
+        json_data = {"ctrl_env": ctrl_env}
+
+        response_data = self._make_request("POST", url, headers, json_data)
+
+        message = response_data.get("message")
+        logger.info(f"int_resources {env} {ctrl_env}, message: {message}")
+
+        return message == "ok"
+
+    def list_resources(self, env: str, ctrl_env: str) -> list[str]:
+        """
+        列举 CTRL 环境资源列表
+
+        Args:
+            env: 环境名称
+            ctrl_env: 控制环境
+
+        Returns:
+            list[str]: CTRL 环境资源列表
+        """
+        # 构建 API URL
+        env_info = self.bytecloud_helper.get_env_info(env)
+        url = f"{env_info.endpoint}/api/v1/bytedts/api/bytedts/v3/DescribeResources"
+
+        # 准备请求头
+        headers = self._build_headers(env)
+
+        # 构建请求数据
+        json_data = {"offset": 0, "limit": 10, "ctrl_env": ctrl_env}
+
+        response_data = self._make_request("POST", url, headers, json_data)
+
+        try:
+            data = cast(dict, response_data.get("data", {}))
+            items = cast(list, data.get("items", []))
+            return [item["name"] for item in items]
+        except (KeyError, AttributeError, Exception) as e:
+            raise Exception(f"无法从响应中提取 CTRL 环境资源列表数据: {str(e)}")

dtsdance/s3.py

@@ -1,7 +1,6 @@
 import os
 from typing import Optional
 import boto3
-from botocore.exceptions import ClientError
 from loguru import logger
 
 
@@ -35,6 +34,6 @@ class S3Client:
             file_url = f"{self.base_url}/{s3_file_name}"
             logger.info(f"文件已成功上传到 S3: {file_url}")
             return file_url
-        except ClientError as e:
+        except Exception as e:
             logger.error(f"上传文件到 S3 时出错: {e}")
             return None