dts-dance 0.2.2__tar.gz → 0.2.4__tar.gz

{dts_dance-0.2.2 → dts_dance-0.2.4}/PKG-INFO

@@ -1,12 +1,13 @@
 Metadata-Version: 2.4
 Name: dts-dance
-Version: 0.2.2
+Version: 0.2.4
 Summary: dts dance lib
 Keywords: observation,tools
 Requires-Python: >=3.12
-Requires-Dist: boto3<2.0.0,>=1.42.29
+Requires-Dist: boto3<2.0.0,>=1.42.30
 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
 Requires-Dist: requests<3.0.0,>=2.32.5
 Description-Content-Type: text/markdown
 

{dts_dance-0.2.2 → dts_dance-0.2.4}/dtsdance/bytecloud.py

@@ -20,9 +20,9 @@ class ByteCloudClient:
     """
 
     # 每小时刷新一次，单位为秒
-    _REFRESH_INTERVAL = 1 * 60 * 60
+    _REFRESH_INTERVAL = 2 * 60 * 60
 
-    def __init__(self, enable_jwt_cache: bool, sites: dict[str, SiteConfig] | None = None) -> None:
+    def __init__(self, sites: dict[str, SiteConfig] | None = None, enable_jwt_cache: bool = False) -> None:
         """
         初始化 ByteCloud Client
         从配置文件加载所有环境的信息，并为每个环境初始化 JWT 令牌

{dts_dance-0.2.2 → dts_dance-0.2.4}/dtsdance/dflow.py

@@ -150,7 +150,7 @@ class DFlowClient:
         """
         # 根据环境生成对应的 scope 参数
         site_info = self.bytecloud_client.get_site_config(site)
-        return f"{site_info.endpoint}/bytedts/datasync/detail/{task_id}?scope={site}"
+        return f"{site_info.endpoint}/bytedts/datasync/detail/{task_id}"
 
     def init_resources(self, site: str, ctrl_env: str) -> bool:
         """
