zhukang
1c7abd81d1
- 310 registry:skill items, 1528 files, 7 categories (hub tags) - registry.json generated by @kortix/registry buildRegistry - nested SKILL.md in sn-da-* renamed SUBSKILL.md (no phantom skills) - provenance + licenses: SOURCES.md
326 lines
7.9 KiB
Python
326 lines
7.9 KiB
Python
#!/usr/bin/env python3
|
||
"""
|
||
通用工具函数
|
||
提供时间格式转换、文件名清理、路径处理等功能
|
||
"""
|
||
|
||
import re
|
||
import os
|
||
from pathlib import Path
|
||
from datetime import datetime
|
||
|
||
|
||
def time_to_seconds(time_str: str) -> float:
|
||
"""
|
||
将时间字符串转换为秒数
|
||
|
||
支持格式:
|
||
- HH:MM:SS.mmm
|
||
- MM:SS.mmm
|
||
- SS.mmm
|
||
|
||
Args:
|
||
time_str: 时间字符串
|
||
|
||
Returns:
|
||
float: 秒数
|
||
|
||
Examples:
|
||
>>> time_to_seconds("01:23:45.678")
|
||
5025.678
|
||
>>> time_to_seconds("23:45.678")
|
||
1425.678
|
||
>>> time_to_seconds("45.678")
|
||
45.678
|
||
"""
|
||
# 移除空格
|
||
time_str = time_str.strip()
|
||
|
||
# 分割时间部分
|
||
parts = time_str.split(':')
|
||
|
||
if len(parts) == 3:
|
||
# HH:MM:SS.mmm
|
||
hours, minutes, seconds = parts
|
||
return int(hours) * 3600 + int(minutes) * 60 + float(seconds)
|
||
elif len(parts) == 2:
|
||
# MM:SS.mmm
|
||
minutes, seconds = parts
|
||
return int(minutes) * 60 + float(seconds)
|
||
else:
|
||
# SS.mmm
|
||
return float(parts[0])
|
||
|
||
|
||
def seconds_to_time(seconds: float, include_hours: bool = True, use_comma: bool = False) -> str:
|
||
"""
|
||
将秒数转换为时间字符串
|
||
|
||
Args:
|
||
seconds: 秒数
|
||
include_hours: 是否包含小时（即使为0）
|
||
use_comma: 使用逗号分隔毫秒（SRT 格式需要）
|
||
|
||
Returns:
|
||
str: 时间字符串
|
||
|
||
Examples:
|
||
>>> seconds_to_time(5025.678)
|
||
'01:23:45.678'
|
||
>>> seconds_to_time(1425.678, include_hours=False)
|
||
'23:45.678'
|
||
>>> seconds_to_time(5025.678, use_comma=True)
|
||
'01:23:45,678'
|
||
"""
|
||
hours = int(seconds // 3600)
|
||
minutes = int((seconds % 3600) // 60)
|
||
secs = seconds % 60
|
||
|
||
separator = ',' if use_comma else '.'
|
||
|
||
if include_hours or hours > 0:
|
||
return f"{hours:02d}:{minutes:02d}:{secs:06.3f}".replace('.', separator)
|
||
else:
|
||
return f"{minutes:02d}:{secs:06.3f}".replace('.', separator)
|
||
|
||
|
||
def sanitize_filename(filename: str, max_length: int = 100) -> str:
|
||
"""
|
||
清理文件名，移除非法字符
|
||
|
||
Args:
|
||
filename: 原始文件名
|
||
max_length: 最大长度
|
||
|
||
Returns:
|
||
str: 清理后的文件名
|
||
|
||
Examples:
|
||
>>> sanitize_filename("Hello: World?")
|
||
'Hello_World'
|
||
>>> sanitize_filename("AGI 不是时间点，是指数曲线")
|
||
'AGI_不是时间点_是指数曲线'
|
||
"""
|
||
# 移除或替换非法字符
|
||
# Windows/Mac/Linux 通用的非法字符
|
||
illegal_chars = r'[<>:"/\\|?*]'
|
||
filename = re.sub(illegal_chars, '_', filename)
|
||
|
||
# 移除开头和结尾的空格和点
|
||
filename = filename.strip('. ')
|
||
|
||
# 替换空格为下划线
|
||
filename = filename.replace(' ', '_')
|
||
|
||
# 移除连续的下划线
|
||
filename = re.sub(r'_+', '_', filename)
|
||
|
||
# 限制长度
|
||
if len(filename) > max_length:
|
||
# 保留扩展名
|
||
name, ext = os.path.splitext(filename)
|
||
if ext:
|
||
max_name_length = max_length - len(ext)
|
||
filename = name[:max_name_length] + ext
|
||
else:
|
||
filename = filename[:max_length]
|
||
|
||
return filename
|
||
|
||
|
||
def create_output_dir(base_dir: str = None) -> Path:
|
||
"""
|
||
创建输出目录（带时间戳）
|
||
|
||
Args:
|
||
base_dir: 基础目录，默认为当前工作目录下的 youtube-clips
|
||
|
||
Returns:
|
||
Path: 输出目录路径
|
||
|
||
Examples:
|
||
>>> create_output_dir()
|
||
PosixPath('/path/to/current/dir/youtube-clips/20260121_143022')
|
||
"""
|
||
if base_dir is None:
|
||
base_dir = Path.cwd() / "youtube-clips"
|
||
else:
|
||
base_dir = Path(base_dir)
|
||
|
||
# 创建带时间戳的子目录
|
||
timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
|
||
output_dir = base_dir / timestamp
|
||
|
||
# 创建目录
|
||
output_dir.mkdir(parents=True, exist_ok=True)
|
||
|
||
return output_dir
|
||
|
||
|
||
def format_file_size(size_bytes: int) -> str:
|
||
"""
|
||
格式化文件大小为人类可读格式
|
||
|
||
Args:
|
||
size_bytes: 文件大小（字节）
|
||
|
||
Returns:
|
||
str: 格式化后的大小
|
||
|
||
Examples:
|
||
>>> format_file_size(1024)
|
||
'1.0 KB'
|
||
>>> format_file_size(1536)
|
||
'1.5 KB'
|
||
>>> format_file_size(1048576)
|
||
'1.0 MB'
|
||
"""
|
||
for unit in ['B', 'KB', 'MB', 'GB', 'TB']:
|
||
if size_bytes < 1024.0:
|
||
return f"{size_bytes:.1f} {unit}"
|
||
size_bytes /= 1024.0
|
||
return f"{size_bytes:.1f} PB"
|
||
|
||
|
||
def parse_time_range(time_range: str) -> tuple:
|
||
"""
|
||
解析时间范围字符串
|
||
|
||
Args:
|
||
time_range: 时间范围字符串，如 "00:00 - 03:15" 或 "00:00-03:15"
|
||
|
||
Returns:
|
||
tuple: (start_seconds, end_seconds)
|
||
|
||
Examples:
|
||
>>> parse_time_range("00:00 - 03:15")
|
||
(0.0, 195.0)
|
||
>>> parse_time_range("01:30:00-01:33:15")
|
||
(5400.0, 5595.0)
|
||
"""
|
||
# 移除空格并分割
|
||
parts = time_range.replace(' ', '').split('-')
|
||
if len(parts) != 2:
|
||
raise ValueError(f"Invalid time range format: {time_range}")
|
||
|
||
start_time = time_to_seconds(parts[0])
|
||
end_time = time_to_seconds(parts[1])
|
||
|
||
if start_time >= end_time:
|
||
raise ValueError(f"Start time must be before end time: {time_range}")
|
||
|
||
return start_time, end_time
|
||
|
||
|
||
def adjust_subtitle_time(time_seconds: float, offset: float) -> float:
|
||
"""
|
||
调整字幕时间戳（用于剪辑后的字幕）
|
||
|
||
Args:
|
||
time_seconds: 原始时间（秒）
|
||
offset: 偏移量（秒），通常是剪辑起始时间
|
||
|
||
Returns:
|
||
float: 调整后的时间
|
||
|
||
Examples:
|
||
>>> adjust_subtitle_time(125.5, 120.0)
|
||
5.5
|
||
"""
|
||
adjusted = time_seconds - offset
|
||
return max(0.0, adjusted) # 确保不为负数
|
||
|
||
|
||
def get_video_duration_display(seconds: float) -> str:
|
||
"""
|
||
获取视频时长的显示格式
|
||
|
||
Args:
|
||
seconds: 时长（秒）
|
||
|
||
Returns:
|
||
str: 格式化的时长
|
||
|
||
Examples:
|
||
>>> get_video_duration_display(125.5)
|
||
'02:05'
|
||
>>> get_video_duration_display(3725.5)
|
||
'1:02:05'
|
||
"""
|
||
hours = int(seconds // 3600)
|
||
minutes = int((seconds % 3600) // 60)
|
||
secs = int(seconds % 60)
|
||
|
||
if hours > 0:
|
||
return f"{hours}:{minutes:02d}:{secs:02d}"
|
||
else:
|
||
return f"{minutes:02d}:{secs:02d}"
|
||
|
||
|
||
def validate_url(url: str) -> bool:
|
||
"""
|
||
验证 YouTube URL 格式
|
||
|
||
Args:
|
||
url: YouTube URL
|
||
|
||
Returns:
|
||
bool: 是否有效
|
||
|
||
Examples:
|
||
>>> validate_url("https://youtube.com/watch?v=Ckt1cj0xjRM")
|
||
True
|
||
>>> validate_url("https://youtu.be/Ckt1cj0xjRM")
|
||
True
|
||
>>> validate_url("invalid_url")
|
||
False
|
||
"""
|
||
# 支持的 YouTube URL 格式
|
||
patterns = [
|
||
r'https?://(?:www\.)?youtube\.com/watch\?v=[\w-]+',
|
||
r'https?://(?:www\.)?youtu\.be/[\w-]+',
|
||
r'https?://(?:www\.)?youtube\.com/embed/[\w-]+',
|
||
]
|
||
|
||
return any(re.match(pattern, url) for pattern in patterns)
|
||
|
||
|
||
def ensure_directory(path: Path) -> Path:
|
||
"""
|
||
确保目录存在，不存在则创建
|
||
|
||
Args:
|
||
path: 目录路径
|
||
|
||
Returns:
|
||
Path: 目录路径
|
||
"""
|
||
path = Path(path)
|
||
path.mkdir(parents=True, exist_ok=True)
|
||
return path
|
||
|
||
|
||
if __name__ == "__main__":
|
||
# 测试代码
|
||
print("Testing utils.py...")
|
||
|
||
# 测试时间转换
|
||
assert time_to_seconds("01:23:45.678") == 5025.678
|
||
assert time_to_seconds("23:45.678") == 1425.678
|
||
assert time_to_seconds("45.678") == 45.678
|
||
|
||
# 测试文件名清理
|
||
assert sanitize_filename("Hello: World?") == "Hello_World"
|
||
assert sanitize_filename("AGI 不是时间点，是指数曲线") == "AGI_不是时间点_是指数曲线"
|
||
|
||
# 测试时间范围解析
|
||
assert parse_time_range("00:00 - 03:15") == (0.0, 195.0)
|
||
assert parse_time_range("01:30:00-01:33:15") == (5400.0, 5595.0)
|
||
|
||
# 测试 URL 验证
|
||
assert validate_url("https://youtube.com/watch?v=Ckt1cj0xjRM") == True
|
||
assert validate_url("https://youtu.be/Ckt1cj0xjRM") == True
|
||
assert validate_url("invalid_url") == False
|
||
|
||
print("✅ All tests passed!")
|