tushare数据写入SQLite 数据库

原帖： http://www.30daydo.com/article/73

这一节 我们学习如何把得到的数据写入数据库。
 
虽然也可以写入excel或者json，不过考虑到后面用的的排序和其他python脚本的调用，最后选择了轻量级的数据库SQLiite作为首选。

# -*-coding=utf-8-*-
#数据库的操作

import sqlite3, time, datetime

__author__ = 'rocchen'


class SqliteDb():


    def __init__(self,dbtable):
        '''
        self.today = time.strftime("%Y-%m-%d")
        self.DBname = self.today + '.db'
        self.conn = sqlite3.connect(self.DBname)
        '''
        today = time.strftime("%Y-%m-%d")
        DBname = today + '.db'
        self.conn = sqlite3.connect(DBname)
        self.dbtable=dbtable
        create_tb = "CREATE TABLE %s (date varchar(10),id varchar(6), name varchar(30), p_change REAL,turnover REAL);" %self.dbtable
        self.conn.execute(create_tb)
        self.conn.commit()

    def store_break_high(self,price_high_data):

        #data 是创新高的个股信息  dataframe
        #print today
        #create_tb = 'CREATE TABLE STOCK (date TEXT,id text PRIMARY KEY, p_change REAL,turnover REAL);'

        #conn.commit()
        #print "(%s,%s,%f,%f)" %(price_high_data[0], price_high_data[1], price_high_data[2], price_high_data[3])
        insert_data_cmd = "INSERT INTO %s(date,id,name,p_change,turnover) VALUES(\"%s\",\"%s\",\"%s\",%f,%f);" %(self.dbtable,price_high_data[0], price_high_data[1], price_high_data[2], price_high_data[3],price_high_data[4])
        self.conn.execute(insert_data_cmd)
        #self.conn.execute('INSERT INTO STOCK(date,id,name,p_change,turnover) VALUES(?,?,?,?,?)',(price_high_data[0], price_high_data[1], price_high_data[2], price_high_data[3],price_high_data[4]))
        self.conn.commit()


    def close(self):
        self.conn.close()

上面创建的表名是 以日期为命名的（前面的下划线是因为数据库的命名规则不能以数字为首）
 
上一篇：30天学会量化交易模型 Day04 （tushare获取破新高的股票）
http://www.30daydo.com/article/70