@@ -179,7 +179,7 @@ class DFlowClient:
 
     def list_resources(self, site: str, ctrl_env: str) -> list[str]:
         """
-        列举 CTRL 环境资源列表
+        列举 CTRL 环境资源列表，连同 agent 列表
 
         Args:
             site: 站点名称
@@ -198,8 +198,8 @@ class DFlowClient:
         response_data = self._make_request("POST", url, self.bytecloud_client.build_request_headers(site), json_data)
 
         try:
-            data = cast(dict, response_data.get("data", {}))
-            items = cast(list, data.get("items", []))
+            data = response_data.get("data", {})
+            items = data.get("items", [])
             return [item["name"] for item in items]
         except (KeyError, AttributeError, Exception) as e:
             raise Exception(f"无法从响应中提取 CTRL 环境资源列表数据: {str(e)}")

dts_dance-0.2.4/dtsdance/feishu_base.py

@@ -0,0 +1,296 @@
+from datetime import datetime, timedelta
+import threading
+import json
+from loguru import logger
+from typing import Any
+from requests_toolbelt import MultipartEncoder
+
+import requests
+
+URL_FEISHU_OPEN_API: str = "https://fsopen.bytedance.net/open-apis"
+
+
+class FeishuBase:
+    """飞书基础类，传入机器人认证信息，支持自动token续期管理"""
+
+    def __init__(self, bot_app_id: str, bot_app_secret: str):
+        """
+        初始化飞书基础类
+        """
+        self.tenant_access_token: str | None = None
+        self.token_expire_time: datetime | None = None
+        self._token_lock = threading.Lock()
+
+        self.bot_app_id = bot_app_id
+        self.bot_app_secret = bot_app_secret
+
+    def _get_tenant_access_token(self) -> str:
+        """
+        获取tenant_access_token
+
+        Returns:
+            tenant_access_token字符串
+
+        Raises:
+            Exception: 获取token失败时抛出异常
+        """
+        url = f"{URL_FEISHU_OPEN_API}/auth/v3/tenant_access_token/internal"
+        payload = {"app_id": self.bot_app_id, "app_secret": self.bot_app_secret}
+        headers = {"Content-Type": "application/json"}
+
+        try:
+            response = requests.post(url, json=payload, headers=headers)
+            response.raise_for_status()
+            result = response.json()
+            if result.get("code") != 0:
+                raise Exception(f"获取tenant_access_token失败: {result.get('msg')}")
+
+            token = result.get("tenant_access_token")
+            expire_in = result.get("expire", 7200)  # 默认2小时过期
+
+            # 设置过期时间，提前5分钟刷新
+            self.token_expire_time = datetime.now() + timedelta(seconds=expire_in - 300)
+
+            logger.info(f"成功获取tenant_access_token，过期时间: {self.token_expire_time}")
+            return token
+
+        except requests.RequestException as e:
+            logger.warning(f"请求tenant_access_token失败: {e}")
+            raise Exception(f"网络请求失败: {e}")
+        except Exception as e:
+            logger.warning(f"获取tenant_access_token失败: {e}")
+            raise
+
+    def get_access_token(self) -> str:
+        """
+        获取有效的access token，自动处理续期
+
+        Returns:
+            有效的tenant_access_token
+        """
+        with self._token_lock:
+            # 检查token是否存在或已过期
+            if self.tenant_access_token is None or self.token_expire_time is None or datetime.now() >= self.token_expire_time:
+
+                logger.info("Token不存在或已过期，重新获取")
+                self.tenant_access_token = self._get_tenant_access_token()
+
+            return self.tenant_access_token
+
+    def make_request(self, method: str, endpoint: str, **kwargs) -> requests.Response:
+        """
+        发起API请求的通用方法
+
+        Args:
+            method: HTTP方法
+            endpoint: API端点
+            **kwargs: 其他请求参数
+
+        Returns:
+            Response对象
+        """
+        url = f"{URL_FEISHU_OPEN_API}{endpoint}"
+
+        # 获取有效token
+        token = self.get_access_token()
+        # logger.debug(f"使用token: {token} 发送请求: {method} {url}")
+
+        # 设置认证头
+        headers = kwargs.get("headers", {"Content-Type": "application/json"})
+        headers["Authorization"] = f"Bearer {token}"
+        kwargs["headers"] = headers
+
+        try:
+            response = requests.request(method, url, **kwargs)
+            logger.debug(f"response_json: {response.json()}")
+            response.raise_for_status()
+            return response
+        except requests.RequestException as e:
+            logger.warning(f"API请求失败: {method} {url}, error: {e}")
+            raise
+
+    def upload_image(self, image_path: str) -> str | None:
+        """
+        上传图片到飞书
+
+        Args:
+            image_path: 图片文件路径
+            image_type: 图片类型，默认为'message'
+
+        Returns:
+            image_key
+
+        """
+        if not image_path:
+            raise ValueError("必须提供image_path")
+
+        file_obj = None
+        try:
+            # 准备文件
+            file_obj = open(image_path, "rb")
+            file_name = image_path.split("/")[-1]
+
+            # 构建表单数据
+            form = {"image_type": "message", "image": (file_name, file_obj, "image/png")}
+            multi_form = MultipartEncoder(form)
+            headers = {"Content-Type": multi_form.content_type}
+            response = self.make_request("POST", "/im/v1/images", headers=headers, data=multi_form)
+            result = response.json()
+            if result.get("code") != 0:
+                raise Exception(f"上传图片失败: {result.get('msg')}")
+
+            logger.info(f"图片上传成功，image_key: {result.get('data', {}).get('image_key')}")
+            logger.debug(f"上传响应logid: {response.headers.get('X-Tt-Logid')}")
+
+            return result.get("data", {}).get("image_key")
+
+        except Exception as e:
+            logger.warning(f"上传图片失败: {e}")
+            return None
+        finally:
+            # 如果是通过路径打开的文件，需要关闭
+            if image_path and file_obj and hasattr(file_obj, "close"):
+                file_obj.close()
+
+    def send_text_message(self, receive_id: str, text: str, receive_id_type: str) -> dict[str, Any]:
+        """
+        发送文本消息
+
+        Args:
+            receive_id: 接收者ID（群聊ID、用户ID等）
+            text: 消息文本内容
+            receive_id_type: 接收者ID类型，可选值：open_id, user_id, union_id, email, chat_id
+
+        Returns:
+            发送结果
+        """
+        content = json.dumps({"text": text}, ensure_ascii=False)
+        return self.send_message(receive_id, "text", content, receive_id_type)
+
+    def send_image_message(self, receive_id: str, image_key: str, receive_id_type: str) -> dict[str, Any]:
+        """
+        发送图片消息
+
+        Args:
+            receive_id: 接收者ID
+            image_key: 图片key（通过upload_image获取）
+            receive_id_type: 接收者ID类型
+
+        Returns:
+            发送结果
+        """
+        content = json.dumps({"image_key": image_key}, ensure_ascii=False)
+        return self.send_message(receive_id, "image", content, receive_id_type)
+
+    def send_card_message(self, receive_id: str, header: dict, elements: list, receive_id_type: str) -> dict[str, Any]:
+        """
+        发送卡片消息
+
+        Args:
+            receive_id: 接收者ID
+            header: 卡片消息的头部内容（符合飞书卡片规范的JSON）
+            elements: 卡片消息的主体元素列表
+            receive_id_type: 接收者ID类型
+
+        Returns:
+            发送结果
+        """
+        card_content = {"schema": "2.0", "config": {"update_multi": True}, "body": {"direction": "vertical", "elements": elements}, "header": header}
+        content = json.dumps(card_content, ensure_ascii=False)
+        return self.send_message(receive_id, "interactive", content, receive_id_type)
+
+    def send_message(self, receive_id: str, msg_type: str, content: str, receive_id_type: str = "chat_id") -> dict[str, Any]:
+        """
+        发送消息的通用方法
+
+        Args:
+            receive_id: 接收者ID
+            msg_type: 消息类型（text, image, card等）
+            content: 消息内容
+            receive_id_type: 接收者ID类型
+
+        Returns:
+            发送结果
+
+        Raises:
+            Exception: 发送失败
+        """
+        payload = {"receive_id": receive_id, "msg_type": msg_type, "content": content}
+        params = {"receive_id_type": receive_id_type}
+
+        try:
+            response = self.make_request("POST", "/im/v1/messages", json=payload, params=params)
+            result = response.json()
+            if result.get("code") != 0:
+                raise Exception(f"请求失败: {result.get('msg')}")
+
+            logger.info(f"消息发送成功，message_id: {result.get('data', {}).get('message_id')}")
+            logger.debug(f"发送响应logid: {response.headers.get('X-Tt-Logid')}")
+
+            return result.get("data", {})
+
+        except Exception as e:
+            logger.warning(f"发送消息失败: {e}")
+            raise
+
+    def reply_message(self, message_id: str, msg_type: str, content: str) -> dict[str, Any]:
+        """
+        回复某条消息
+
+        Args:
+            message_id: 消息ID
+            msg_type: 消息类型（text, image, card等）
+            content: 消息内容
+
+        Returns:
+            回复结果
+
+        Raises:
+            Exception: 回复失败
+        """
+        payload = {"reply_in_thread": False, "msg_type": msg_type, "content": content}
+
+        try:
+            response = self.make_request("POST", f"/im/v1/messages/{message_id}/reply", json=payload)
+            result = response.json()
+            if result.get("code") != 0:
+                raise Exception(f"请求失败: {result.get('msg')}")
+
+            logger.info(f"消息回复成功，message_id: {result.get('data', {}).get('message_id')}")
+            logger.debug(f"发送响应logid: {response.headers.get('X-Tt-Logid')}")
+
+            return result.get("data", {})
+
+        except Exception as e:
+            logger.warning(f"回复消息失败: {e}")
+            raise
+
+    def batch_get_user_ids(self, emails: list[str]) -> list[str]:
+        """
+        批量获取用户ID
+
+        Args:
+            emails: 用户邮箱列表
+
+        Returns:
+            批量获取结果（用户ID列表）
+
+        Raises:
+            Exception: 获取失败
+        """
+        payload = {"emails": emails, "include_resigned": False}
+
+        try:
+            response = self.make_request("POST", f"/contact/v3/users/batch_get_id?user_id_type=user_id", json=payload)
+            result = response.json()
+            if result.get("code") != 0:
+                raise Exception(f"请求失败: {result.get('msg')}")
+
+            user_ids = [user.get("user_id") for user in result.get("data", {}).get("user_list", [])]
+
+            logger.debug(f"发送响应logid: {response.headers.get('X-Tt-Logid')}")
+
+            return user_ids
+        except Exception as e:
+            logger.warning(f"批量获取用户ID失败: {e}")
+            raise

{dts_dance-0.2.2 → dts_dance-0.2.4}/dtsdance/feishu_table.py

@@ -1,6 +1,6 @@
 from datetime import datetime
 import threading
-from typing import Callable, Optional, Dict, Any
+from typing import Callable, Any
 from loguru import logger
 import requests
 
@@ -16,15 +16,15 @@ class FeishuTable:
         """
         初始化飞书表格
         """
-        self.tenant_access_token: Optional[str] = None
-        self.token_expire_time: Optional[datetime] = None
+        self.tenant_access_token: str | None = None
+        self.token_expire_time: datetime | None = None
         self._token_lock = threading.Lock()
 
         self.feishu_base = feishu_base
         self.table_app_token = table_app_token
         self.table_id = table_id
 
-    def get_app_table_record_id(self, table_view_id: str, task_id: str) -> Optional[str]:
+    def get_app_table_record_id(self, table_view_id: str, task_id: str) -> str | None:
         """
         根据任务ID查询飞书多维表格记录，返回记录ID
         """
@@ -59,7 +59,7 @@ class FeishuTable:
             logger.warning(f"查询 {task_id} 表格记录失败: {e}")
             return None
 
-    def update_app_table_record(self, record_id: str, fields: Dict[str, Any]) -> bool:
+    def update_app_table_record(self, record_id: str, fields: dict[str, Any]) -> bool:
         """
         更新飞书多维表格记录
 
