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) |