freedakgmail
64 lines
2.1 KiB
Python
64 lines
2.1 KiB
Python
"""
|
|
更新股票缓存数据（从上年1月1日起）
|
|
"""
|
|
import os
|
|
import json
|
|
import glob
|
|
from datetime import datetime, timedelta
|
|
from services.stock_service import get_stock_fund_flow, save_cached_data
|
|
|
|
def update_all_caches():
|
|
cache_dir = 'stock_data_cache'
|
|
if not os.path.exists(cache_dir):
|
|
print("缓存目录不存在")
|
|
return
|
|
|
|
cache_files = glob.glob(f'{cache_dir}/*.json')
|
|
print(f"找到 {len(cache_files)} 个缓存文件")
|
|
|
|
# 日期范围：从上年1月1日到今天
|
|
end_date = datetime.now().strftime('%Y-%m-%d')
|
|
start_date = f'{datetime.now().year - 1}-01-01'
|
|
print(f"更新日期范围: {start_date} ~ {end_date}")
|
|
|
|
updated = 0
|
|
failed = 0
|
|
|
|
for i, cache_file in enumerate(cache_files):
|
|
stock_code = os.path.basename(cache_file).replace('.json', '')
|
|
print(f"[{i+1}/{len(cache_files)}] 更新 {stock_code}...", end=" ", flush=True)
|
|
|
|
try:
|
|
# 检查现有缓存数据范围
|
|
with open(cache_file, 'r', encoding='utf-8') as f:
|
|
data = json.load(f)
|
|
|
|
records = data.get('records', [])
|
|
if records:
|
|
first_date = records[0].get('日期', '')
|
|
# 如果已经有上年1月的数据，跳过
|
|
if first_date and first_date <= start_date:
|
|
print(f"已有上年数据({first_date})，跳过")
|
|
continue
|
|
|
|
# 需要更新
|
|
df, name, error = get_stock_fund_flow(stock_code, start_date, end_date)
|
|
if error:
|
|
print(f"错误: {error}")
|
|
failed += 1
|
|
elif df is not None and len(df) > 0:
|
|
save_cached_data(stock_code, df, name)
|
|
print(f"成功 ({len(df)}条)")
|
|
updated += 1
|
|
else:
|
|
print("无数据")
|
|
failed += 1
|
|
except Exception as e:
|
|
print(f"异常: {e}")
|
|
failed += 1
|
|
|
|
print(f"\n完成! 更新: {updated}, 失败: {failed}")
|
|
|
|
if __name__ == '__main__':
|
|
update_all_caches()
|