从tushare上下载数据并批量存入postgres数据库中

将从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

猜你喜欢

转载自blog.csdn.net/ljl86400/article/details/104465637