import argparse from datetime import datetime from system_management.scheduler.task_scheduler import TaskScheduler from system_management.scheduler.task_scheduler import TaskManager from config import Config from utils.logger import CrossPlatformLog # 初始化日志 log = CrossPlatformLog.get_logger("TaskManagement") def main(): # 初始化配置和组件 scheduler = TaskScheduler(Config.MYSQL_CONFIG) manager = TaskManager(scheduler) # 解析命令行参数 parser = argparse.ArgumentParser(description="任务管理工具") subparsers = parser.add_subparsers(dest="command", help="可用命令") # 列出任务 list_parser = subparsers.add_parser("list", help="列出所有任务") list_parser.add_argument("--active-only", action="store_true", help="只显示活跃任务") # 查看任务详情 show_parser = subparsers.add_parser("show", help="显示任务详情") show_parser.add_argument("task_id", type=int, help="任务ID") # 更新任务 update_parser = subparsers.add_parser("update", help="更新任务属性") update_parser.add_argument("task_id", type=int, help="任务ID") update_parser.add_argument("--name", help="任务名称") update_parser.add_argument("--type", help="任务类型") update_parser.add_argument("--module", help="模块路径") update_parser.add_argument("--cron", help="Cron表达式") update_parser.add_argument("--timezone", help="时区") # 启用/禁用任务 toggle_parser = subparsers.add_parser("toggle", help="启用/禁用任务") toggle_parser.add_argument("task_id", type=int, help="任务ID") toggle_parser.add_argument("--activate", action="store_true", help="启用任务") toggle_parser.add_argument("--deactivate", action="store_true", help="禁用任务") # 删除任务 delete_parser = subparsers.add_parser("delete", help="删除任务") delete_parser.add_argument("task_id", type=int, help="任务ID") # 手动执行任务 run_parser = subparsers.add_parser("run", help="手动执行任务") run_parser.add_argument("task_id", type=int, help="任务ID") # 添加任务 add_parser = subparsers.add_parser("add", help="添加新任务") add_parser.add_argument("--name", required=True, help="任务名称") add_parser.add_argument("--type", required=True, help="任务类型") add_parser.add_argument("--module", required=True, help="模块路径") add_parser.add_argument("--cron", required=True, help="Cron表达式") add_parser.add_argument("--timezone", default="Asia/Shanghai", help="时区") args = parser.parse_args() # 执行相应命令 if args.command == "list": try: tasks = manager.get_all_tasks(args.active_only) manager.print_task_table(tasks) log.info(f"列出任务完成，共{len(tasks)}个任务") except Exception as e: log.error(f"列出任务失败: {str(e)}", exc_info=True) elif args.command == "show": try: task = manager.get_task_by_id(args.task_id) if task: print("\n===== 任务详情 =====") for key, value in task.items(): print(f"{key}: {value}") print("====================") log.info(f"显示任务详情成功，任务ID: {args.task_id}") else: log.warning(f"未找到任务ID: {args.task_id}") print(f"任务ID {args.task_id} 不存在") except Exception as e: log.error(f"显示任务详情失败，任务ID: {args.task_id}", exc_info=True) elif args.command == "update": try: updates = {} if args.name: updates['task_name'] = args.name if args.type: updates['task_type'] = args.type if args.module: updates['module_path'] = args.module if args.cron: updates['cron_expression'] = args.cron if args.timezone: updates['time_zone'] = args.timezone if not updates: log.warning("未提供任何更新字段") print("请至少指定一个更新字段") return if manager.update_task(args.task_id, updates): log.info(f"任务ID {args.task_id} 更新成功") print(f"任务ID {args.task_id} 更新成功") else: log.warning(f"任务ID {args.task_id} 更新失败") print(f"任务ID {args.task_id} 更新失败") except Exception as e: log.error(f"更新任务失败，任务ID: {args.task_id}", exc_info=True) elif args.command == "toggle": try: if args.activate and args.deactivate: log.warning("不能同时指定 --activate 和 --deactivate") print("不能同时指定 --activate 和 --deactivate") return if not args.activate and not args.deactivate: log.warning("请指定 --activate 或 --deactivate") print("请指定 --activate 或 --deactivate") return if args.activate: success = manager.toggle_task_status(args.task_id, True) action = "启用" else: success = manager.toggle_task_status(args.task_id, False) action = "禁用" if success: log.info(f"任务ID {args.task_id} {action}成功") print(f"任务ID {args.task_id} {action}成功") else: log.warning(f"任务ID {args.task_id} {action}失败") print(f"任务ID {args.task_id} {action}失败") except Exception as e: log.error(f"切换任务状态失败，任务ID: {args.task_id}", exc_info=True) elif args.command == "delete": try: confirm = input(f"确定要删除任务ID {args.task_id} 吗? (y/n) ") if confirm.lower() == 'y': if manager.delete_task(args.task_id): log.info(f"任务ID {args.task_id} 删除成功") print(f"任务ID {args.task_id} 删除成功") else: log.warning(f"任务ID {args.task_id} 删除失败") print(f"任务ID {args.task_id} 删除失败") else: log.info(f"用户取消删除任务ID {args.task_id}") print("操作已取消") except Exception as e: log.error(f"删除任务失败，任务ID: {args.task_id}", exc_info=True) elif args.command == "run": try: log.info(f"开始手动执行任务ID {args.task_id}") print(f"正在手动执行任务ID {args.task_id}...") if manager.run_task_manually(args.task_id): log.info(f"任务ID {args.task_id} 执行成功") print(f"任务ID {args.task_id} 执行成功") else: log.warning(f"任务ID {args.task_id} 执行失败") print(f"任务ID {args.task_id} 执行失败") except Exception as e: log.error(f"手动执行任务失败，任务ID: {args.task_id}", exc_info=True) elif args.command == "add": try: task_id = scheduler.add_task( task_name=args.name, task_type=args.type, module_path=args.module, cron_expression=args.cron, time_zone=args.timezone ) log.info(f"新任务添加成功，ID: {task_id}") print(f"新任务添加成功，ID: {task_id}") except Exception as e: log.error(f"添加任务失败: {str(e)}", exc_info=True) print(f"添加任务失败: {str(e)}") else: parser.print_help() if __name__ == "__main__": main()