使用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