python讀取excel文件生成sql文件

學了python這麼久，總算是在工做中用到一次。此次是爲了從excel文件中讀取數據而後寫入到數據庫中。這個邏輯用java來寫的話就過重了，因此此次考慮經過python腳原本實現。

在此以前須要給python添加一個xlrd模塊，這個模塊是專門用來操做excel文件的。

在mac中能夠經過easy_install xlrd命令實現自動安裝模塊

import xdrlib ,sys
import xlrd
def open_excel(file= a.xlsx'):
    try:
        data = xlrd.open_workbook(file)#打開excel文件
        return data
    except Exception,e:
        print str(e)

def excel_table_bycol(file='a.xlsx',colindex=[0],table_name='Sheet1'):
    data = open_excel(file)
    table = data.sheet_by_name(table_name)#獲取excel裏面的某一頁
    nrows = table.nrows#獲取行數
    colnames = table.row_values(0)#獲取第一行的值，做爲key來使用，對於不一樣的excel文件能夠進行調整
    list = []
    #（1，nrows）表示取第一行之後的行，由於第一行每每是表頭
    for rownum in range(1,nrows):
         row = table.row_values(rownum)
         if row:
              app = {}
              for i in colindex:
                   app[str(colnames[i]).encode("utf-8")] = str(row[i]).encode("utf-8")#將數據填入一個字典中，同時對數據進行utf-8轉碼，由於有些數據是unicode編碼的
              list.append(app)#將字典加入列表中去
    return list
def main():
   #colindex是一個數組，用來選擇讀取哪一列，由於每每excel中的一小部分纔是咱們須要的
   tables = excel_table_bycol(colindex=[1,4],table_name=u'areaCode')
   file = open('channel_area_code.sql','w')#建立sql文件，並開啓寫模式
   for row in tables:
        if row['area_code'] != '':
                file.write("update table_name set para1='%s' where para2='%s';\n"%(row['para1'],row['para2']))#往文件裏寫入sql語句
if __name__=="__main__":
    main()

這並不是是一個通用的python腳本，仍是須要根據excel文件的格式做出一些調整,可是代碼並不複雜，開發速度也很快，比之前用java是輕鬆多了。