[Protobuf] PB写入数据读出数据实践操作

以你之姓@ 2021-12-11 15:45 262阅读 0赞

以protobuf中的map对象为例，重点介绍写入和读数据

# Proto文件内容： #

syntax = "proto3";
    message HelloRequired {
    
        map<string, string> args = 1;
    
    }

# 写数据 #

from protobuf import map_pb2
    def write_test():
        map_pb = map_pb2.HelloRequired()
        map_file = "./data/map.pb"
        try:
          f = open(map_file, "rb")
          map_pb.ParseFromString(f.read())
          f.close()
        except IOError:
          print(map_file + ": Could not open file. Creating a new one.")
        # PromptForMap(map_pb.args.update())
        map_pb.args["key3"] = "**集团"
        map_pb.args["key4"] = "ay"
        f = open(map_file, "wb")
        f.write(map_pb.SerializeToString())
        f.close()
    
    
    if __name__ == "__main__":
        write_test()

# 读数据 #

class ProtoBuffDecode(object):
        def __init__(self, pb_py_path_str, pb_path_file, fun_name):
            self.pb_py_path_str = pb_py_path_str
            self.pb_path_file = pb_path_file
            self.fun_name = fun_name
    
        def pb_read(self):
            fun_name = self.fun_name
            pb_py_path_str = self.pb_py_path_str
            py_file = "." + pb_py_path_str
            import importlib
            pb_mode = importlib.import_module(py_file, 'protobuf')
            pb_class = getattr(pb_mode, fun_name)()
            print("pb_class:", type(pb_class))
            pb_file = "./data/" + self.pb_path_file
            # print(pb_file)
            pb_file_open = open(pb_file, "rb")
            pb_class.ParseFromString(pb_file_open.read())
            pb_file_open.close()
            # self.list_pb(pb_class)
            # event = protobuf_to_dict(pb_class)
            # print(event)
            event2 = self.list_pb(pb_class)
    
        def list_pb(self, target_pb):
            for i in target_pb.args:
                print(i)
    
    
    if __name__ == "__main__":
        test = ProtoBuffDecode("map_pb2","map.pb","HelloRequired")
        test.pb_read()