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

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

項(xiàng)目GitHub地址：https://github.com/sunny0826/...

生成update語(yǔ)句：

#!/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里面的某一頁(yè)
    nrows = table.nrows  # 獲取行數(shù)
    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")  # 將數(shù)據(jù)填入一個(gè)字典中，同時(shí)對(duì)數(shù)據(jù)進(jìn)行utf-8轉(zhuǎn)碼，因?yàn)橛行?shù)據(jù)是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:    # 創(chuàng)建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語(yǔ)句

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

在Excel文件中，第一行行需要update的表名，第二行為字段名，每一列為該字段的對(duì)應(yīng)值，這里可以進(jìn)行where過濾，只需要修改main函數(shù)下where數(shù)組內(nèi)的數(shù)字即可，需要update的字段同理，執(zhí)行后就會(huì)生成名字為update#表名.sql的SQL文件。

生成insert語(yǔ)句：

#!/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里面的某一頁(yè)
    nrows = table.nrows  # 獲取行數(shù)
    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轉(zhuǎn)為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:  # 創(chuàng)建sql文件，并開啟寫模式
        f.write(sql_line)  # 往文件里寫入sql語(yǔ)句

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

生成insert語(yǔ)句SQL文件的Excel格式與update的相同，但是傳入?yún)?shù)方面，因?yàn)椴恍枰^濾條件，所以只需要往數(shù)組中寫入需要插入字段的列數(shù)就好。

希望對(duì)大家有所幫助^_^