@@ -81,14 +81,14 @@ class FeishuTable:
                 logger.warning(f"更新表格记录失败: {result.get('msg')}")
                 return False
 
-            logger.info(f"成功更新记录ID {record_id}，字段: {fields}")
+            # logger.debug(f"成功更新记录ID {record_id}，字段: {fields}")
             return True
 
         except Exception as e:
             logger.warning(f"更新表格记录失败: {e}")
             return False
 
-    def fetch_records(self, table_view_id: str, field_names: list[str], page_size: int = 100, page_token: Optional[str] = None) -> Dict[str, Any]:
+    def fetch_records(self, table_view_id: str, field_names: list[str], page_size: int = 100, page_token: str | None = None) -> dict[str, Any]:
         """
         获取表格记录
 
@@ -121,7 +121,7 @@ class FeishuTable:
             logger.error(f"获取记录失败: {e}")
             raise
 
-    def parse_record(self, record: Dict[str, Any], field_names: Optional[list[str]] = None) -> Dict[str, str]:
+    def parse_record(self, record: dict[str, Any], field_names: list[str] | None = None) -> dict[str, str]:
         """
         解析单条记录，动态提取指定字段
 
@@ -157,7 +157,7 @@ class FeishuTable:
         return result
 
     def loop_all(
-        self, table_view_id: str, field_names: list[str], callback: Optional[Callable[[Dict[str, str]], None]] = None, limit: Optional[int] = None
+        self, table_view_id: str, field_names: list[str], callback: Callable[[dict[str, str]], None] | None = None, limit: int | None = None
     ):
         """
         遍历视图中的所有记录
@@ -222,3 +222,65 @@ class FeishuTable:
 
         logger.info(f"总共获取到 {total_count} 条记录，成功处理 {processed_count} 条")
         logger.info("数据遍历完成")
+
+    def fetch_all(self, table_view_id: str, field_names: list[str], limit: int | None = None) -> list[dict[str, str]]:
+        """
+        抓取视图中的所有记录
+
+        Args:
+            limit: 可选的记录处理限制数量
+        """
+        logger.info("开始获取飞书多维表格数据...")
+
+        page_count = 0
+        total_count = 0
+        processed_count = 0
+        page_token = None
+
+        records: list[dict[str, str]] = []
+        while True:
+            page_count += 1
+            logger.info(f"正在获取第 {page_count} 页数据...")
+
+            try:
+                data = self.fetch_records(table_view_id, field_names, page_size=200, page_token=page_token)
+                items = data.get("items", [])
+                size = len(items)
+                total_count += size
+                page_token = data.get("page_token")
+                logger.info(f"第 {page_count} 页获取到 {size} 条记录，next page_token: {page_token}")
+
+                # 处理每条记录
+                for item in items:
+                    logger.info(f"正在处理第 {processed_count + 1} 条数据...")
+                    # 检查是否达到处理限制
+                    if limit and processed_count >= limit:
+                        logger.info(f"已达到处理限制 {limit}，停止处理")
+                        break
+
+                    try:
+                        parsed_record = self.parse_record(item)
+                        processed_count += 1
+
+                        logger.info(f"任务ID: {parsed_record['task_id']}")
+                        records.append(parsed_record)
+
+                    except Exception as e:
+                        logger.error(f"处理记录失败: {e}, 记录: {item}")
+                        continue
+
+                # 如果达到处理限制，退出外层循环
+                if limit and processed_count >= limit:
+                    break
+
+                # 检查是否还有更多数据
+                if not data.get("has_more", False) or not page_token:
+                    break
+
+            except Exception as e:
+                logger.error(f"获取第 {page_count} 页数据失败: {e}")
+                break
+
+        logger.info(f"总共获取到 {total_count} 条记录，成功处理 {processed_count} 条")
+        logger.info("数据遍历完成")
+        return records

{dts_dance-0.2.2 → dts_dance-0.2.4}/dtsdance/spacex_bytedts.py

@@ -40,6 +40,41 @@ class SpaceXClient:
             logger.warning(f"do quest queryServerMeta exception: {e}")
             raise
 
+    def register_resource(self, site: str, data_raw: 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))
+            # print(f"response status code: {response.status_code}, response text: {response.text}")
+
+            response.raise_for_status()
+
+            result = response.json()
+            return result.get("message", "") == "ok"
+
+        except Exception as e:
+            logger.warning(f"do quest registerResource exception: {e}")
+            raise
+
+    def list_resources(self, site: str, data_raw: 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))
+            # print(f"response status code: {response.status_code}, response text: {response.text}")
+
+            response.raise_for_status()
+
+            result = response.json()
+            if not result.get("message", "") == "ok":
+                return []
+
+            items = result.get("data", {}).get("items", [])
+            return [item["name"] for item in items]
+        except Exception as e:
+            logger.warning(f"do quest listResource exception: {e}")
+            raise
+
     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"
@@ -71,3 +106,16 @@ class SpaceXClient:
         except Exception as e:
             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]:
+        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))
+            # print(f"response status code: {response.status_code}, response text: {response.text}")
+
+            response.raise_for_status()
+            return response.json()
+        except Exception as e:
+            logger.warning(f"do quest gateway_operation exception: {e}")
+            raise

{dts_dance-0.2.2 → dts_dance-0.2.4}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "dts-dance"
-version = "0.2.2"
+version = "0.2.4"
 description = "dts dance lib"
 readme = "README.md"
 keywords = ["tools", "observation"]
@@ -12,8 +12,9 @@ requires-python = ">=3.12"
 dependencies = [
     "requests>=2.32.5,<3.0.0",
     "loguru>=0.7.3,<0.8.0",
-    "boto3>=1.42.29,<2.0.0",
+    "boto3>=1.42.30,<2.0.0",
     "pyyaml>=6.0.3,<7.0.0",
+    "requests_toolbelt==1.0.0",
 ]
 
 [tool.hatch.build.targets.wheel]

{dts_dance-0.2.2 → dts_dance-0.2.4}/uv.lock

@@ -4,30 +4,30 @@ requires-python = ">=3.12"
 
 [[package]]
 name = "boto3"
-version = "1.42.29"
+version = "1.42.30"
 source = { registry = "https://pypi.org/simple" }
 dependencies = [
     { name = "botocore" },
     { name = "jmespath" },
     { name = "s3transfer" },
 ]
-sdist = { url = "https://files.pythonhosted.org/packages/5c/24/1dd85b64004103c2e60476d0fa8d78435f5fed9db1129cd2cd332784037a/boto3-1.42.29.tar.gz", hash = "sha256:247e54f24116ad6792cfc14b274288383af3ec3433b0547da8a14a8bd6e81950", size = 112810, upload-time = "2026-01-15T20:36:39.404Z" }
+sdist = { url = "https://files.pythonhosted.org/packages/42/79/2dac8b7cb075cfa43908ee9af3f8ee06880d84b86013854c5cca8945afac/boto3-1.42.30.tar.gz", hash = "sha256:ba9cd2f7819637d15bfbeb63af4c567fcc8a7dcd7b93dd12734ec58601169538", size = 112809, upload-time = "2026-01-16T20:37:23.636Z" }
 wheels = [
-    { url = "https://files.pythonhosted.org/packages/51/30/2c25d7be8418e7f137ffece6097c68199dbd6996da645ec9b5a5a9647123/boto3-1.42.29-py3-none-any.whl", hash = "sha256:6c9c4dece67bf72d82ba7dff48e33a56a87cdf9b16c8887f88ca7789a95d3317", size = 140574, upload-time = "2026-01-15T20:36:37.206Z" },
+    { url = "https://files.pythonhosted.org/packages/52/b3/2c0d828c9f668292e277ca5232e6160dd5b4b660a3f076f20dd5378baa1e/boto3-1.42.30-py3-none-any.whl", hash = "sha256:d7e548bea65e0ae2c465c77de937bc686b591aee6a352d5a19a16bc751e591c1", size = 140573, upload-time = "2026-01-16T20:37:22.089Z" },
 ]
 
 [[package]]
 name = "botocore"
-version = "1.42.29"
+version = "1.42.30"
 source = { registry = "https://pypi.org/simple" }
 dependencies = [
     { name = "jmespath" },
     { name = "python-dateutil" },
     { name = "urllib3" },
 ]
-sdist = { url = "https://files.pythonhosted.org/packages/70/08/8a8e0255949845f764c5126f97b1bc09a6484077f124c2177b979ecfbbff/botocore-1.42.29.tar.gz", hash = "sha256:0fe869227a1dfe818f691a31b8c1693e39be8056a6dff5d6d4b3fc5b3a5e7d42", size = 14890916, upload-time = "2026-01-15T20:36:28.241Z" }
+sdist = { url = "https://files.pythonhosted.org/packages/44/38/23862628a0eb044c8b8b3d7a9ad1920b3bfd6bce6d746d5a871e8382c7e4/botocore-1.42.30.tar.gz", hash = "sha256:9bf1662b8273d5cc3828a49f71ca85abf4e021011c1f0a71f41a2ea5769a5116", size = 14891439, upload-time = "2026-01-16T20:37:13.77Z" }
 wheels = [
-    { url = "https://files.pythonhosted.org/packages/94/76/cfa6a934ee5a8a87f626b38275193a046da894d2f9021e001587fc2e8c7d/botocore-1.42.29-py3-none-any.whl", hash = "sha256:b45f8dfc1de5106a9d040c5612f267582e68b2b2c5237477dff85c707c1c5d11", size = 14563947, upload-time = "2026-01-15T20:36:23.828Z" },
+    { url = "https://files.pythonhosted.org/packages/3d/8d/6d7b016383b1f74dd93611b1c5078bbaddaca901553ab886dcda87cae365/botocore-1.42.30-py3-none-any.whl", hash = "sha256:97070a438cac92430bb7b65f8ebd7075224f4a289719da4ee293d22d1e98db02", size = 14566340, upload-time = "2026-01-16T20:37:10.94Z" },
 ]
 
 [[package]]
@@ -107,21 +107,23 @@ wheels = [
 
 [[package]]
 name = "dts-dance"
-version = "0.2.2"
+version = "0.2.4"
 source = { editable = "." }
 dependencies = [
     { name = "boto3" },
     { name = "loguru" },
     { name = "pyyaml" },
     { name = "requests" },
+    { name = "requests-toolbelt" },
 ]
 
 [package.metadata]
 requires-dist = [
-    { name = "boto3", specifier = ">=1.42.29,<2.0.0" },
+    { name = "boto3", specifier = ">=1.42.30,<2.0.0" },
     { name = "loguru", specifier = ">=0.7.3,<0.8.0" },
     { name = "pyyaml", specifier = ">=6.0.3,<7.0.0" },
     { name = "requests", specifier = ">=2.32.5,<3.0.0" },
+    { name = "requests-toolbelt", specifier = "==1.0.0" },
 ]
 
 [[package]]
@@ -228,6 +230,18 @@ wheels = [
     { url = "https://files.pythonhosted.org/packages/1e/db/4254e3eabe8020b458f1a747140d32277ec7a271daf1d235b70dc0b4e6e3/requests-2.32.5-py3-none-any.whl", hash = "sha256:2462f94637a34fd532264295e186976db0f5d453d1cdd31473c85a6a161affb6", size = 64738, upload-time = "2025-08-18T20:46:00.542Z" },
 ]
 
+[[package]]
+name = "requests-toolbelt"
+version = "1.0.0"
+source = { registry = "https://pypi.org/simple" }
+dependencies = [
+    { name = "requests" },
+]
+sdist = { url = "https://files.pythonhosted.org/packages/f3/61/d7545dafb7ac2230c70d38d31cbfe4cc64f7144dc41f6e4e4b78ecd9f5bb/requests-toolbelt-1.0.0.tar.gz", hash = "sha256:7681a0a3d047012b5bdc0ee37d7f8f07ebe76ab08caeccfc3921ce23c88d5bc6", size = 206888, upload-time = "2023-05-01T04:11:33.229Z" }
+wheels = [
+    { url = "https://files.pythonhosted.org/packages/3f/51/d4db610ef29373b879047326cbf6fa98b6c1969d6f6dc423279de2b1be2c/requests_toolbelt-1.0.0-py2.py3-none-any.whl", hash = "sha256:cccfdd665f0a24fcf4726e690f65639d272bb0637b9b92dfd91a5568ccf6bd06", size = 54481, upload-time = "2023-05-01T04:11:28.427Z" },
+]
+
 [[package]]
 name = "s3transfer"
 version = "0.16.0"

dts_dance-0.2.2/dtsdance/feishu_base.py

@@ -1,107 +0,0 @@
-from datetime import datetime, timedelta
-import threading
-from typing import Optional
-from loguru import logger
-import requests
-
-URL_FEISHU_OPEN_API: str = "https://fsopen.bytedance.net/open-apis"
-
-
-class FeishuBase:
-    """飞书基础类，传入机器人认证信息，支持自动token续期管理"""
-
-    def __init__(self, bot_app_id: str, bot_app_secret: str):
-        """
-        初始化飞书基础类
-        """
-        self.tenant_access_token: Optional[str] = None
-        self.token_expire_time: Optional[datetime] = None
-        self._token_lock = threading.Lock()
-
-        self.bot_app_id = bot_app_id
-        self.bot_app_secret = bot_app_secret
-
-    def _get_tenant_access_token(self) -> str:
-        """
-        获取tenant_access_token
-
-        Returns:
-            tenant_access_token字符串
-
-        Raises:
-            Exception: 获取token失败时抛出异常
-        """
-        url = f"{URL_FEISHU_OPEN_API}/auth/v3/tenant_access_token/internal"
-        payload = {"app_id": self.bot_app_id, "app_secret": self.bot_app_secret}
-        headers = {"Content-Type": "application/json"}
-
-        try:
-            response = requests.post(url, json=payload, headers=headers)
-            response.raise_for_status()
-            result = response.json()
-            if result.get("code") != 0:
-                raise Exception(f"获取tenant_access_token失败: {result.get('msg')}")
-
-            token = result.get("tenant_access_token")
-            expire_in = result.get("expire", 7200)  # 默认2小时过期
-
-            # 设置过期时间，提前5分钟刷新
-            self.token_expire_time = datetime.now() + timedelta(seconds=expire_in - 300)
-
-            logger.info(f"成功获取tenant_access_token，过期时间: {self.token_expire_time}")
-            return token
-
-        except requests.RequestException as e:
-            logger.warning(f"请求tenant_access_token失败: {e}")
-            raise Exception(f"网络请求失败: {e}")
-        except Exception as e:
-            logger.warning(f"获取tenant_access_token失败: {e}")
-            raise
-
-    def get_access_token(self) -> str:
-        """
-        获取有效的access token，自动处理续期
-
-        Returns:
-            有效的tenant_access_token
-        """
-        with self._token_lock:
-            # 检查token是否存在或已过期
-            if self.tenant_access_token is None or self.token_expire_time is None or datetime.now() >= self.token_expire_time:
-
-                logger.info("Token不存在或已过期，重新获取")
-                self.tenant_access_token = self._get_tenant_access_token()
-
-            return self.tenant_access_token
-
-    def make_request(self, method: str, endpoint: str, **kwargs) -> requests.Response:
-        """
-        发起API请求的通用方法
-
-        Args:
-            method: HTTP方法
-            endpoint: API端点
-            **kwargs: 其他请求参数
-
-        Returns:
-            Response对象
-        """
-        url = f"{URL_FEISHU_OPEN_API}{endpoint}"
-
-        # 获取有效token
-        token = self.get_access_token()
-        # logger.debug(f"使用token: {token} 发送请求: {method} {url}")
-
-        # 设置认证头
-        headers = kwargs.get("headers", {"Content-Type": "application/json"})
-        headers["Authorization"] = f"Bearer {token}"
-        kwargs["headers"] = headers
-
-        try:
-            response = requests.request(method, url, **kwargs)
-            # logger.debug(f"response_json: {response.json()}")
-            response.raise_for_status()
-            return response
-        except requests.RequestException as e:
-            logger.warning(f"API请求失败: {method} {url}, error: {e}")
-            raise