补充Mock规则命中模式启动参数
This commit is contained in:
@@ -3,6 +3,9 @@
|
||||
|
||||
基于 FastAPI 实现的 Mock 服务器,用于模拟流水分析平台的 7 个核心接口
|
||||
"""
|
||||
import argparse
|
||||
import os
|
||||
|
||||
from fastapi import FastAPI
|
||||
from routers import api
|
||||
from config.settings import settings
|
||||
@@ -68,9 +71,23 @@ async def health_check():
|
||||
}
|
||||
|
||||
|
||||
def parse_args(argv=None):
|
||||
parser = argparse.ArgumentParser()
|
||||
parser.add_argument("--rule-hit-mode", choices=["subset", "all"], default="subset")
|
||||
return parser.parse_args(argv)
|
||||
|
||||
|
||||
def apply_rule_hit_mode(rule_hit_mode: str) -> None:
|
||||
os.environ["RULE_HIT_MODE"] = rule_hit_mode
|
||||
settings.RULE_HIT_MODE = rule_hit_mode
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
import uvicorn
|
||||
|
||||
args = parse_args()
|
||||
apply_rule_hit_mode(args.rule_hit_mode)
|
||||
|
||||
# 启动服务器
|
||||
uvicorn.run(
|
||||
app,
|
||||
|
||||
Reference in New Issue
Block a user