如何从字符串动态创建Protobuf对象？嵌套消息通用填充方法问询

给定如下Protobuf定义：

message MyProto {
    optional string ip = 1; 
    optional int32 port = 2;
    optional Flags flags = 3;
}

message Flags {
    optional bool flag1 = 1;
    optional bool flag2 = 2;
}

需要从以下结构化字符串生成对应的MyProto对象：

ip:127.0.0.1
port:8080
Flags:
    flag1: True
    flag2: False

已经掌握普通字段的设置方法（比如setattr(my_proto, "ip", "127.0.0.1")），但不知道如何处理嵌套的消息类型，需要一套通用方案，能自动识别并填充任意嵌套的Protobuf消息字段。

实现步骤 #

1. 将结构化字符串解析为嵌套字典

首先把带缩进的字符串转成嵌套字典结构，同时将字符串中的大写键（如Flags）转换为Protobuf定义里的小写字段名（flags），方便后续匹配：

def parse_structured_string(s):
    lines = [line.strip() for line in s.split('\n') if line.strip()]
    result = {}
    current_level = [result]
    indent_level = 0
    for line in lines:
        # 计算当前行缩进量（假设缩进为4个空格）
        current_indent = len(line) - len(line.lstrip())
        # 调整当前层级
        while current_indent <= indent_level:
            current_level.pop()
            indent_level -= 4
        indent_level = current_indent
        key, value = line.split(':', 1)
        key = key.strip().lower()
        value = value.strip()
        if not value:
            # 嵌套消息，创建新字典
            new_dict = {}
            current_level[-1][key] = new_dict
            current_level.append(new_dict)
        else:
            # 转换值类型
            if value.isdigit():
                parsed_value = int(value)
            elif value.lower() in ('true', 'false'):
                parsed_value = value.lower() == 'true'
            else:
                parsed_value = value
            current_level[-1][key] = parsed_value
    return result

2. 递归填充Protobuf对象

编写通用函数，通过Protobuf的DESCRIPTOR判断字段类型，递归处理嵌套消息：

def fill_proto(proto_obj, data_dict):
    for key, value in data_dict.items():
        field_desc = proto_obj.DESCRIPTOR.fields_by_name.get(key)
        if not field_desc:
            continue
        if field_desc.type == field_desc.TYPE_MESSAGE:
            # 嵌套消息类型，创建对应实例并递归填充
            nested_proto = getattr(proto_obj, key)
            fill_proto(nested_proto, value)
        else:
            # 普通字段，按类型转换后赋值
            if field_desc.type in (field_desc.TYPE_INT32, field_desc.TYPE_INT64):
                setattr(proto_obj, key, int(value))
            elif field_desc.type == field_desc.TYPE_BOOL:
                setattr(proto_obj, key, bool(value))
            elif field_desc.type in (field_desc.TYPE_STRING, field_desc.TYPE_BYTES):
                setattr(proto_obj, key, str(value))
            # 其他类型（如float、enum）可按需扩展

使用示例 #

# 导入protoc生成的Protobuf Python代码
from my_proto_pb2 import MyProto

# 目标输入字符串
input_str = """ip:127.0.0.1
port:8080
Flags:
    flag1: True
    flag2: False"""

# 解析字符串为字典
data = parse_structured_string(input_str)
# 创建空的MyProto对象
my_proto = MyProto()
# 填充对象
fill_proto(my_proto, data)

# 验证结果
print(my_proto.ip)          # 输出: 127.0.0.1
print(my_proto.port)        # 输出: 8080
print(my_proto.flags.flag1) # 输出: True
print(my_proto.flags.flag2) # 输出: False

关键细节说明 #

• 缩进适配：代码默认以4个空格为缩进单位，可根据实际字符串格式调整缩进量的计算逻辑；
• 字段匹配：通过Protobuf的DESCRIPTOR获取字段元信息，确保能识别任意嵌套消息类型；
• 类型兼容：针对常见Protobuf字段类型做了转换处理，可根据需求扩展更多类型（如枚举、浮点数等）；
• 通用性：该方案不依赖具体的Protobuf消息定义，只要输入字符串格式统一，就能自动适配任意嵌套结构。

内容的提问来源于stack exchange，提问作者Ames ISU