修改启动参数

This commit is contained in:
xhy 2025-04-04 17:42:25 +08:00
parent e4287c6605
commit d064fdc1f9

View File

@ -66,7 +66,7 @@ class MainApp:
# 添加 web 服务器参数 # 添加 web 服务器参数
parser.add_argument( parser.add_argument(
"--web", action="store_true", help="启动 web 服务器,启动后将忽略其他选项" "--web", action="store_true", help="启动 web 服务器"
) )
parser.add_argument( parser.add_argument(
"--web-only", action="store_true", help="启动 web 服务器,但是不启动引擎" "--web-only", action="store_true", help="启动 web 服务器,但是不启动引擎"
@ -94,7 +94,7 @@ class MainApp:
if invalid_modes: if invalid_modes:
parser.error(f'无效的运行模式: {", ".join(invalid_modes)}') parser.error(f'无效的运行模式: {", ".join(invalid_modes)}')
args.report = reports args.report = reports
else: elif args.report is not None:
args.report = ["pc", "site", "wap"] args.report = ["pc", "site", "wap"]
# 检查输入的文件是否存在 # 检查输入的文件是否存在
@ -133,16 +133,18 @@ class MainApp:
signal.signal(signal.SIGINT, self.exit_handler) signal.signal(signal.SIGINT, self.exit_handler)
# 启动所有的 engine # 启动所有的 engine
if not self.args.web_only: if self.args.crawl:
self.crawl_engine = CrawlEngine() self.crawl_engine = CrawlEngine()
self.crawl_engine.start() self.crawl_engine.start()
logger.info("crawl 启动") logger.info("crawl 启动")
if self.args.evidence:
self.evidence_engine = EvidenceEngine() self.evidence_engine = EvidenceEngine()
self.evidence_engine.start() self.evidence_engine.start()
logger.info("evidence 启动") 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() self.report_engine.start()
logger.info("report 启动") logger.info("report 启动")