python 生成 树状结构

Posted 2021-11-28 520zm

树状结构: 字典里只有一个键值对, key 为根, 值为一个列表, 列表里的某个或多个元素可以再进行分支(分支还是列表)

比如: 邮件的发件人, 收件人, 转发关系树状结构

forwarding_recp = re.compile(".*?发件人:(.*?);.*?发送时间:.*?收件人:(.*?)主题", re.S)

def parse_addr(addr, split_str=‘,‘):
    l = []
    if addr:
        for a in addr.split(split_str):
            l.append(utils.parseaddr(a)[1])
    return l

def gen_forwarding(con):  # 参数是解析过后的邮件正文
    re_ret = forwarding_recp.findall(con)
    forwarding = dict()
    for i in re_ret:
        from_addr = parse_addr(i[0])[0]
        to_addr = parse_addr(i[1], ";")
        del to_addr[-1]

        for k, v in forwarding.items():
            forwarding.clear()
            to_addr.remove(k)
            to_addr.append({k: v})
            forwarding[from_addr] = to_addr
        forwarding[from_addr] = to_addr
    return json.dumps(forwarding)