How to get top-level protobuf enum value name by number in python?

python protocol-buffers
abyss.7 picture abyss.7 · Jul 16, 2012 · Viewed 41.7k times · Source

For example, I have proto-file File.proto:

enum Test {
  ONE = 1;
  TWO = 2;
}

I generate file File_pb2.py with protoc from File.proto. I want in a python-code get string "ONE" (that corresponds to the name of File_pb2.ONE) by value 1 (that corresponds to the value of File_pb2.ONE) from generated file File_pb2.py without defining my own dictionaries. How can I do that?

Answer

Tisho picture Tisho · Jul 16, 2012

Assuming the generated python is located in File_pb2.py code Try this:

file_pb2._TEST.values_by_number[1].name

In your case, this should give 'ONE'

The reverse is :

file_pb2._TEST.values_by_name['ONE'].number

will give 1.

EDIT: As correctly pointed by @dyoo in the comments, a new method was later introduced in protobuf library:

file_pb2.Test.Name(1)
file_pb2.Test.Value('One')

EDIT: This has changed again in proto3. Now the Name() and Value() methods belong to the EnumTypeWrapper class so they can be accessed like:

file_pb2.Name(1)
file_pb2.Value('One')

Related questions

ImportError: No module named google.protobuf

I am following this guide (https://developers.google.com/protocol-buffers/docs/pythontutorial) and using the exact sample of addressbook.proto. Post the content of compiler generated addressbook_pb2.py file as well. When I run the following simple program, there …

Read More
No module named google.protobuf

I am trying to run Google's deep dream. For some odd reason I keep getting ImportError: No module named google.protobuf after trying to import protobuf. I have installed protobuf using sudo install protobuf. I am running python 2.7 OSX Yosemite 10.10.3. …

Read More
How to assign to repeated field?

I am using protocol buffers in python and I have a Person message repeated uint64 id but when I try to assign a value to it like: person.id = [1, 32, 43432] I get an error: Assigment not allowed for repeated field "id" …

Read More