"""命令行入库工具。

用法：
  python cli.py init                 # 仅建库建表
  python cli.py ingest               # 全量入库（默认股票池）
  python cli.py ingest 600519 000001 # 指定股票入库（含快照）
"""
import sys

from db import init_db
import ingest
import init_auth
import watchlist_manager as wl


def main():
    init_db()
    args = sys.argv[1:]
    if not args or args[0] == "init":
        init_auth.init_default_admin()
        wl.init_default_groups()
        print("init done")
        return
    if args[0] == "ingest":
        codes = args[1:] or None
        res = ingest.run_daily_ingest(universe=codes)
        print(res)
    elif args[0] == "ingest_all":
        days = int(args[1]) if len(args) > 1 else 250
        # 先抓快照类数据，再全市场日线
        ingest.run_daily_ingest(with_quotes=False)
        res = ingest.ingest_quotes_all(days=days)
        print(res)


if __name__ == "__main__":
    main()