external_demand/app/hot_content/cli.py

"""热点内容流程命令行入口。"""

from __future__ import annotations

import argparse
import json

from app.hot_content.config import load_flow_config
from app.hot_content.service import run_once


def parse_args() -> argparse.Namespace:
    parser = argparse.ArgumentParser(description="热点内容 MySQL 入库与定时调度流程")
    parser.add_argument("--once", action="store_true", help="只执行一次，不进入循环调度")
    return parser.parse_args()


def main() -> None:
    args = parse_args()
    config = load_flow_config()
    if args.once:
        summary = run_once(config)
        print(json.dumps(summary, ensure_ascii=False, indent=2))
        return

    from app.scheduler import start_scheduler

    start_scheduler()


if __name__ == "__main__":
    main()