Commit 7d9c88d8f93d1603e072783701709d78a8b0ed8d
1 parent
ce072bf5
del generator.py
Showing
1 changed file
with
0 additions
and
117 deletions
Show diff stats
generator.py deleted
| @@ -1,117 +0,0 @@ | @@ -1,117 +0,0 @@ | ||
| 1 | -import os | ||
| 2 | - | ||
| 3 | -ProtoCodeGO = './cmd/gameserver/action/protocode.go' | ||
| 4 | -ProtoCodeGOTest = './cmd/test/action/protocode.go' | ||
| 5 | -ProtoCodeFile = './protos/protocode.proto' | ||
| 6 | -ProtoFileDir = "./protos" | ||
| 7 | - | ||
| 8 | -ProtoCodeStr = "syntax = \"proto3\";\noption go_package = \"../pb;pb\";\n\npackage protocode;\n\nenum ProtoCode\n{{\n " \ | ||
| 9 | - "UNKNOWN = 0;\n {}\n}} " | ||
| 10 | -ProtoCodeLineReq = "\t{}Rpc = {};\n" | ||
| 11 | -ProtoCodeLineRsp = "\t{}Rsp = {};\n" | ||
| 12 | -ProtoCodeLineNty = "\t{}Nty = {};\n" | ||
| 13 | - | ||
| 14 | -GoProtoCodeStr = "package action\n\nimport (\n\t\"pro2d/common/logger\"\n\t\"pro2d/pb\"\n)\n\nfunc GetActionMap() " \ | ||
| 15 | - "map[interface{{}}]interface{{}} {{\n\tlogger.Debug(\"init protocode...\")\n\tam := make(map[interface{{" \ | ||
| 16 | - "}}]interface{{}})\n{}\n\treturn am\n}}" | ||
| 17 | -GoProtoCodeLine = "\tam[uint32(pb.ProtoCode_{}Rpc)] = {}Rpc\n" | ||
| 18 | - | ||
| 19 | -GoProtoCodeTestStr = "package action\n\nimport (\n\t\"pro2d/pb\"\n)\n\nfunc GetTestActionMap() " \ | ||
| 20 | - "map[interface{{}}]interface{{}} {{\n\tam := make(map[interface{{" \ | ||
| 21 | - "}}]interface{{}})\n{}\n\treturn am\n}}" | ||
| 22 | -GoProtoCodeTestReqLine = "\tam[uint32(pb.ProtoCode_{}Rpc)] = {}Rsp\n" | ||
| 23 | -GoProtoCodeTestRspLine = "\tam[uint32(pb.ProtoCode_{}Rsp)] = {}Rsp\n" | ||
| 24 | -GoProtoCodeTestNtyLine = "\tam[uint32(pb.ProtoCode_{}Nty)] = {}Nty\n" | ||
| 25 | - | ||
| 26 | -def generatorProto(path): | ||
| 27 | - RpcStartCode = 500 | ||
| 28 | - NtyStartCode = 1000 | ||
| 29 | - | ||
| 30 | - files = os.listdir(path) | ||
| 31 | - ProtoCodeData = "" | ||
| 32 | - GoCodeData = "" | ||
| 33 | - GoCodeTestData = "" | ||
| 34 | - for file in files: | ||
| 35 | - if file.find("account.proto") != -1: | ||
| 36 | - continue | ||
| 37 | - | ||
| 38 | - if os.path.isdir(file): | ||
| 39 | - continue | ||
| 40 | - | ||
| 41 | - with open(path + "/" + file, 'r', encoding='utf-8', errors='ignore') as f: | ||
| 42 | - lines = f.readlines() # 读取所有行 | ||
| 43 | - firstline = lines[0] | ||
| 44 | - if firstline.find("proto3") == -1: | ||
| 45 | - continue | ||
| 46 | - | ||
| 47 | - # req + rar | ||
| 48 | - for line in lines: | ||
| 49 | - if line.find("message") == -1: | ||
| 50 | - continue | ||
| 51 | - sline = line.split(' ') | ||
| 52 | - if len(sline) < 2: | ||
| 53 | - continue | ||
| 54 | - | ||
| 55 | - messageStr = sline[1].replace('\n', '').replace('{', "") | ||
| 56 | - n1 = messageStr.find('Req') | ||
| 57 | - n2 = messageStr.find('Rar') | ||
| 58 | - loginReq = messageStr.find('LoginReq') | ||
| 59 | - | ||
| 60 | - if n1 != -1: | ||
| 61 | - RpcStartCode = RpcStartCode + 1 | ||
| 62 | - ProtoCodeData += ProtoCodeLineReq.format(messageStr[:n1], RpcStartCode) | ||
| 63 | - if loginReq != -1: | ||
| 64 | - continue | ||
| 65 | - GoCodeData += GoProtoCodeLine.format(messageStr[:n1], messageStr[:n1]) | ||
| 66 | - GoCodeTestData += GoProtoCodeTestReqLine.format(messageStr[:n1], messageStr[:n1]) | ||
| 67 | - elif n2 != -1: | ||
| 68 | - RpcStartCode = RpcStartCode + 1 | ||
| 69 | - ProtoCodeData += ProtoCodeLineReq.format(messageStr[:n2], RpcStartCode) | ||
| 70 | - if loginReq != -1: | ||
| 71 | - continue | ||
| 72 | - GoCodeData += GoProtoCodeLine.format(messageStr[:n2], messageStr[:n2]) | ||
| 73 | - GoCodeTestData += GoProtoCodeTestReqLine.format(messageStr[:n2], messageStr[:n2]) | ||
| 74 | - ProtoCodeData += '\n' | ||
| 75 | - # nty | ||
| 76 | - for line in lines: | ||
| 77 | - if line.find("message") == -1: | ||
| 78 | - continue | ||
| 79 | - sline = line.split(' ') | ||
| 80 | - if len(sline) < 2: | ||
| 81 | - continue | ||
| 82 | - | ||
| 83 | - messageStr = sline[1].replace('\n', '').replace('{', "") | ||
| 84 | - n3 = messageStr.find('Nty') | ||
| 85 | - loginReq = messageStr.find('LoginReq') | ||
| 86 | - | ||
| 87 | - if n3 != -1: | ||
| 88 | - NtyStartCode = NtyStartCode + 1 | ||
| 89 | - ProtoCodeData += ProtoCodeLineNty.format(messageStr[:n3], NtyStartCode) | ||
| 90 | - if loginReq != -1: | ||
| 91 | - continue | ||
| 92 | - GoCodeTestData += GoProtoCodeTestNtyLine.format(messageStr[:n3], messageStr[:n3]) | ||
| 93 | - | ||
| 94 | - | ||
| 95 | - | ||
| 96 | - | ||
| 97 | - # protocode.go | ||
| 98 | - gostr = GoProtoCodeStr.format(GoCodeData) | ||
| 99 | - fo = open(ProtoCodeGO, "w") | ||
| 100 | - fo.write(gostr) | ||
| 101 | - fo.close() | ||
| 102 | - | ||
| 103 | - # protocode.go | ||
| 104 | - gostr = GoProtoCodeTestStr.format(GoCodeTestData) | ||
| 105 | - fo = open(ProtoCodeGOTest, "w") | ||
| 106 | - fo.write(gostr) | ||
| 107 | - fo.close() | ||
| 108 | - | ||
| 109 | - #protocode.proto | ||
| 110 | - protostr = ProtoCodeStr.format(ProtoCodeData) | ||
| 111 | - fo = open(ProtoCodeFile, "w") | ||
| 112 | - fo.write(protostr) | ||
| 113 | - fo.close() | ||
| 114 | - | ||
| 115 | - | ||
| 116 | -if __name__ == "__main__": | ||
| 117 | - generatorProto(ProtoFileDir) |