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

使用IntelliJ IDEA编辑器编写Lua脚本的时候,安装一个插件 “EmmyLua” 可以对有代码提示功能
想把protoBuf协议文件转成Lua代码,然后给出代码提示

第一次使用python脚本
使用到的几个功能记录一下

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