python从sqlite中提取数据到excel

 import sqlite3 as sqlite
 from xlwt import *
 import sys

 def sqlite_get_col_names(cur, select_sql):
     cur.execute(select_sql)
     return [tuple[0] for tuple in cur.description]

 def query_by_sql(cur, select_sql):
     cur.execute(select_sql)
     return cur.fetchall()

 def sqlite_to_workbook_with_head(cur, table, select_sql, workbook):
     ws = workbook.add_sheet(table)
     print('create table %s.' % table)
     #enumerate针对一个可迭代对象，生成的是序号加上内容

     for colx, heading in enumerate(sqlite_get_col_names(cur, select_sql)):
         ws.write(0, colx, heading)    #在第1行的colx列写上头部信息

     for rowy, row in enumerate(query_by_sql(cur, select_sql)):
         for colx, text in enumerate(row):    #row是一行的内容
             ws.write(rowy + 1, colx, text)    #在rowy+1行，colx写入数据库内容text

 def sqlite_to_workbook_without_head(cur, table, select_sql, workbook):
     ws = workbook.add_sheet(table)

     for rowy, row in enumerate(query_by_sql(cur, select_sql)):
         for colx, text in enumerate(row):    #row是一行的内容
             ws.write(rowy, colx, text)    #在rowy行，colx写入数据库内容text

 def dump_db_to_excel(cur, workbook):
     for tbl_name in [row[0] for row in query_by_sql(cur, "select tbl_name FROM sqlite_master where type = 'table'")]:
         select_sql = "select * from '%s'" % tbl_name
         sqlite_to_workbook_with_head(cur, tbl_name, select_sql, workbook)

 def main(dbpath):
     xlspath = dbpath[:dbpath.rfind('.')] + '.xls'
     print("<%s> --> <%s>" % (dbpath, xlspath))

     db = sqlite.connect(dbpath)
     cur = db.cursor()
     w = Workbook()

     dump_db_to_excel(cur, w)    #把所有的db中的表数据导出到excel中，每个table对应一个sheet页

     #按照条件查询数据并导出到excel中
     #sheet_name = '测试'
     #query_data_sql = "select 100-id as used from cpu_info where cpu_name = '%Cpu0'"
     #sqlite_to_workbook_without_head(cur, sheet_name, query_data_sql, w)

     cur.close()
     db.close()

     w.save(xlspath)

 if __name__ == "__main__":
     # arg == database path
     main(sys.argv[1])