import pandas as pd
import tushare as ts
import json
import os

# 从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()

'''
# 获取20250813日所有的ST股票
df = pro.stock_basic(exchange='', list_status='L')
df = df[df['name'].str.contains('ST')]
print(df)

''' 
stock_list = None

stock_list = pd.read_csv('stock_list.csv')
#判断是否有stock_list.csv
if os.path.exists('stock_list.csv'):
    stock_list = pd.read_csv('stock_list.csv')
else:
    pass

if True:
    # 获取A股大盘股票列表
    stock_list = pro.stock_basic(exchange='', list_status='L',fields='ts_code,symbol,name,area,industry,list_date')[:100]
    #将stock_list 保存到 stock_list.csv
    stock_list.to_csv('stock_list.csv',encoding='utf-8', index=False)
    print("股票列表已保存为stock_list.csv")
# 获取今日交易数据
#today = pd.Timestamp.now().strftime('%Y%m%d')
today =  '20250903'
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("今日无交易数据")