记录——excel导出lua工具（python实现）

　　项目需要一个从excel导出lua配置表的工具，之前的工具是主程写的，效率极差，i7 CPU 一次全部导出要花掉1个多小时。匪夷所思的是，这么渣的效率，居然用了整整一年。当

然，中途有人反映效率差，主程直接无视了，随着呼声越来越高，主程终于有空行动了。

　　一天下来，主程宣布，其工具效率提升了近50%，并且说了很多非专业人士听不懂的话，什么换库了，什么小文件没法提速，什么python效率就是这么差……

当即就有同志体验了改进过的工具，导出某几个文件速度的的确确，确确实实，实实在在提升了50%，于是公司内部群里好几个同志对主程竖起了大拇指，主程表示举手之劳，不足以

挂齿，随即便在公司论坛发帖《转：python之excel哪家强》，内容是介绍python几个常用excel库的优劣，以及自己在考虑众多因素之后果断的抉择。

　　在无比佩服的同时，我决定重写这个工具，一接手，深似海，非常不规范的excel表，非常不灵活的格式，还有大堆针对某些表的特殊处理，整个设计用一句话来形容：跪了。

在折腾了一天之后终于搞定了，跟原工具用一样的库openpyxl，兼容99.9%原配置表，而性能却比之前改进过的工具快了230倍左右。

　　废话说了一大堆。

支持类型：

int,bool,float,string,

struct<[int,bool,float,string]...>,

list<int>,

list<bool>,

list<float>,

list<string>,

list<struct<[int,bool,float,string]...>>,

填写格式：

line 1: 注释

line 2: 字段类型:字段名字 => int:name | list<int:>:name | struct<int:n1,int:n2>:name

line 3：内容

line N：内容

源码：

 # -*- coding:utf-8 -*- 

 #    excel 导出 lua 表
 #    支持类型，int，bool，float，string，list，struct，list<int>，list<bool>，list<float>，list<string>，list<struct>

 import openpyxl
 import codecs
 import time

 TInt = "int"
 TList = "list"
 TBool = "bool"
 TFloat = "float"
 TString = "string"
 TStruct = "struct"

 def ValToKey(val):
     return val.isdecimal() and "[" + val + "]" or val

 def CheckInt(data, args = None):
     return data

 def CheckBool(data, args = None):
     " == data and "false" or "true"

 def CheckFloat(data, args = None):
     return data

 def CheckString(data, args = None):
     return "\"" + data.replace("\n", "\\n") + "\""

 def CheckList(data, args):
     vals = data.split(";")
     vals.pop()
     if len(vals) == 0:
         return ""
     func = args[0]["func"]
     args = args[0]["args"]
     array = [func(v, args) for k, v in enumerate(vals)]
     return "{ " + ", ".join(array) + " }"

 def CheckStruct(data, args):
     vals = data.split(",")
     vals.pop()
     if len(vals) == 0:
         return ""
     array = []
     for k, v in enumerate(vals):
         val = args[k]["func"](v, args[k]["args"])
         name = args[k]["name"]
         if len(val) != 0:
             array.append(name + " = " + val)
     return "{ " + ", ".join(array) + " }"

 def GetValue(sheet, row, col):
     return unicode(sheet.cell(row = row, column = col).value or "")

 def GetLine(sheet, row):
     return [GetValue(sheet, row, col) for col in xrange(1, sheet.max_column + 1)]

 def CheckParses(fields):
     result = []
     for k, field in enumerate(fields):
         pos = field.rfind(":")
         type = field[0:pos]
         name = field[pos+1:]
         parse = { "args": None }
         if TInt == type[:len(TInt)]:
             parse["func"] = CheckInt
         elif TBool == type[:len(TBool)]:
             parse["func"] = CheckBool
         elif TFloat == type[:len(TFloat)]:
             parse["func"] = CheckFloat
         elif TString == type[:len(TString)]:
             parse["func"] = CheckString
         elif TList == type[:len(TList)]:
             parse["func"] = CheckList
             parse["args"] = CheckParses([type[len(TList)+1:-1]])
         elif TStruct == type[:len(TStruct)]:
             parse["func"] = CheckStruct
             parse["args"] = CheckParses(type[len(TStruct)+1:-1].split(","))
         if len(name) == 0:
             parse["name"] = ValToKey(unicode(k))
         else:
             parse["name"] = ValToKey(name)
         result.append(parse)
     return result

 def CheckChunk(parses, fields):
     chunk = []
     for index, field in enumerate(fields):
         key = parses[index]["name"]
         val = parses[index]["func"](field, parses[index]["args"])
         if len(val) != 0:
             chunk.append(key + " = " + val)
     return "{ " + ", ".join(chunk) + " }"

 def Export(sheet):
     #    第一行注释
     #    第二行类型
     parses = CheckParses(GetLine(sheet, 2))
     #    第三行内容
     chunks, indexs, parse0 = [], [], parses[0]
     for row in xrange(3, sheet.max_row + 1):
         value = GetValue(sheet, row, 1)
         index = parse0["func"](value, parse0["args"])
         indexs.append(index)

         chunk = CheckChunk(parses, GetLine(sheet, row))
         chunks.append("%s = %s" % (ValToKey(index), chunk))
         print row, index

     result = [    "return ", \
                 "{ " + "\n".join(chunks) + " }", \
                 ",", \
                 "{ " + ", ".join(indexs) + " }",]
     return "\n".join(result)

 def Main(opath, ipath):
     xlsx = openpyxl.load_workbook(opath)
     sheet = xlsx.get_sheet_by_name(xlsx.sheetnames[0])
     result = Export(sheet)
     with codecs.open(ipath, "w", "utf-8") as f:
         f.write(result)

 if __name__ == "__main__":
     Main("D:\\MyWork\\py\\a.xlsx", "D:\\MyWork\\py\\a.lua")

因为公司项目配置表格式很烂，所以这一份是另外写的，方便以后我自己用的，未经测试！！！

下载