将文本（或 ASCII）目录树解析为 Python 对象

Question

这或多或少是与List directory tree structure in python? 和这个答案https://stackoverflow.com/a/51991554/7422721 的反向问题

鉴于我们有文本艺术风格的目录树（ASCII，但不限于），作为tree 程序或等效 Python 脚本的输出生成：

a/
├── b/
│   └── c/
│       ├── file_1
│       └── file_2
└── d/

每个文件名都以'/', '=', '*', '@', '|' or '>' 之一结束，指示文件类型，与ls -F 或tree -F 约定一样。所以以/结尾的文件名是目录，=是socket等。如果没有结尾，我们假设它是普通文件。

如何将其读回本机数据结构，例如：

[
  "a": [
    "b": [
      "c": [
        "file_1", "file_2"
      ]
    ],
    "d": []
  ]
]  

或者其他可以在内存文件系统中表示的对象？

这里的目标是为操作磁盘上文件的脚本创建可读的单元测试。

Answer 1

我能够将示例树从您的问题转换为 dict 格式：

{'a': {'b': {'c': {'file_1': {}, 'file_2': {}}}, 'd': {}}}

脚本：

import re

def find_children(tree):
    result = {}
    for i, line in enumerate(tree.split('\n')):
        child = re.findall('^\w+', line)
        if child:
            parent = child[0]
            child_tree = '\n'.join([x[len(parent)+3:] for x in tree.split('\n')[i+1:]])
            result[parent] = find_children(child_tree)
    return result

with open('tree.txt', 'r', encoding='utf-8') as f:
    tree_dict = find_children(f.read())