Function _parse

proto_edit.py:86–120

(buf: bytes, start: int, end: int)

def _parse(buf: bytes, start: int, end: int) -> Message:
    msg = Message()
    pos = start
    while pos < end:
        tag, pos = _read_varint(buf, pos)
        wt = tag & 7
        field_no = tag >> 3
        if wt == WT_VARINT:
            val, pos = _read_varint(buf, pos)
            msg.fields.append(Field(field_no, wt, val))
        elif wt == WT_I64:
            v = buf[pos : pos + 8]
            pos += 8
            msg.fields.append(Field(field_no, wt, v))
        elif wt == WT_I32:
            v = buf[pos : pos + 4]
            pos += 4
            msg.fields.append(Field(field_no, wt, v))
        elif wt == WT_LEN:
            length, pos = _read_varint(buf, pos)
            data = buf[pos : pos + length]
            pos += length
            child: Union[bytes, Message]
            if _looks_like_msg(data):
                try:
                    child = _parse(data, 0, len(data))
                except Exception:
                    child = data
            else:
                child = data
            msg.fields.append(Field(field_no, wt, child))
        else:
            # SGROUP/EGROUP or unknown: bail out silently with what we have so far
            break
    return msg


def _looks_like_msg(buf: bytes) -> bool: