從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