Initial commit: lesson-highlights generator

src/core/utils.py

@@ -0,0 +1,112 @@
+# -*- coding: utf-8 -*-
+"""
+通用工具函数
+
+提供跨模块使用的通用功能
+"""
+
+import os
+import re
+import subprocess
+import logging
+
+logger = logging.getLogger(__name__)
+
+
+def ensure_dir(path):
+    """确保目录存在"""
+    os.makedirs(path, exist_ok=True)
+    return path
+
+
+def get_clip_num(path):
+    """从路径中提取clip编号"""
+    match = re.search(r'clip(\d+)', path)
+    return int(match.group(1)) if match else 0
+
+
+def run_cmd(cmd, capture=True, timeout=300):
+    """
+    执行shell命令
+
+    Args:
+        cmd: 命令字符串
+        capture: 是否捕获输出
+        timeout: 超时时间(秒)
+
+    Returns:
+        True if success, False otherwise
+    """
+    try:
+        if capture:
+            result = subprocess.run(
+                cmd,
+                shell=True,
+                capture_output=True,
+                text=True,
+                encoding='utf-8',
+                errors='replace',
+                timeout=timeout
+            )
+            return result.returncode == 0
+        else:
+            result = subprocess.run(cmd, shell=True, timeout=timeout)
+            return result.returncode == 0
+    except subprocess.TimeoutExpired:
+        logger.error(f"Command timeout: {cmd[:100]}...")
+        return False
+    except Exception as e:
+        logger.error(f"Command failed: {e}")
+        return False
+
+
+def to_srt_time(t):
+    """将秒数转换为SRT时间格式 (HH:MM:SS,mmm)"""
+    hours = int(t // 3600)
+    minutes = int((t % 3600) // 60)
+    seconds = int(t % 60)
+    millis = int((t % 1) * 1000)
+    return f"{hours:02d}:{minutes:02d}:{seconds:02d},{millis:03d}"
+
+
+def to_ass_time(t):
+    """将秒数转换为ASS时间格式 (HH:MM:SS.cc)"""
+    hours = int(t // 3600)
+    minutes = int((t % 3600) // 60)
+    seconds = int(t % 60)
+    centis = int((t % 1) * 100)
+    return f"{hours:02d}:{minutes:02d}:{seconds:02d}.{centis:02d}"
+
+
+def format_duration(seconds):
+    """格式化时长为可读字符串"""
+    if seconds < 60:
+        return f"{seconds:.1f}秒"
+    elif seconds < 3600:
+        return f"{seconds/60:.1f}分钟"
+    else:
+        return f"{seconds/3600:.1f}小时"
+
+
+def format_filesize(bytes_size):
+    """格式化文件大小为可读字符串"""
+    mb = bytes_size / (1024 * 1024)
+    if mb < 1024:
+        return f"{mb:.1f} MB"
+    else:
+        return f"{mb/1024:.1f} GB"
+
+
+class SubtitleError(Exception):
+    """字幕处理异常"""
+    pass
+
+
+class VideoError(Exception):
+    """视频处理异常"""
+    pass
+
+
+class LLMError(Exception):
+    """LLM调用异常"""
+    pass