Python-从文件夹目录创建字典

Question

我正在尝试解决一个问题，但有点令人困惑。

我想要做的是创建一个字典对象，其中包含目录文件夹及其文件作为子，如下所示:

vm.folder = {
    id: 'root',
    name: 'Root',
    type: "folder",
    children: [
        {
            id: "Folder 1",
            name: "1",
            type: "folder",
            children: [
                {
                    id: "Folder 1a",
                    name: "1a",
                    type: "folder",
                    children: [
                        {
                            id: "1a1",
                            name: "1a1",
                            type: "file"
                        },
                        {
                            id: "1a2",
                            name: "1a2",
                            type: "file"
                        }
                    ]
                }
            ]
        }
    ]
}

我这里的代码只是获取目录名和文件名，将它们作为键和值分配给字典:

def pathto_dict(path):
    file_token = ''
    for root, dirs, files in os.walk(path):
        tree = {dir: pathto_dict(os.path.join(root, dir)) for dir in dirs}
        tree.update({file: file_token for file in files})
        return tree

Answer 1

尝试附加的代码。它不包含您指定的 id 属性，但可以轻松地将其添加到代码的 tree = {} 部分中。

def pathto_dict(path):
    for root, dirs, files in os.walk(path_):
        tree = {"name": root, "type":"folder", "children":[]}
        tree["children"].extend([pathto_dict(os.path.join(root, d)) for d in dirs])
        tree["children"].extend([{"name":os.path.join(root, f), "type":"file"} for f in files])
        return tree