ADD 增加视频帧位计算、字符串判空节点

PERF 重构项目结构

nodes/vod2local.py

@@ -0,0 +1,112 @@
+import os
+from pathlib import Path
+
+import requests
+from tencentcloud.common import credential
+from tencentcloud.common.profile.client_profile import ClientProfile
+from tencentcloud.common.profile.http_profile import HttpProfile
+from tencentcloud.vod.v20180717 import vod_client, models
+
+
+class VodToLocalNode:
+
+    def __init__(self):
+        self.secret_id = "AKIDsrihIyjZOBsjimt8TsN8yvv1AMh5dB44"
+        self.secret_key = "CPZcxdk6W39Jd4cGY95wvupoyMd0YFqW"
+        self.vod_client = self.init_vod_client()
+
+    def init_vod_client(self):
+        """初始化VOD客户端"""
+        try:
+            http_profile = HttpProfile(endpoint="vod.tencentcloudapi.com")
+            client_profile = ClientProfile(httpProfile=http_profile)
+            cred = credential.Credential(self.secret_id, self.secret_key)
+            return vod_client.VodClient(cred, "ap-shanghai", client_profile)
+        except Exception as e:
+            raise RuntimeError(f"VOD client initialization failed: {e}")
+
+    @classmethod
+    def INPUT_TYPES(cls):
+        return {
+            "required": {
+                "file_id": ("STRING", {"default": ""}),
+                "sub_app_id": ("STRING", {"default": ""}),
+            }
+        }
+
+    RETURN_TYPES = ("STRING",)
+    RETURN_NAMES = ("local_path",)
+    FUNCTION = "execute"
+    CATEGORY = "video"
+
+    def execute(self, file_id, sub_app_id):
+        # 调用下载逻辑
+        local_path = self.download_vod(file_id, sub_app_id)
+        print(f"下载成功: {local_path}")
+        return (local_path,)
+
+    def _get_download_url(self, file_id, sub_app_id):
+        """获取媒体文件下载地址"""
+        try:
+            req = models.DescribeMediaInfosRequest()
+            req.FileIds = [file_id]
+            req.SubAppId = int(sub_app_id)
+
+            resp = self.vod_client.DescribeMediaInfos(req)
+            if not resp.MediaInfoSet:
+                raise ValueError("File not found")
+
+            media_info = resp.MediaInfoSet[0]
+            if not media_info.BasicInfo.MediaUrl:
+                raise ValueError("No download URL available")
+
+            return media_info.BasicInfo.MediaUrl
+        except Exception as e:
+            raise RuntimeError(f"Tencent API error: {e}")
+
+    def create_directory(self, path):
+        p = Path(path)
+        if not p.exists():
+            p.mkdir(
+                parents=True, exist_ok=True
+            )  # parents=True会自动创建所有必需的父目录，exist_ok=True表示如果目录已存在则不会引发异常
+            print(f"目录已创建: {path}")
+        else:
+            print(f"目录已存在: {path}")
+
+    def download_vod(self, file_id, sub_app_id):
+        """
+        需要补充腾讯云VOD SDK调用逻辑
+        返回本地文件路径
+        """
+        media_url = self._get_download_url(file_id=file_id, sub_app_id=sub_app_id)
+        print(f"download from url: {media_url}")
+        # 生成一个临时目录路径名并创建该目录
+        self.create_directory(
+            os.path.join(
+                os.path.dirname(os.path.abspath(__file__)), "download", f"{sub_app_id}"
+            )
+        )
+        output_dir = os.path.join(
+            os.path.dirname(os.path.abspath(__file__)),
+            "download",
+            f"{sub_app_id}",
+            f"{file_id}.mp4",
+        )
+        # 判断文件是否存在
+        if os.path.exists(output_dir):
+            return output_dir
+        return self._download_file(url=media_url, save_path=output_dir, timeout=60 * 10)
+
+    def _download_file(self, url: str, save_path: str, timeout: int = 30):
+        """下载文件到本地"""
+        try:
+            with requests.get(url, stream=True, timeout=timeout) as response:
+                response.raise_for_status()
+                with open(save_path, "wb") as f:
+                    for chunk in response.iter_content(chunk_size=8192):
+                        if chunk:
+                            f.write(chunk)
+            return save_path
+        except Exception as e:
+            raise RuntimeError(f"Download error: {e}")