from __future__ import annotations

import argparse
from datetime import datetime

from _bootstrap import add_src_to_path

add_src_to_path()

from lhbfx.pipeline import import_daily


def main() -> None:
    parser = argparse.ArgumentParser(description="导入同花顺某日全量龙虎榜数据")
    parser.add_argument("--date", default=datetime.now().strftime("%Y-%m-%d"), help="交易日期，格式 YYYY-MM-DD")
    args = parser.parse_args()

    result = import_daily(args.date)
    print(f"导入日期: {args.date}")
    print(f"概览记录数: {result.overview_count}")
    print(f"营业部明细数: {result.detail_count}")


if __name__ == "__main__":
    main()