#!/usr/bin/env python3
"""
過去レース結果の取り込み（雛形）

データ取得は利用規約・ライセンスに従うこと。
取り込み本体は import_csv 等で実装予定。
"""
from __future__ import annotations

import argparse
import logging
import os
import sys

sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))

from collectors.db_util import connect  # noqa: E402

logging.basicConfig(level=logging.INFO, format="%(asctime)s [%(levelname)s] %(message)s")
logger = logging.getLogger(__name__)


def main() -> None:
    parser = argparse.ArgumentParser(description="競馬レース結果取り込み")
    parser.add_argument("--check-db", action="store_true", help="DB接続と件数確認のみ")
    args = parser.parse_args()

    if args.check_db:
        conn = connect()
        try:
            with conn.cursor() as cur:
                cur.execute("SELECT COUNT(*) AS c FROM races")
                r = cur.fetchone()
                logger.info("races count: %s", r["c"])
                cur.execute("SELECT COUNT(*) AS c FROM tracks")
                t = cur.fetchone()
                logger.info("tracks count: %s", t["c"])
        finally:
            conn.close()
        return

    logger.info("取り込み処理はデータソース決定後に実装")


if __name__ == "__main__":
    main()