#!/usr/bin/env python
# -*- coding:utf-8 -*-
# Author: Hogan
import xlrd
import sqlite3
def read_excel(fileName):
# 打開文件excel
workBook = xlrd.open_workbook(fileName)
# 打開表格
table = workBook.sheets()[0]
# 計算文檔有多少行
all_row = table.nrows
# 返回打開文檔的對象,和文檔的總行數
return table, all_row
def create_con(dbname):
# conn = sqlite3.connect('example2.db') # 連接數據庫
conn = sqlite3.connect(dbname) # 連接數據庫
# connect()方法,可以判斷一個數據庫文件是否存在,如果不存在就自動創建一個,如果存在的話,就打開那個數據庫。
cus = conn.cursor() # 創建游標
return cus, conn
def sql_dao(da, cus):
# cus.execute('''CREATE TABLE stocks(id real ,lng REAL ,lat REAL,slp REAL ,intensity real ,utc text )''')
# 向表中插入一條數據
sql = '''insert into stocks values(%s,'%s','%s','%s','%s','%s')''' % (da[0], da[1], da[2], da[3], da[4], da[5])
cus.execute(sql)
def submit_close(conn):
# 提交當前事務,保存數據
conn.commit()
# 關閉數據庫連接
conn.close()
if __name__ == '__main__':
row_all_obj, all_row_num = read_excel("datastr.xlsx")
conn, cus = create_con()
for i in range(1, all_row_num + 1):
data = row_all_obj.row_values(i)
sql_dao(data,cus )
print("插入第", i, "條")
submit_close(conn)