Jak uzyskać nazwę najwyższego poziomu protobuf enum według numeru w pythonie?

Question

Na przykład, mieć proto plik File.proto:

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

wygenerować plik File_pb2.py z protoc z File.proto. Chcę w kodzie Pythona uzyskać ciąg "ONE" (który odpowiada nazwie File_pb2.ONE) o wartość 1 (która odpowiada wartości File_pb2.ONE) z wygenerowanego pliku File_pb2.py bez definiowania własnych słowników. Jak mogę to zrobić?

Answer 1

Zakładając wygenerowany Pythona znajduje się w kodzie File_pb2.py Spróbuj tego:

file_pb2._TEST.values_by_number[1].name 

W twoim przypadku, to powinien dać 'ONE'

Odwrotna jest:

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

da 1.

EDIT: Jak słusznie zauważył przez @dyoo w komentarzach, nowa metoda została później wprowadzona w Protobuf biblioteki:

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

EDIT: To po raz kolejny zmieniła się w proto3. Teraz nazwa() and Value() metody należą do klasy EnumTypeWrapper dzięki czemu mogą one być dostępne jak:

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