vent_agent_2.0/scripts/mock_result_platform.py

#!/usr/bin/env python3
"""独立启动结果接收平台，所有回执存入自己的 SQLite。"""
import argparse
import os
from pathlib import Path
import sys

ROOT = Path(__file__).resolve().parents[1]
sys.path.insert(0, str(ROOT))


def main():
    parser = argparse.ArgumentParser()
    parser.add_argument("--port", type=int, default=8767)
    parser.add_argument("--db", type=Path, default=ROOT / "data/mock_result_platform.sqlite")
    parser.add_argument("--fail-first", type=int, default=0, help="前 N 次推送返回503，用于验证重试")
    args = parser.parse_args()
    import uvicorn
    from mock_result_platform.app import LOCAL_TOKEN, ResultStore, create_app
    app = create_app(ResultStore(args.db), token=os.getenv("MOCK_RESULT_PLATFORM_TOKEN", LOCAL_TOKEN),
                     fail_first=max(0, args.fail_first))
    uvicorn.run(app, host="127.0.0.1", port=args.port)


if __name__ == "__main__":
    main()