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