python + openpyxl + Jinja2（解析excel生成javaBean和json）
阿新 • • 發佈：2019-02-17
模版
package {{ package }};
import java.io.BufferedReader;
import java.io.File;
import java.io.FileInputStream;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.Collection;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
import java.util.HashMap;
import python程式碼
import jinja2
import openpyxl
import os
import sys
import json
import collections
import datetime
RESOURCE_PATH = "../resource"
TIME_FORMAT = "%Y-%m-%d %H:%M:%S"
PACKAGE = "mmorpg.server.game.common.conf"
JAVA_FILE_PATH = "../mmorpg/src/mmorpg/server/game/common/conf"
sheetNames = []
javaFiles = []
allFils = []
def allExcelFils(path):
parents = os.listdir(path)
for parent in parents:
childPath = os.path.join(path,parent)
if os.path.isdir(childPath):
allExcelFils(childPath)
elif childPath.endswith(".xlsx") and childPath.find("~") == -1:
allFils.append(childPath)
else:
pass
def getRowContents(row):
contents = []
for cell in row:
if cell.value == None:
print("ERROR TITLE getRowContents:",row)
sys.exit(0)
contents.append(cell.value)
return contents
def getNameAndIndexs(row):
names = []
indexs = []
for cell in row:
if cell.value == None:
print("ERROR TITLE getNameAndIndexs:",row)
sys.exit(0)
pos = str(cell.value).find("-key")
if pos != -1:
names.append(cell.value[0:pos])
indexs.append(cell.value[0:pos])
else:
names.append(cell.value)
return names,indexs
def getIntValue(value):
return int(value)
def getStrValue(value):
if isinstance(value,datetime.datetime):
return datetime.datetime.strftime(value,TIME_FORMAT)
return str(value)
def getBoolValue(value):
return bool(value)
def getFloatValue(value):
return float(value)
def parseCellValue(type,value):
if type in ("byte","short","int","long"):
if value == None or value == "":
return 0
else:
return getIntValue(value)
elif type in ("float","double"):
if value == None:
return 0.0
elif isinstance(value,str):
value = value.replace(" ","")
else:
if value == "":
return 0.0
else:
return getFloatValue(value)
elif type == "String":
if value == None:
return ""
else:
return getStrValue(value)
elif type == "boolean":
if value == None or value == "":
return False
else:
return getBoolValue(value)
else:
print("WARN:unkonw field type !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!")
return value
def generateJSON(sheetName,names,types,rows):
pos = sheetName.find("|")
if pos != -1:
sheetName = sheetName[0:pos]
pos = sheetName.find("-")
if pos != -1:
sheetName = sheetName[0:pos]
pathDir = RESOURCE_PATH+"/conf"
if not os.path.exists(pathDir):
os.makedirs(pathDir)
filePah = pathDir+"/Conf"+sheetName+".json"
print(filePah)
file = open(filePah,"a+",encoding="utf8")
for row in rows:
cells = list(row)
data = collections.OrderedDict()
for index in range(len(names)):
data[names[index]] = parseCellValue(types[index],cells[index].value)
jsonData = json.dumps(data,ensure_ascii=False)
file.write(jsonData+"\n")
file.close()
"""
生成檔案內容
"""
def parseTemplate(templateFile,context):
env = jinja2.Environment(loader=jinja2.FileSystemLoader("config"))
template = env.get_template(templateFile)
return template.render(context)
"""
生成java檔案
"""
def generateJAVA(excelPath,sheetName,names,noteds,indexs,types):
superClassName = ""
pos = sheetName.find("|")
if pos != -1:
superClassName = sheetName[pos+1:len(sheetName)]
sheetName = sheetName[0:pos]
pos = sheetName.find("-")
if pos != -1:
sheetName = sheetName[0:pos]
if sheetName in javaFiles:
return
context = {}
context["package"] = PACKAGE
context["superClass"] = superClassName
context["excelName"] = os.path.basename(excelPath)
context["sheetName"] = sheetName
indexStr = "{"
for ind in range(len(indexs)):
if ind == len(indexs) - 1:
indexStr += "\""+indexs[ind]+"\""
else:
indexStr+= "\""+indexs[ind]+"\","
indexStr += "}"
context["indexs"] = indexStr
fields = []
for index in range(len(names)):
data = {}
data["name"] = names[index]
data["uname"] = names[index][0].upper()+names[index][1:]
data["note"] = noteds[index]
data["type"] = types[index]
data["utype"] = types[index][0].upper()+types[index][1:]
fields.append(data)
context["fileds"] = fields
javaClassContent = parseTemplate("template.html",context)
filePah = JAVA_FILE_PATH+"/Conf"+sheetName+".java"
print(filePah)
file = open(filePah,"a+",encoding="utf8")
file.write(javaClassContent)
file.close()
javaFiles.append(sheetName)
def generate():
for excelPath in allFils:
wb = openpyxl.load_workbook(excelPath,read_only=True,keep_links=False)
for sheet in wb:
if sheet.title.find("策劃") > -1 or sheet.title.startswith("sheet") or sheet.title.startswith("Sheet") or sheet.title.find("coder") > -1 or sheet.title[0].islower():
continue
if sheet.title in sheetNames:
print("相同的sheet名稱："+sheet.title)
sys.exit(0)
sheetNames.append(sheet.title)
rows = list(sheet.rows)
if len(rows) < 3:
continue
#註釋
note_row = rows.pop(0)
field_row = rows.pop(0)
type_row = rows.pop(0)
#名字
names,indexs = getNameAndIndexs(field_row)
if names[0] != "sn":
continue
notes = getRowContents(note_row)
#型別
row_field_types = getRowContents(type_row)
if len(names) != len(row_field_types):
print("excel 資料錯誤："+excelPath)
sys.exit(0)
generateJSON(sheet.title,names,row_field_types,rows)
generateJAVA(excelPath,sheet.title,names,notes,indexs,row_field_types)
def startup():
allExcelFils(RESOURCE_PATH)
generate()
#啟動
if __name__ == '__main__':
startup()