摘要：之前寫過一篇使用腳本生成文件的文章，時隔很久這種終于有空寫該庫的兄弟庫，用來讀取文件。

之前寫過一篇使用Python腳本生成Excel文件的文章，時隔很久這種終于有空寫該庫的兄弟庫xlrd，用來讀取Excel文件。
最近被調到電商項目，由于種種原因（設計不完善、SQL語句書寫不規范，甲方太奇葩...槽點太多，就不在這里吐槽了）需要經常進行刷數據操作（批量修改錯誤數據），一般就是寫一寫SQL文件來直接操作，但是在進行了反復的ctrl+c和ctrl+v的操作之后，終于決定寫一個腳本，直接將Excel文件中需要insert和update讀取出來，并生成SQL文件。

項目GitHub地址：https://github.com/sunny0826/...

生成update語句：

#!/usr/bin/env python
#encoding: utf-8
#Author: guoxixi
import xlrd
import sys
reload(sys)
sys.setdefaultencoding("utf-8")

def open_excel(file="test.xls"):
    try:
        data = xlrd.open_workbook(file)  # 打開excel文件
        return data
    except Exception, e:
        print str(e)


def excel_table_bycol(file="", where=[0], colindex=[0], table_name="Sheet1"):
    data = open_excel(file)
    table = data.sheet_by_name(table_name)  # 獲取excel里面的某一頁
    nrows = table.nrows  # 獲取行數
    t_name = table.row_values(0)[0].encode("utf8") #表名
    colnames = table.row_values(1)  # 獲取第一行的值，作為key來使用
    list = []
    # （2，nrows）表示取第二行以后的行，第一行為表名，第二行為表頭
    for rownum in range(2, nrows):
        row = table.row_values(rownum)
        if row:
            whe = {}
            for n in where:
                whe[str(colnames[n]).encode("utf-8")] = str(row[n]).encode("utf-8")  #輸入的篩選字段
            app = {}
            for i in colindex:
                app[str(colnames[i]).encode("utf-8")] = str(row[i]).encode("utf-8")  # 將數據填入一個字典中，同時對數據進行utf-8轉碼，因為有些數據是unicode編碼的
            list.append({"where":whe,"app":app})  # 將字典加入列表中去
    return list,t_name


def main(file,where,colindex):
    # colindex為需要更新的列，where為篩選的列
    tables = excel_table_bycol(file,where,colindex, table_name=u"Sheet1")
    with open("./sql_result/update#"+tables[1]+".sql", "w") as f:    # 創建sql文件，并開啟寫模式
        for info in tables[0]:
            sql_line = "UPDATE "+tables[1]+" SET"
            apps = info.get("app")
            for key,value in apps.items():
                if sql_line.endswith("SET"):
                    sql_line += " "+key+"=""+value+"" "
                else:
                    sql_line += ", " + key + "="" + value + "" "
            sql_line += " WHERE"
            where = info.get("where")
            for key, value in where.items():
                if sql_line.endswith("WHERE"):
                    sql_line += " "+key+"=""+value+"" "
                else:
                    sql_line += "AND " + key + "="" + value + "" "
            sql_line+="
"
            f.write(sql_line)  # 往文件里寫入sql語句

if __name__ == "__main__":
    file_name = "test.xls"  # 導入xls文件名
    where = [0,1,2]         # 條件字段
    colindex = [3, 4]       # 需要插入的列
    main(file_name,where,colindex)

在Excel文件中，第一行行需要update的表名，第二行為字段名，每一列為該字段的對應值，這里可以進行where過濾，只需要修改main函數下where數組內的數字即可，需要update的字段同理，執行后就會生成名字為update#表名.sql的SQL文件。

生成insert語句：

#!/usr/bin/env python
#encoding: utf-8
#Author: guoxixi
import xlrd
import sys
reload(sys)
sys.setdefaultencoding("utf-8")

def open_excel(file="test.xls"):
    try:
        data = xlrd.open_workbook(file)  # 打開excel文件
        return data
    except Exception, e:
        print str(e)


def excel_table_bycol(file="", colindex=[0], table_name="Sheet1"):
    data = open_excel(file)
    table = data.sheet_by_name(table_name)  # 獲取excel里面的某一頁
    nrows = table.nrows  # 獲取行數
    t_name = table.row_values(0)[0].encode("utf8") #表名
    colnames = table.row_values(1)  # 獲取第一行的值，作為key來使用
    list = []
    # （2，nrows）表示取第二行以后的行，第一行為表名，第二行為表頭
    list.append(t_name)
    list.append(colnames)
    for rownum in range(2, nrows):
        row = table.row_values(rownum)
        if row:
            app = []
            for i in colindex:
                app.append(str(row[i]).encode("utf-8") )
            list.append(app)  # 將字典加入列表中去
    return list


def main(file_name,colindex):
    # colindex為需要插入的列
    tables = excel_table_bycol(file_name,colindex, table_name=u"Sheet1")
    t_name = tables.pop(0)
    key_list = ",".join(tables.pop(0)).encode("utf8")   #list轉為str
    sql_line = "INSERT INTO "+t_name+"（"+key_list+"）VALUE"
    line = ""
    for info in tables:
        content = ",".join(info)
        if line != "":
            line =line + ",(" + content + ")"
        else:
            line = "("+content+")"
    sql_line = sql_line + line + ";"
    with open("./sql_result/insert#" + t_name + ".sql", "w") as f:  # 創建sql文件，并開啟寫模式
        f.write(sql_line)  # 往文件里寫入sql語句

if __name__ == "__main__":
    file_name = "test.xls"          #導入xls文件名
    colindex = [0, 1, 2, 3, 4]      #需要插入的列
    main(file_name,colindex)

生成insert語句SQL文件的Excel格式與update的相同，但是傳入參數方面，因為不需要過濾條件，所以只需要往數組中寫入需要插入字段的列數就好。

希望對大家有所幫助^_^