将从tushare上下载到的数据批量存储到postgres库-SZ基表-SZ000001分表中
# -*- coding:utf-8 -*-
"""
@author:Shawn.li
@file:insertData.py
@time:2020/2/23 15:05
"""
import psycopg2
import tushare as ts
# 获取股票数据,并转换数据格式为字典模式
stockCode = "000001.SZ"
pro = ts.pro_api()
data = pro.daily(ts_code=stockCode, start_date='20170701', end_date='20180718')
# 将数据从dataframe格式转换成split字典格式,并通过get获取其中的数据值
data = data.to_dict(orient='split')
data = data.get('data')
# 将获取的数据插入数据库
conn = psycopg2.connect(database="postgres", user="postgres", password="5432", host="127.0.0.1", port="5432")
cur = conn.cursor()
sqlIndex = "insert into SZ000001 (" \
"ts_code," \
"trade_date," \
"price_open," \
"price_high," \
"price_low," \
"price_close," \
"price_pre_close," \
"price_change," \
"price_pct_chg," \
"vol," \
"amount) " \
"values (%s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s)"
cur.executemany(sqlIndex, data)
conn.commit