Python将策划表xlsx转为Lua可用文件

#!/usr/bin/env python
# -*- coding: UTF-8 -*-
# @Date    : 2020-03-21 18:06:21
# @Author  : yan nan fei
# @Version : $Id$

import os,sys
import xlrd

template="""local DataTable = \n{\n[]\n}
function GetTable() return DataTable end
function GetContent(SheetName) return DataTable[SheetName].Content end
"""

tableTemplate="""%{tablename}={{\n[]\n%}},\n"""


KeyValueTemp="""%{key}={value},\n"""

StringValTemp="""%{key}=[=[{value}]=],\n"""

BoolValTemp="""%{key}={value},\n"""

FuncValTemp="""%{key}=function() return {value} end,"""

vTypeDic={"Int":KeyValueTemp,"String":StringValTemp,"RawData":KeyValueTemp,"Bool":BoolValTemp,"Func":FuncValTemp}


def TemplateParse(template,reg):
	template=template.replace("%",reg)
	return template


def MakeTable(template,tablename,content,depth):
	template=TemplateParse(template,depth*"\t")
	if content!=None:
		template=template.replace("[]",content)
	if isinstance(tablename,float):#如果表key为数字
		tablename="[{key}]".format(key=int(tablename))
	if "tablename" in template:
		template=template.format(tablename=tablename)
	return template

def MakeSubTable(parent,tble2):
	tble=parent.replace("[]",tble2)
	return tble

def MakeKeyValue(vaType,key,value,depth):#根据类型拿到相应模板
	if len(str(key))==0 or len(str(value))==0 or "//" in str(key):
		return ""
	if vaType in vTypeDic:
		template=vTypeDic[vaType]
		template=TemplateParse(template,depth*"\t")
		template=template.format(key=key,value=value)
		return template

class ReadSheetContent:
	def __init__(self,workbook,sheetName):
		self.workbook=workbook
		self.sheetName=sheetName
	def GetContent(self):
		content=""
		sheetContent=self.workbook.sheet_by_name(self.sheetName)
		for row in range(0,sheetContent.nrows):
			if row<3: #前三行分别为备注，变量，变量类型
				continue
			rowTableName=sheetContent.cell_value(row,0)
			rowTableObject=MakeTable(tableTemplate,rowTableName,"[]",3)
			colContent=""
			for col in range(0,sheetContent.ncols):
				key=sheetContent.cell_value(1,col) #第二行表示变量
				vaType=sheetContent.cell_value(2,col)#第三行表示变量类型
				value=sheetContent.cell_value(row,col)
				keyValueObj=MakeKeyValue(vaType,key,value,5)
				colContent=colContent+keyValueObj
			rowTableObject=rowTableObject.replace("[]",colContent)
			content=content+rowTableObject
		return content



if __name__=="__main__":
	if len(sys.argv)>1:
		targetFileName=sys.argv[1]
		targetDir=os.path.dirname(targetFileName)
		if os.path.isdir(targetDir):
			os.makedirs(targetDir)
		workbook=xlrd.open_workbook(targetFileName)
		allSheetNames = workbook.sheet_names();
		fileNameNoExten=targetFileName.split('.')[0]
		with open(fileNameNoExten+".lua","w") as file:
			luaContent=""
			for sheetName in allSheetNames:
				if sheetName=="main":
					continue
				sheetTable=MakeTable(tableTemplate,sheetName,"[]",1)
				ContentTable=MakeTable(tableTemplate,"Content","[]",2)
				sheetTable=MakeSubTable(sheetTable,ContentTable)
				readSheetContent=ReadSheetContent(workbook,sheetName)
				SheetContent=readSheetContent.GetContent()
				luaContent=luaContent+MakeSubTable(sheetTable,SheetContent)
			template=template.replace("[]",luaContent)
			file.write(template)