111 lines
3.9 KiB
Python
111 lines
3.9 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
恢复中断的小说生成任务
|
|
"""
|
|
|
|
import os
|
|
import sys
|
|
import json
|
|
from pathlib import Path
|
|
|
|
def find_unfinished_projects():
|
|
"""查找未完成的项目"""
|
|
base_dir = Path(__file__).parent
|
|
unfinished = []
|
|
|
|
for item in base_dir.iterdir():
|
|
if item.is_dir() and item.name.startswith('n') and item.name[1:].isdigit():
|
|
progress_file = item / ".generation_progress.json"
|
|
if progress_file.exists():
|
|
try:
|
|
with open(progress_file, 'r', encoding='utf-8') as f:
|
|
progress = json.load(f)
|
|
|
|
if progress.get('stage') != 'complete':
|
|
unfinished.append({
|
|
'dir': item.name,
|
|
'stage': progress.get('stage', 'unknown'),
|
|
'progress': progress.get('progress_percent', 0),
|
|
'last_update': progress.get('last_update', 'unknown')
|
|
})
|
|
except:
|
|
continue
|
|
|
|
return sorted(unfinished, key=lambda x: x['last_update'], reverse=True)
|
|
|
|
def main():
|
|
print("🔍 查找未完成的小说生成任务...")
|
|
|
|
unfinished = find_unfinished_projects()
|
|
|
|
if not unfinished:
|
|
print("✅ 没有找到未完成的任务")
|
|
print("💡 运行 python ViralNovel.py 开始新的生成")
|
|
return
|
|
|
|
print(f"\n📋 找到 {len(unfinished)} 个未完成的任务:")
|
|
print("-" * 60)
|
|
|
|
for i, proj in enumerate(unfinished, 1):
|
|
print(f"{i}. 项目: {proj['dir']}")
|
|
print(f" 阶段: {proj['stage']}")
|
|
print(f" 进度: {proj['progress']:.1f}%")
|
|
print(f" 更新: {proj['last_update']}")
|
|
print()
|
|
|
|
if len(unfinished) == 1:
|
|
choice = input("是否继续这个任务？(y/n): ").strip().lower()
|
|
if choice == 'y':
|
|
selected = unfinished[0]
|
|
else:
|
|
print("❌ 已取消")
|
|
return
|
|
else:
|
|
try:
|
|
choice = input(f"选择要继续的任务 (1-{len(unfinished)}): ").strip()
|
|
idx = int(choice) - 1
|
|
if 0 <= idx < len(unfinished):
|
|
selected = unfinished[idx]
|
|
else:
|
|
print("❌ 无效选择")
|
|
return
|
|
except ValueError:
|
|
print("❌ 请输入数字")
|
|
return
|
|
|
|
print(f"\n🚀 继续生成任务: {selected['dir']}")
|
|
print(f"📊 当前进度: {selected['stage']} ({selected['progress']:.1f}%)")
|
|
print("⏳ 正在恢复...")
|
|
|
|
try:
|
|
# 导入并运行恢复函数
|
|
sys.path.append(str(Path(__file__).parent))
|
|
from ViralNovel import resume_generation_from_progress
|
|
|
|
result = resume_generation_from_progress(selected['dir'])
|
|
|
|
print(f"\n✅ 生成完成！")
|
|
print(f"📖 标题: {result['title']}")
|
|
print(f"📁 保存位置: {result['project_dir']}")
|
|
|
|
except ImportError:
|
|
print("❌ 无法导入 ViralNovel 模块")
|
|
print("💡 请确保 ViralNovel.py 文件存在")
|
|
|
|
except Exception as e:
|
|
error_str = str(e)
|
|
print(f"\n❌ 恢复失败: {error_str}")
|
|
|
|
if "网络连接" in error_str or "超时" in error_str:
|
|
print("\n🔧 网络问题解决建议:")
|
|
print("1. 检查网络连接")
|
|
print("2. 等待几分钟后重试")
|
|
print("3. 运行: python resume_generation.py")
|
|
|
|
elif "rate limit" in error_str.lower():
|
|
print("\n⏰ API调用频率限制:")
|
|
print("1. 等待几分钟后重试")
|
|
print("2. 运行: python resume_generation.py")
|
|
|
|
if __name__ == "__main__":
|
|
main() |