原帖：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,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\",%f);" %(self.dbtable,price_high_data[0],price_high_data[3],price_high_data[4])
        self.conn.execute(insert_data_cmd)
        #self.conn.execute('INSERT INTO STOCK(date,turnover) VALUES(?,?,?)',(price_high_data[0],price_high_data[4]))
        self.conn.commit()


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