from xlwt import *
'''对excel增加行'''
def writeRow(tableObj,rowNum,tmpList,xlsStyleDict={}):
for num,o in enumerate(tmpList):
if num in xlsStyleDict:
style = xlsStyleDict.get(num)
tableObj.write(rowNum,num,o,style)
else:
tableObj.write(rowNum,num,o)
'''@param fileName:文件名
@param dataDict:数据
dataDict格式 {"_sheetName":[u'a',u'b',u'c'],
"_sheetData":{"a":[['1','2','3'],['a','b','c']],
"b":[['11','22','33'],['aa','bb','cc']],
"c":[['111','222','333'],['aaa','bbb','ccc']]
},
"_sheetTitle":{"a":[[u'日期',u'visits数量',u'leads数量']]}
"_sheetStyle":{"a":{0:XFStyleobject,1:XFStyleobject}}
}
'''
def createExcelFile(fileName,dataDict):
if dataDict:
excelfile = Workbook()
sheetNameList = dataDict.get("_sheetName",[])
sheetDataDict = dataDict.get("_sheetData",{})
sheetTitleDict = dataDict.get("_sheetTitle",{})
sheetStyleDict = dataDict.get("_sheetStyle",{})
for sheetName in sheetNameList:
ws = excelfile.add_sheet(sheetName)
titleList = sheetTitleDict.get(sheetName,[])
styleDict = sheetStyleDict.get(sheetName,{})
dataList = sheetDataDict.get(sheetName,[])
rowIdx = 0
for title in titleList:
writeRow(ws,rowIdx,title)
rowIdx +=1
for data in dataList:
writeRow(ws,rowIdx,data,styleDict)
rowIdx +=1
excelfile.save(fileName)
