[转]用Python做一个自动生成读表代码的小脚本

写在开始（本片文章不是写给小白的，至少你应该知道一些常识！）

大家在Unity开发中，肯定会把一些数据放到配置文件中，尤其是大一点的项目，每次开发一个新功能的时候，都要重复的写那些读表代码。非常烦。来个实用小工具，大家随便看看。

 #-*- coding: utf-8 -*-
 #----------------------------------------------------------#
 #   版本：python-3.5.0a3-amd64
 #   功能：生成读表代码文件
 #----------------------------------------------------------#
 import os
 import sys
 import re
 import string
 import codecs

 templateFileName = "./template.cs"
 outputPathName = "./"

 def InputFileName2RowClassName(_inputFileName):
     _className = ""
     _className = "Row_" + _inputFileName.split(".")[0]
     return _className

 def InputFileName2TableClassName(_inputFileName):
     _className = ""
     _className = "Table_" + _inputFileName.split(".")[0]
     return _className

 def InputFileName2FileName(_inputFileName):
     _className = ""
     _className = _inputFileName.split(".")[0]
     return _className

 def GenerateRowFields(_type,_name,_desc):
     _returnValue = ""
     _returnValue += "\t//" + _desc + "\n"
     if _type.find("FLOAT")!=-1:
         _returnValue += "\tprivate float m_float" + _name + ";\n"
         _returnValue += "\tpublic float " + _name + "{get{return m_float" + _name + ";}}\n"
     if _type.find("STRING")!=-1:
         _returnValue += "\tprivate string m_string" + _name + ";\n"
         _returnValue += "\tpublic string " + _name + "{get{return m_string" + _name + ";}}\n"
     if _type.find("INT")!=-1:
         _returnValue += "\tprivate int m_int" + _name + ";\n"
         _returnValue += "\tpublic int " + _name + "{get{return m_int" + _name + ";}}\n"
     return _returnValue

 def GenerateRowFunction(_type,_name,_desc,_i):
     _returnValue = ""
     if _type.find("FLOAT")!=-1:
         _returnValue = "\t\t\tm_float" + _name + " = float.Parse(strCols[i++]);\n"
     if _type.find("STRING")!=-1:
         _returnValue = "\t\t\tm_string" + _name + " = strCols[i++];\n"
     if _type.find("INT")!=-1:
         _returnValue = "\t\t\tm_int" + _name + " = Convert.ToInt32(strCols[i++]);\n"
     #print(_returnValue)
     return _returnValue

 def GenerateFileds(_inputfilename):
     outputText = "";
     lineNum = 0
     listTypes = []
     listNames = []
     listDescs = []
     textFile = open( _inputfilename, "r", encoding= 'utf-8' )
     for line in textFile.readlines():
         lineNum = lineNum + 1
         #只读前三行
         if lineNum <= 3:
             #第一行类型
             if lineNum == 1:
                 listTypes = line.strip().split("\t")
             #第二行名称
             if lineNum == 2:
                 listNames = line.strip().split("\t")
             #第三行注释
             if lineNum == 3:
                 listDescs = line.strip().split("\t")
     textFile.close()
     for i in range( 0, len(listTypes) ):
         outputText += GenerateRowFields(listTypes[i],listNames[i],listDescs[i])
     return outputText

 def GenerateFromText(_inputfilename):
     outputText = "";
     lineNum = 0
     listTypes = []
     listNames = []
     listDescs = []
     textFile = open( _inputfilename, "r", encoding= 'utf-8' )
     for line in textFile.readlines():
         lineNum = lineNum + 1
         #只读前三行
         if lineNum <= 3:
             #第一行类型
             if lineNum == 1:
                 listTypes = line.strip().split("\t")
             #第二行名称
             if lineNum == 2:
                 listNames = line.strip().split("\t")
             #第三行注释
             if lineNum == 3:
                 listDescs = line.strip().split("\t")
     textFile.close()
     for j in range( 0, len(listTypes) ):
         #print(listTypes[j],listNames[j],listDescs[j],j)
         outputText += GenerateRowFunction(listTypes[j],listNames[j],listDescs[j],j)
     return outputText

 #根据TXT生成客户端代码文件
 def GenarateCodeByText(_inputfilename):
     #templateFileName = "template.cs"
     templateFile = open( templateFileName, "r" )
     templateText = templateFile.read()
     templateText = templateText.replace( "#{RowClass}", InputFileName2RowClassName(_inputfilename) )
     templateText = templateText.replace( "#{Fileds}", GenerateFileds(_inputfilename) )
     templateText = templateText.replace( "#{FromText}", GenerateFromText(_inputfilename) )
     templateText = templateText.replace( "#{TableClass}", InputFileName2TableClassName(_inputfilename) )
     templateText = templateText.replace( "#{FileName}", InputFileName2FileName(_inputfilename) )
     templateFile.close()
     return templateText

 #自动写文件
 def AutoWriteToFile(_inputfilename):
     outputFileName = outputPathName + InputFileName2TableClassName(_inputfilename) + ".cs"
     outputText = GenarateCodeByText(_inputfilename).encode("utf-8")
     outputFile = open( outputFileName, "wb" )
     outputFile.write(outputText)
     outputFile.close()

 #获取文件夹下所有文件路径
 def walk(path):
     f1 = os.listdir(path)
     for f in f1:
         #如果是子目录，则递归遍历
         if f.find(".svn") == -1 and os.path.isdir(os.path.join(path,f)):
             #walk(os.path.join(path,f))
             None
         else:
             if f.find(".txt") != -1:
                 t_name = os.path.join(path,f)
                 listfilepath.append(t_name)
     return listfilepath

 listfilepath = []

 if __name__ == '__main__':
     if len(sys.argv) == 1:
         listfilepath = walk("./")
         print(len(listfilepath))
         for files in listfilepath:
             files = files.split("/")[-1]
             print(files)
             AutoWriteToFile(files)
     sys.exit(0)

伸手党请看下面，大神们误喷，请绕行。

code

• 本文固定链接: http://4past.net/?p=48
• 转载请注明: X先生 2015年09月02日 于 X先生 发表