| 101 | LogPrinter.info('Setting Lang env done, locale is %s.%s' % (code, encoding)) |
| 102 | |
| 103 | def main(self): |
| 104 | args = self._params |
| 105 | LogPrinter.print_logo() |
| 106 | |
| 107 | if args.command == 'localscan': |
| 108 | '''执行本地项目扫描''' |
| 109 | from node.localtask.localrunner import LocalRunner |
| 110 | LocalRunner(args).run() |
| 111 | |
| 112 | elif args.command == 'start': |
| 113 | '''启动任务执行端,持续获取并执行任务''' |
| 114 | from node.servertask.looprunner import LoopRunner |
| 115 | if args.token: |
| 116 | LoopRunner(args).run() |
| 117 | else: |
| 118 | LogPrinter.error("缺少token参数,请通过-t <token>启动start命令.") |
| 119 | |
| 120 | elif args.command == 'quickinit': |
| 121 | '''快速分析初始化工具''' |
| 122 | from node.quicktask.toolloader import QuickScanToolLoader |
| 123 | QuickScanToolLoader.load_tools(args) |
| 124 | |
| 125 | elif args.command == 'quickscan': |
| 126 | from node.quicktask.quickrunner import QuickRunner |
| 127 | QuickRunner(args).run() |
| 128 | |
| 129 | elif args.command == 'task': |
| 130 | '''执行单个工具任务''' |
| 131 | from node.testrunner import TestRunner |
| 132 | if args.request_file: |
| 133 | TestRunner(args.request_file).run() |
| 134 | else: |
| 135 | LogPrinter.error("请输入request.json文件, 输入 -h 查看帮助文档.") |
| 136 | |
| 137 | elif args.command == 'updatetool': |
| 138 | '''更新工具库''' |
| 139 | # 从git拉取工具配置库 |
| 140 | ToolConfigLoader().load_tool_config() |
| 141 | if args.tool: # 更新指定工具 |
| 142 | tool_name_list = StringMgr.str_to_list(args.tool) |
| 143 | ToolLoader(tool_names=tool_name_list, os_type=args.os_type, include_common=False).git_load_tools(print_enable=False) |
| 144 | elif args.all_tools: # 更新全量工具 |
| 145 | ToolLoader(os_type=args.os_type, config_all_tools=True, include_common=False).git_load_tools(print_enable=False) |
| 146 | else: |
| 147 | LogPrinter.error("请输入必要的参数(-a|-t)! 输入 -h 查看帮助文档.") |
| 148 | |
| 149 | elif args.command == 'help': |
| 150 | '''输出帮助文档''' |
| 151 | CmdArgParser.print_help() |
| 152 | |
| 153 | else: |
| 154 | CmdArgParser.print_help() |
| 155 | |
| 156 | |
| 157 | if __name__ == "__main__": |