#!/bin/bash
# 批量查询并缓存热门线路高铁时刻表
# 使用方法：./update_cache.sh YOUR_API_KEY

KEY=${1:-""}

if [ -z "$KEY" ]; then
    echo "用法：$0 <API_KEY>"
    echo "示例：$0 abc123def456..."
    exit 1
fi

SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
cd "$SCRIPT_DIR"

echo "========================================"
echo "高铁时刻表缓存更新"
echo "开始时间：$(date '+%Y-%m-%d %H:%M:%S')"
echo "API: 天行数据"
echo "========================================"
echo ""

# 北京出发（10 条）
echo "【北京出发】"
python query.py --from 北京 --to 西安 --api tianapi --key "$KEY"
python query.py --from 北京 --to 上海 --api tianapi --key "$KEY"
python query.py --from 北京 --to 广州 --api tianapi --key "$KEY"
python query.py --from 北京 --to 深圳 --api tianapi --key "$KEY"
python query.py --from 北京 --to 成都 --api tianapi --key "$KEY"
python query.py --from 北京 --to 重庆 --api tianapi --key "$KEY"
python query.py --from 北京 --to 武汉 --api tianapi --key "$KEY"
python query.py --from 北京 --to 南京 --api tianapi --key "$KEY"
python query.py --from 北京 --to 杭州 --api tianapi --key "$KEY"
python query.py --from 北京 --to 长沙 --api tianapi --key "$KEY"
echo ""

# 上海出发（8 条）
echo "【上海出发】"
python query.py --from 上海 --to 北京 --api tianapi --key "$KEY"
python query.py --from 上海 --to 杭州 --api tianapi --key "$KEY"
python query.py --from 上海 --to 南京 --api tianapi --key "$KEY"
python query.py --from 上海 --to 苏州 --api tianapi --key "$KEY"
python query.py --from 上海 --to 无锡 --api tianapi --key "$KEY"
python query.py --from 上海 --to 宁波 --api tianapi --key "$KEY"
python query.py --from 上海 --to 合肥 --api tianapi --key "$KEY"
python query.py --from 上海 --to 福州 --api tianapi --key "$KEY"
echo ""

# 广州出发（6 条）
echo "【广州出发】"
python query.py --from 广州 --to 深圳 --api tianapi --key "$KEY"
python query.py --from 广州 --to 珠海 --api tianapi --key "$KEY"
python query.py --from 广州 --to 北京 --api tianapi --key "$KEY"
python query.py --from 广州 --to 上海 --api tianapi --key "$KEY"
python query.py --from 广州 --to 成都 --api tianapi --key "$KEY"
python query.py --from 广州 --to 重庆 --api tianapi --key "$KEY"
echo ""

# 西安出发（6 条）
echo "【西安出发】"
python query.py --from 西安 --to 北京 --api tianapi --key "$KEY"
python query.py --from 西安 --to 上海 --api tianapi --key "$KEY"
python query.py --from 西安 --to 宝鸡 --api tianapi --key "$KEY"
python query.py --from 西安 --to 咸阳 --api tianapi --key "$KEY"
python query.py --from 西安 --to 汉中 --api tianapi --key "$KEY"
python query.py --from 西安 --to 成都 --api tianapi --key "$KEY"
echo ""

# 成都出发（5 条）
echo "【成都出发】"
python query.py --from 成都 --to 北京 --api tianapi --key "$KEY"
python query.py --from 成都 --to 上海 --api tianapi --key "$KEY"
python query.py --from 成都 --to 西安 --api tianapi --key "$KEY"
python query.py --from 成都 --to 重庆 --api tianapi --key "$KEY"
python query.py --from 成都 --to 贵阳 --api tianapi --key "$KEY"
echo ""

echo "========================================"
echo "缓存更新完成"
echo "结束时间：$(date '+%Y-%m-%d %H:%M:%S')"
echo "========================================"

# 显示缓存统计
echo ""
echo "缓存统计："
sqlite3 train_schedule.db "SELECT COUNT(*) as '总车次数', COUNT(DISTINCT train_no) as '唯一车次', COUNT(DISTINCT from_station || '-' || to_station) as '线路数' FROM trains;"