From d064fdc1f9c5b1980ca0a7673270588f51c00668 Mon Sep 17 00:00:00 2001 From: xhy Date: Fri, 4 Apr 2025 17:42:25 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9=E5=90=AF=E5=8A=A8=E5=8F=82?= =?UTF-8?q?=E6=95=B0?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- app/app.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/app/app.py b/app/app.py index 533ac34..ea0fb64 100644 --- a/app/app.py +++ b/app/app.py @@ -66,7 +66,7 @@ class MainApp: # 添加 web 服务器参数 parser.add_argument( - "--web", action="store_true", help="启动 web 服务器，启动后将忽略其他选项" + "--web", action="store_true", help="启动 web 服务器" ) parser.add_argument( "--web-only", action="store_true", help="启动 web 服务器，但是不启动引擎" @@ -94,7 +94,7 @@ class MainApp: if invalid_modes: parser.error(f'无效的运行模式: {", ".join(invalid_modes)}') args.report = reports - else: + elif args.report is not None: args.report = ["pc", "site", "wap"] # 检查输入的文件是否存在 @@ -133,16 +133,18 @@ class MainApp: signal.signal(signal.SIGINT, self.exit_handler) # 启动所有的 engine - if not self.args.web_only: + if self.args.crawl: self.crawl_engine = CrawlEngine() self.crawl_engine.start() logger.info("crawl 启动") + if self.args.evidence: self.evidence_engine = EvidenceEngine() self.evidence_engine.start() logger.info("evidence 启动") - self.report_engine = Reporter(["pc", "site", "wap"]) + if self.args.report: + self.report_engine = Reporter(self.args.report) self.report_engine.start() logger.info("report 启动")