Get Protobuf option value and others

proto_entity.py

"""
Proto

enum EntityType {
  GEOMETRY = 0;
  PHYSICAL = 1;
}

extend google.protobuf.MessageOptions {
  EntityType entity_type = 100;
}


message Point {
  float x = 1;
  float y = 2;
  float z = 3;
  option (common.entity_type) = GEOMETRY;
}
"""
import proto_pkg

def get_proto_entity_type(message: Message) -> str:
    """ Get the entity_type option declared in the message, if present """
    entity_type_enum_int = message.DESCRIPTOR.GetOptions().Extensions[
        proto_pkg.entity_type
    ]
    return proto_pkg.EntityType.Name(entity_type_enum_int)

  
def get_proto_package(message: Message) -> str:
    """ Get the package name as defined in the proto `package x` directive """
    return message.DESCRIPTOR.file.package


def get_proto_message_field_names(message: Message) -> List[str]:
    """ Get all field names of a message """
    return [f.name for f in message.DESCRIPTOR.fields]


def get_proto_message_name(message: Message) -> str:
    return message.DESCRIPTOR.name