cfdaily
54 lines
1.4 KiB
Python
54 lines
1.4 KiB
Python
#!/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❌ 还有问题，需要继续修复")
|