51 lines
1.6 KiB
Python
51 lines
1.6 KiB
Python
import tushare as ts
|
|
|
|
from stock_config import TUSHARE_TOKEN, get_engine
|
|
|
|
# 1. 配置参数
|
|
# 在 Tushare 官网个人主页获取你的 token
|
|
TOKEN = TUSHARE_TOKEN
|
|
# 数据库连接配置 (格式: mysql+pymysql://用户名:密码@主机地址:端口/数据库名)
|
|
# DB_CONN_STR = 'mysql+pymysql://yangfan:aA%8023321088@8.159.129.156:10836/level?charset=utf8'
|
|
|
|
|
|
def sync_stock_basic():
|
|
# 2. 初始化 tushare 接口
|
|
ts.set_token(TOKEN)
|
|
pro = ts.pro_api()
|
|
|
|
print("正在从 Tushare 获取股票列表数据...")
|
|
|
|
# 3. 获取数据 (对应 doc_id=25 的接口参数)
|
|
# 获取当前所有正常上市交易的股票列表
|
|
df = pro.stock_basic(
|
|
exchange='',
|
|
list_status='L',
|
|
fields='ts_code,symbol,name,area,industry,fullname,enname,cnspell,market,exchange,curr_type,list_status,list_date,delist_date,is_hs,act_name,act_ent_type'
|
|
)
|
|
|
|
if df.empty:
|
|
print("未获取到数据，请检查接口权限或Token设置。")
|
|
return False
|
|
|
|
# 4. 写入数据库
|
|
try:
|
|
# 创建数据库引擎
|
|
engine = get_engine()
|
|
|
|
# 将数据写入数据库表 'stock_basic'
|
|
# if_exists='replace': 如果表存在则替换（适合全量更新基础信息）
|
|
# index=False: 不保存 DataFrame 的索引
|
|
df.to_sql('stock_basic', engine, if_exists='replace', index=False)
|
|
|
|
print(f"成功将 {len(df)} 条股票基础信息写入数据库表 [stock_basic]。")
|
|
return True
|
|
|
|
except Exception as e:
|
|
print(f"写入数据库失败: {e}")
|
|
return False
|
|
|
|
|
|
if __name__ == "__main__":
|
|
sync_stock_basic()
|