Function main

agent/batch_manager.py:198–266

主函数

()

                    print(f"     - {rel_type}: {count}")

def main():
    """主函数"""
    parser = argparse.ArgumentParser(
        description='分批处理管理器 - 管理菜谱处理的分批保存和断点续传',
        formatter_class=argparse.RawDescriptionHelpFormatter,
        epilog="""
使用示例:
  python batch_manager.py status                    # 查看处理状态
  python batch_manager.py continue ./HowToCook-master  # 继续中断的处理
  python batch_manager.py merge                     # 合并批次数据
  python batch_manager.py clean-progress            # 清理进度文件
  python batch_manager.py clean-batches             # 清理批次数据
  python batch_manager.py details                   # 显示所有批次详情
  python batch_manager.py details -b 1              # 显示指定批次详情
        """)
    
    subparsers = parser.add_subparsers(dest='command', help='可用命令')
    
    # status命令
    subparsers.add_parser('status', help='查看处理进度状态')
    
    # continue命令
    continue_parser = subparsers.add_parser('continue', help='继续中断的处理')
    continue_parser.add_argument('recipe_dir', help='菜谱目录路径')
    
    # merge命令
    subparsers.add_parser('merge', help='合并所有批次数据')
    
    # clean命令
    subparsers.add_parser('clean-progress', help='清理进度文件')
    subparsers.add_parser('clean-batches', help='清理所有批次数据')
    
    # details命令
    details_parser = subparsers.add_parser('details', help='显示批次详细信息')
    details_parser.add_argument('-b', '--batch', type=int, help='指定批次编号')
    
    # 全局参数
    parser.add_argument('-o', '--output', default='./ai_output', help='输出目录路径')
    
    args = parser.parse_args()
    
    if not args.command:
        parser.print_help()
        return
    
    # 确保输出目录存在
    os.makedirs(args.output, exist_ok=True)
    
    print("🛠️  分批处理管理器")
    print("=" * 50)
    
    try:
        if args.command == 'status':
            show_progress_status(args.output)
        elif args.command == 'continue':
            continue_processing(args.recipe_dir, args.output)
        elif args.command == 'merge':
            merge_batches(args.output)