Question

我收到了一个使用Python的Google ProtoBuf，并试图将枚举值与其字符串表示形式进行比较。根据{{3}}和this，我应该可以使用类似enum_values_by_name的内容来获取我需要的信息。但是，所有enum*相关属性都是空的：

>>> type(my_message)
<class 'myObjects_pb2.myObject'>

>>> my_message
# nID: 53564
# nAge: 2
# type: OBJECT_CLASS_SIGN
# view: OBJECT_VIEW_FRONT   

>>> my_message.type
# 1

>>> filter(lambda s: s.startswith('enum'), dir(my_message.DESCRIPTOR))
# ['enum_types', 'enum_types_by_name', 'enum_values_by_name']

>>> my_message.DESCRIPTOR.enum_types
# []

>>> my_message.DESCRIPTOR.enum_types_by_name
# {}

>>> my_message.DESCRIPTOR.enum_values_by_name
# {}

也许这与我的protobuf在许多文件中定义的事实有关，我想要的枚举没有在我导入的主文件中定义（但是用于解码my_message）？

为什么我会收到这些空集合（更重要的是）如何查找有关枚举的信息？

Answer 1

我不知道为什么邮件的DESCRIPTOR包含未填充的枚举属性。（这对我来说似乎是一个错误。）但是，至少有两个解决方案：

1）如果您知道定义枚举的文件的名称，您可以“破解”＃39;通过名称获取枚举值：

# This is the root ProtoBuf definition
from mydir import rootapi_pb2 

# We happen to know that the enums are defined in myenums.proto
enum_file = rootapi_pb2.myenums__pb2 # NOTE: Extra Underscore!
enum_value = getattr(enum_file, 'OBJECT_CLASS_SIGN')

但是，如果你不想依赖这个黑客，你最终可以通过以下方式找到枚举描述符，从而找到名称中的值：

my_message.DESCRIPTOR.fields_by_name['type'].enum_type.values_by_name['OBJECT_CLASS_SIGN'].number

因为它太可怕了，所以它被包装成一个安全，可重复使用的功能：

def enum_value(msg, field_name, enum_name):
    """Return the integer for the enum name of a field,
       or None if the field does not exist, is not an enum, or the
       enum does not have a value with the supplied name."""
    field = msg.DESCRIPTOR.fields_by_name.get(field_name,None)
    if field and field.enum_type:
        enum = field.enum_type.values_by_name.get(enum_name,None)
        return enum and enum.number

print(enum_value(my_message, 'type', 'OBJECT_CLASS_SIGN'))
# 1

Answer 2

用于Python的ProtoBuf非常丑陋...但是无论如何我都必须使用它...

我觉得@Phrogz的功能可以简化一些。这是我附带的功能：

def get_enum_name_by_number(parent, field_name):
    field_value = getattr(parent, field_name)
    return parent.DESCRIPTOR.fields_by_name[field_name].enum_type.values_by_number.get(field_value).name

print(my_message.type)
# 1
print(get_enum_name_by_number(my_message, 'type'))
# OBJECT_CLASS_SIGN

查找Python DESCRIPTOR for ProtoBuf

2 个答案: