2569718930@qq.com
50 lines
1.4 KiB
Python
50 lines
1.4 KiB
Python
from __future__ import annotations
|
|
|
|
import argparse
|
|
import os
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
ROOT = Path(__file__).resolve().parents[1]
|
|
if str(ROOT) not in sys.path:
|
|
sys.path.insert(0, str(ROOT))
|
|
|
|
|
|
def parse_args() -> argparse.Namespace:
|
|
from src.utils.prewarm_dashboard import DEFAULT_CITIES
|
|
|
|
parser = argparse.ArgumentParser(
|
|
description="Prewarm PolyWeather summary/detail/market caches for selected cities.",
|
|
)
|
|
parser.add_argument(
|
|
"--base-url",
|
|
default=os.getenv("POLYWEATHER_BACKEND_URL", "http://127.0.0.1:8000"),
|
|
help="Backend base URL, defaults to POLYWEATHER_BACKEND_URL or http://127.0.0.1:8000",
|
|
)
|
|
parser.add_argument(
|
|
"--cities",
|
|
default=",".join(DEFAULT_CITIES),
|
|
help="Comma-separated city names to prewarm",
|
|
)
|
|
parser.add_argument("--force-refresh", action="store_true")
|
|
parser.add_argument("--include-detail", action="store_true")
|
|
parser.add_argument("--include-market", action="store_true")
|
|
return parser.parse_args()
|
|
|
|
|
|
def main() -> int:
|
|
args = parse_args()
|
|
from src.utils.prewarm_dashboard import run_prewarm
|
|
|
|
return run_prewarm(
|
|
base_url=args.base_url,
|
|
cities=args.cities,
|
|
force_refresh=bool(args.force_refresh),
|
|
include_detail=bool(args.include_detail),
|
|
include_market=bool(args.include_market),
|
|
)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
raise SystemExit(main())
|