wehub-resource-sync
201 lines
6.6 KiB
Python
201 lines
6.6 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
修复项目缩略图脚本
|
|
处理链接导入和文件导入项目的缩略图问题
|
|
"""
|
|
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
# 添加项目根目录到Python路径
|
|
project_root = Path(__file__).parent.parent.parent
|
|
if str(project_root) not in sys.path:
|
|
sys.path.insert(0, str(project_root))
|
|
|
|
from backend.core.database import SessionLocal
|
|
from backend.models.project import Project
|
|
from backend.utils.thumbnail_generator import generate_project_thumbnail
|
|
import requests
|
|
import base64
|
|
import logging
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
def fix_project_thumbnail(project_id: str):
|
|
"""修复指定项目的缩略图"""
|
|
db = SessionLocal()
|
|
try:
|
|
project = db.query(Project).filter(Project.id == project_id).first()
|
|
|
|
if not project:
|
|
print(f"❌ 项目 {project_id} 不存在")
|
|
return False
|
|
|
|
if project.thumbnail:
|
|
print(f"✅ 项目 {project_id} 已有缩略图，跳过")
|
|
return True
|
|
|
|
print(f"🔧 修复项目 {project_id} 的缩略图...")
|
|
|
|
# 检查项目类型和来源
|
|
source_url = project.project_metadata.get('source_url') if project.project_metadata else None
|
|
is_bilibili_project = source_url and 'bilibili.com' in source_url
|
|
has_video_file = project.video_path and Path(project.video_path).exists()
|
|
|
|
if is_bilibili_project:
|
|
# 链接导入项目 - 尝试从B站获取缩略图
|
|
print(f"📺 检测到B站项目，尝试获取原视频缩略图...")
|
|
success = fix_bilibili_thumbnail(project, db)
|
|
elif has_video_file:
|
|
# 文件导入项目 - 从视频文件生成缩略图
|
|
print(f"📁 检测到文件导入项目，从视频文件生成缩略图...")
|
|
success = fix_file_import_thumbnail(project, db)
|
|
else:
|
|
# 没有原始视频文件，尝试从切片生成缩略图
|
|
print(f"🎬 没有原始视频文件，尝试从切片生成缩略图...")
|
|
success = fix_clip_thumbnail(project, db)
|
|
|
|
if success:
|
|
print(f"✅ 项目 {project_id} 缩略图修复成功")
|
|
else:
|
|
print(f"❌ 项目 {project_id} 缩略图修复失败")
|
|
|
|
return success
|
|
|
|
except Exception as e:
|
|
print(f"❌ 修复项目 {project_id} 缩略图时发生错误: {e}")
|
|
return False
|
|
finally:
|
|
db.close()
|
|
|
|
def fix_bilibili_thumbnail(project, db):
|
|
"""修复B站项目的缩略图"""
|
|
try:
|
|
# 从项目设置中获取B站信息
|
|
if not project.processing_config:
|
|
return False
|
|
|
|
bilibili_info = project.processing_config.get('bilibili_info', {})
|
|
if not bilibili_info:
|
|
return False
|
|
|
|
# 尝试从B站API获取缩略图
|
|
# 这里需要根据实际的B站API来实现
|
|
# 暂时返回False，表示需要手动处理
|
|
print("⚠️ B站缩略图获取需要API支持，暂时跳过")
|
|
return False
|
|
|
|
except Exception as e:
|
|
logger.error(f"修复B站缩略图失败: {e}")
|
|
return False
|
|
|
|
def fix_file_import_thumbnail(project, db):
|
|
"""修复文件导入项目的缩略图"""
|
|
try:
|
|
video_path = Path(project.video_path)
|
|
if not video_path.exists():
|
|
print(f"⚠️ 视频文件不存在: {video_path}")
|
|
return False
|
|
|
|
# 生成缩略图
|
|
thumbnail_data = generate_project_thumbnail(project.id, video_path)
|
|
|
|
if thumbnail_data:
|
|
# 保存到数据库
|
|
project.thumbnail = thumbnail_data
|
|
db.commit()
|
|
return True
|
|
else:
|
|
print("⚠️ 缩略图生成失败")
|
|
return False
|
|
|
|
except Exception as e:
|
|
logger.error(f"修复文件导入缩略图失败: {e}")
|
|
return False
|
|
|
|
def fix_clip_thumbnail(project, db):
|
|
"""从切片生成缩略图"""
|
|
try:
|
|
# 查找项目目录中的切片文件
|
|
project_dir = Path(f"/Users/zhoukk/autoclip/data/projects/{project.id}")
|
|
clips_dir = project_dir / "output" / "clips"
|
|
|
|
if not clips_dir.exists():
|
|
print(f"⚠️ 切片目录不存在: {clips_dir}")
|
|
return False
|
|
|
|
# 获取第一个切片文件
|
|
clip_files = list(clips_dir.glob("*.mp4"))
|
|
if not clip_files:
|
|
print(f"⚠️ 没有找到切片文件")
|
|
return False
|
|
|
|
first_clip = clip_files[0]
|
|
print(f"🎬 使用切片文件生成缩略图: {first_clip.name}")
|
|
|
|
# 生成缩略图
|
|
thumbnail_data = generate_project_thumbnail(project.id, first_clip)
|
|
|
|
if thumbnail_data:
|
|
# 保存到数据库
|
|
project.thumbnail = thumbnail_data
|
|
db.commit()
|
|
return True
|
|
else:
|
|
print("⚠️ 从切片生成缩略图失败")
|
|
return False
|
|
|
|
except Exception as e:
|
|
logger.error(f"从切片生成缩略图失败: {e}")
|
|
return False
|
|
|
|
def fix_all_project_thumbnails():
|
|
"""修复所有项目的缩略图"""
|
|
db = SessionLocal()
|
|
try:
|
|
# 查找所有没有缩略图的项目
|
|
projects = db.query(Project).filter(Project.thumbnail.is_(None)).all()
|
|
|
|
if not projects:
|
|
print("✅ 所有项目都已有缩略图")
|
|
return True
|
|
|
|
print(f"📋 找到 {len(projects)} 个需要修复缩略图的项目")
|
|
|
|
success_count = 0
|
|
for project in projects:
|
|
if fix_project_thumbnail(project.id):
|
|
success_count += 1
|
|
|
|
print(f"🎉 完成！成功修复 {success_count}/{len(projects)} 个项目的缩略图")
|
|
return True
|
|
|
|
except Exception as e:
|
|
print(f"❌ 修复所有项目缩略图时发生错误: {e}")
|
|
return False
|
|
finally:
|
|
db.close()
|
|
|
|
def main():
|
|
"""主函数"""
|
|
if len(sys.argv) > 1:
|
|
# 修复指定项目
|
|
project_id = sys.argv[1]
|
|
print(f"🚀 开始修复项目 {project_id} 的缩略图...")
|
|
if fix_project_thumbnail(project_id):
|
|
print("🎉 缩略图修复完成！")
|
|
else:
|
|
print("❌ 缩略图修复失败")
|
|
sys.exit(1)
|
|
else:
|
|
# 修复所有项目
|
|
print("🚀 开始修复所有项目的缩略图...")
|
|
if fix_all_project_thumbnails():
|
|
print("🎉 所有缩略图修复完成！")
|
|
else:
|
|
print("❌ 缩略图修复失败")
|
|
sys.exit(1)
|
|
|
|
if __name__ == "__main__":
|
|
main()
|