def _addenumvalues(message_descriptor, dictionary):
"""sets class-level attributes for all enum fields defined in this message.
args:
message_descriptor: descriptor object for this message type.
dictionary: class dictionary that should be populated.
"""
for enum_type in message_descriptor.enum_types:
dictionary[enum_type.name] = enum_type_wrapper.enumtypewrapper(enum_type)
for enum_value in enum_type.values:
dictionary[enum_value.name] = enum_value.number
def _addenumvalues(descriptor, cls):
"""sets class-level attributes for all enum fields defined in this message.
also exporting a class-level object that can name enum values.
args:
descriptor: descriptor object for this message type.
cls: class we're constructing for this message type.
"""
for enum_type in descriptor.enum_types:
setattr(cls, enum_type.name, enum_type_wrapper.enumtypewrapper(enum_type))
for enum_value in enum_type.values:
setattr(cls, enum_value.name, enum_value.number)