#!/usr/bin/env python3
"""
测试修复后的日线数据下载
"""
import sys
import os
sys.path.append(os.path.dirname(os.path.abspath(__file__)))

from a_stock_daily_data_downloader import AStockDailyDownloader

print("="*70)
print("🧪 测试修复后的日线数据下载")
print("="*70)

# 创建下载器，只测试10只股票
downloader = AStockDailyDownloader(
    base_dir="/tmp/test_daily_download",
    start_date="2024-01-01",
    end_date="2024-12-31"
)

# 获取股票列表
stocks = downloader.get_all_a_stock_codes()
print(f"\n📊 获取到 {len(stocks)} 只股票")

# 测试前10只
print("\n🧪 测试前10只股票下载...")
success_count = 0
fail_count = 0

for i, stock in enumerate(stocks[:10]):
    print(f"\n{i+1}. {stock['symbol']} {stock['name']}:", end=" ")
    df = downloader.download_stock_daily(
        stock['symbol'], 
        stock['code'], 
        stock['name']
    )
    
    if df is not None and not df.empty:
        print(f"✅ {len(df)} 条记录")
        success_count += 1
    else:
        print(f"❌ 失败")
        fail_count += 1

print(f"\n{'='*70}")
print(f"📊 测试结果: 成功 {success_count}, 失败 {fail_count}, 成功率 {success_count/10*100:.1f}%")
print(f"{'='*70}")

if success_count > 0:
    print("\n🎉 修复成功！可以开始全量下载了")
else:
    print("\n❌ 还有问题，需要继续修复")