在 Python 中循环 Protocol Buffers 属性

Question

我需要递归循环协议缓冲区消息中包含的所有属性/子对象的帮助，假设我们不知道它们的名称或有多少个。

作为示例，请从 google 网站上的教程中获取以下 .proto 文件：

  message Person {
    required string name = 1;
    required int32 id = 2;
    optional string email = 3;

  enum PhoneType {
    MOBILE = 0;
    HOME = 1;
    WORK = 2;
  }

  message PhoneNumber {
    required string number = 1;
    optional PhoneType type = 2 [default = HOME];
  }

  repeated PhoneNumber phone = 4;
}

并使用它...:

person = tutorial.Person()
person.id = 1234
person.name = "John Doe"
person.email = "[email protected]"
phone = person.phone.add()
phone.number = "555-4321"
phone.type = tutorial.Person.HOME

给定

Person

，如何访问每个元素的属性名称及其值：

person.id

、

person.name

、

person.email

、

person.phone.number

、

person.phone.type

？

我已经尝试了以下方法，但它似乎没有重复出现

person.phone.number

或

person.phone.type

。

object_of_interest = Person

while( hasattr(object_of_interest, "_fields") ):
  for obj in object_of_interest._fields:
    # Do_something_with_object(obj) # eg print obj.name
    object_of_interest = obj

我尝试使用

obj.DESCRIPTOR.fields_by_name.keys

访问子元素，但这些是子对象的字符串表示形式，而不是对象本身。

obj.name 为我提供了名称的属性，但我不确定如何实际获取该属性的值，例如 obj.name 可能会给我“名称”，但我如何从中获取“john doe”？

Answer 1

我对 protobuf 不是很熟悉，所以很可能有一种更简单的方法或 api 来处理这种事情。但是，下面显示了如何迭代/内省和对象字段并将其打印出来的示例。希望至少足以让您朝着正确的方向前进......

import addressbook_pb2 as addressbook

person = addressbook.Person(id=1234, name="John Doe", email="[email protected]")
person.phone.add(number="1234567890")


def dump_object(obj):
    for descriptor in obj.DESCRIPTOR.fields:
        value = getattr(obj, descriptor.name)
        if descriptor.type == descriptor.TYPE_MESSAGE:
            if descriptor.label == descriptor.LABEL_REPEATED:
                map(dump_object, value)
            else:
                dump_object(value)
        elif descriptor.type == descriptor.TYPE_ENUM:
            enum_name = descriptor.enum_type.values[value].name
            print "%s: %s" % (descriptor.full_name, enum_name)
        else:
            print "%s: %s" % (descriptor.full_name, value)

dump_object(person)

输出

tutorial.Person.name: John Doe
tutorial.Person.id: 1234
tutorial.Person.email: [email protected]
tutorial.Person.PhoneNumber.number: 1234567890
tutorial.Person.PhoneNumber.type: HOME

Answer 2

如果您想获取底层原型类中定义的所有字段（而不仅仅是给定实例中设置的字段），您可以执行以下操作：

def print_fields(message_desc):
  for field in message_desc.fields:
    if field.type == field.TYPE_MESSAGE:
      print_fields(field.message_type)
    else:
      print(field.full_name)

然后在消息的描述符上调用它，例如，

print_fields(tutorial.Person.DESCRIPTOR)

。