#!/bin/bash
# A股历史数据获取 - V2优化版

cd "$(dirname "$0")"

# 检查配置
if [ ! -f "config.txt" ]; then
    echo "请先配置 Tushare Token:"
    echo "  echo 'your_token' > config.txt"
    exit 1
fi

# 检查股票列表
if [ ! -f "A股股票列表.csv" ]; then
    echo "缺少股票列表文件: A股股票列表.csv"
    exit 1
fi

# 运行
echo "启动 A股历史数据获取 V2..."
python3 fetch_history_v2.py

# 完成后生成汇总文件
if [ -f "data/stock_daily_data.parquet" ]; then
    echo "生成CSV汇总文件..."
    python3 -c "
import pandas as pd
from pathlib import Path
from datetime import datetime

df = pd.read_parquet('data/stock_daily_data.parquet')
timestamp = datetime.now().strftime('%Y%m%d')
df.to_csv(f'data/A股日线数据_{timestamp}.csv', index=False)
print(f'CSV文件: data/A股日线数据_{timestamp}.csv ({len(df)}条记录)')
"
fi