import pandas as pd
import tushare as ts
import json

# 从config.json读取tushare token
with open('config.json', 'r') as f:
    config = json.load(f)
    ts.set_token(config['APIKEY'])

# 初始化pro接口
pro = ts.pro_api()

# 获取A股大盘股票列表
stock_list = pro.stock_basic(exchange='', list_status='L')[:100]

# 获取今日交易数据
today = pd.Timestamp.now().strftime('%Y%m%d')
data = []
for ts_code in stock_list['ts_code']:
    df = pro.daily(ts_code=ts_code, trade_date=today)
    if not df.empty:
        data.append(df)

# 合并数据并保存为CSV
if data:
    all_data = pd.concat(data)
    all_data.to_csv('stock_data.csv', index=False)
    print("数据已保存为stock_data.csv")
else:
    print("今日无交易数据")