Question

我有一个为python 2编译的google协议缓冲区，我正在尝试将其移植到python 3.不幸的是，我找不到我用来在任何地方生成编译协议缓冲区的proto文件。我如何恢复proto文件，以便我可以为python 3编译一个新文件。我不知道使用了什么proto版本，我所拥有的是.py文件意味着在python 2.6上运行。

Answer 1

您必须编写代码（例如Python）来遍历消息描述符的树。它们原则上应该包含原始原始文件的完整信息，但代码注释除外。您在posession中生成的Python模块应该允许您将proto文件的文件描述符序列化为文件描述符原型消息，然后可以将其提供给表示为原型代码的代码。

作为指南，您应该查看protoc的各种代码生成器，它们实际上也是这样做的：它们将文件描述符作为protobuf消息读入，分析它并生成代码。

以下是如何使用Python编写Protobuf插件的基本介绍

https://www.expobrain.net/2015/09/13/create-a-plugin-for-google-protocol-buffer/

这是protoc插件的官方列表

https://github.com/google/protobuf/blob/master/docs/third_party.md

这是一个生成LUA代码的protoc插件，用Python编写。

https://github.com/sean-lin/protoc-gen-lua/blob/master/plugin/protoc-gen-lua

我们来看看主代码块

def main():
    plugin_require_bin = sys.stdin.read()
    code_gen_req = plugin_pb2.CodeGeneratorRequest()
    code_gen_req.ParseFromString(plugin_require_bin)

    env = Env()
    for proto_file in code_gen_req.proto_file:
        code_gen_file(proto_file, env,
                      proto_file.name in code_gen_req.file_to_generate)

    code_generated = plugin_pb2.CodeGeneratorResponse()
    for k in  _files:
        file_desc = code_generated.file.add()
        file_desc.name = k
        file_desc.content = _files[k]

    sys.stdout.write(code_generated.SerializeToString())

循环for proto_file in code_gen_req.proto_file:实际上循环遍历文件描述符对象，protoc要求代码生成器插件生成LUA代码。 所以现在你可以这样做：

# This should get you the file descriptor for your proto file
file_descr = your_package_pb2.sometype.GetDescriptor().file
# serialized version of file descriptor
filedescr_msg = file_descr.serialized_pb
# required by lua codegen
env = Env()
# create LUA code -> modify it to create proto code
code_gen_file(filedescr, env, "your_package.proto")

Answer 2

如其他帖子所述，您需要遍历描述符消息树并构建原型文件内容。

您可以在协议缓冲区 github repository 中找到完整的 C++ 示例。以下是链接中的一些 C++ 代码片段，旨在让您了解如何在 Python 中实现这一点：

  // Special case map fields.
  if (is_map()) {
    strings::SubstituteAndAppend(
        &field_type, "map<$0, $1>",
        message_type()->field(0)->FieldTypeNameDebugString(),
        message_type()->field(1)->FieldTypeNameDebugString());
  } else {
    field_type = FieldTypeNameDebugString();
  }

  std::string label = StrCat(kLabelToName[this->label()], " ");

  // Label is omitted for maps, oneof, and plain proto3 fields.
  if (is_map() || containing_oneof() ||
      (is_optional() && !has_optional_keyword())) {
    label.clear();
  }

  SourceLocationCommentPrinter comment_printer(this, prefix,
                                               debug_string_options);
  comment_printer.AddPreComment(contents);

  strings::SubstituteAndAppend(
      contents, "$0$1$2 $3 = $4", prefix, label, field_type,
      type() == TYPE_GROUP ? message_type()->name() : name(), number());

FieldTypeNameDebugString 函数如下所示：

// The field type string used in FieldDescriptor::DebugString()
std::string FieldDescriptor::FieldTypeNameDebugString() const {
  switch (type()) {
    case TYPE_MESSAGE:
      return "." + message_type()->full_name();
    case TYPE_ENUM:
      return "." + enum_type()->full_name();
    default:
      return kTypeToName[type()];
  }
}

如何将编译的协议缓冲区转换回.proto文件？

2 个答案: