初次使用Python脚本,proto协议文件转Lua代码

1.设置为文件格式备注为utf-8

#!/usr/bin/python
# -*- coding: UTF-8 -*-

2.定义一个方法,可以返回多个值

#--------------------------------------------
# 替换字符串方法
#--------------------------------------------
#define a function
def ReplaceStr(s):
#TODO ... (自己的逻辑)
return s

#备注:方法可以返回多个值
def MyFunc(s):
local s1 = "我是s1"
return s,s1

#调用时,用两个变量接
rtS,rtS1 = MyFunc(str)

2.替换字符串

s = s.replace('=', ' ');    #先把所有的'='替换成'空格'

3.用正则替换任意个数的’空格’换成’-‘

s = re.sub(r'\s+', '-', s); #把任意多个'空格'替换成'-'

4.把字符串分割成数组

arrSplit = s.split('-') #把"1-2-3" 分割成 [1,2,3]

5.逻辑判断 if-else 和 与或非

typeValue = ""
if s == "int32":
typeValue = "0"
elif s == "int64":
typeValue = "0"
else:
typeValue = "{}"

# 判断符号
# 等于'=='

6.把字符串分割成数组

arrSplit = s.split('-') #把"1-2-3" 分割成 [1,2,3]

7.读文件

fileRead = open("client2server.proto")             # 返回一个文件对象
line = fileRead.readline()             # 调用文件的 readline()方法
while line:
print line
# 读取下一行,while循环,直到读完最后一行
line = fileRead.readline()

8.写文件

fileWrite = open('LuaProto.lua', 'w')
fileWrite.write(allStr)
fileWrite.close( )
print "文件保存成功,当前目录下的LuaProto.lua"

Proto文件

// msgType = 1
// 客户端 -> 服务器
// 发送登录消息，包括账号和密码
message Login {
required string user = 1;
required string pwd = 2;
}

message LoginRt {
required int32 rt = 1;
required string token = 2;
}

python脚本

#!/usr/bin/python
# -*- coding: UTF-8 -*-

import re

#--------------------------------------------
# 替换字符串方法
#--------------------------------------------
#define a function
def ReplaceStr(s):
s = s.replace('=', ' ');    #先把所有的'='替换成'空格'
s = s.replace('=', ' ');    #先把所有的'='替换成'空格'
s = s.replace('{', ' ');    #先把所有的'{'替换成'空格'
s = s.replace('\t', ' ');   #先把所有的'\t'替换成'空格'
s = re.sub(r'\s+', '-', s); #把任意多个'空格'替换成'-'
return s

#--------------------------------------------
# 检测类型赋值
#--------------------------------------------
#define a function
def GetTypeValue(s):
typeValue = ""
if s == "int32":
typeValue = "0"
elif s == "int64":
typeValue = "0"
elif s == "string":
typeValue = "\"\""
elif s == "string":
typeValue = "\"\""
else:
typeValue = "{}"
return typeValue

#--------------------------------------------
# 检测类型替换后的字符串
#--------------------------------------------
#define a function
def CheckType(s,clsStr):
rtStr = ""
arrSplit = s.split('-')
newCls = ""
if arrSplit[0] == "message":
newCls = arrSplit[1]
rtStr += "\n---@class " + newCls + '\n'
rtStr += newCls + " = {}"
rtStr += '\n'
elif (len(arrSplit) > 3) and (arrSplit[1] == "required" or arrSplit[1] == "repeated" or arrSplit[1] == "optional"):
rtStr +=  clsStr + "." + arrSplit[3] + " = " + GetTypeValue(arrSplit[2])
rtStr += '\n'
#elif arrSplit[0] == '}':
#rtStr += '}'
return rtStr, newCls

#--------------------------------------------
# 读取文件
#--------------------------------------------
allStr = ""    #拼接的字符串,用来写文件
lastClass = "" #记录上一个ClassName
fileRead = open("client2server.proto")             # 返回一个文件对象
line = fileRead.readline()             # 调用文件的 readline()方法
while line:
rptStr = ""
rptStr = ReplaceStr(line)          #自己写的替换字符串
rptStr,tmpCls = CheckType(rptStr,lastClass)
if tmpCls != "":
lastClass = tmpCls

allStr += rptStr
line = fileRead.readline()

#--------------------------------------------
# 写文件
#--------------------------------------------
fileWrite = open('LuaProto.lua', 'w')
fileWrite.write(allStr)
fileWrite.close( )
print "文件保存成功,当前目录下的LuaProto.lua"

生成后的Lua文件

---@class Login
Login = {}
Login.user= ""
Login.pwd = ""

---@class LoginRt
LoginRt = {}
LoginRt.rt = 0
LoginRt.token = ""

API 参考

菜鸟教程 Python 基础教程 http://www.runoob.com/python/python-tutorial.html[/code]