在Python中递归添加到字典

Question

我有一个目录结构：

test
    test2 
        test3

并且我正在尝试制作一个看起来像这样的字典：

{'name': 'test', 'sub': [{'name': 'test2', 'sub': [{'name': 'test3', 'sub': []}]}, ]}

为此，我使用以下递归函数：

def dir_struct(rootdir):
    a = {}
    rootdir = rootdir.rstrip(os.sep)
    a["name"] = rootdir
    a["sub"] = []
    for path, dirs, files in os.walk(rootdir):
        for dir in dirs:
            b = dir_struct(dir)
            a["sub"].append(b)
    return a

但是有了它我得到了

{'name': 'test', 'sub': [{'name': 'test2', 'sub': []}, {'name': 'test3', 'sub': []}]}

有没有办法解决这个问题？

Answer 1

问题在于os.walk已经是一个递归函数，它遍历整个目录结构。

一种解决方案是使用os.listdir而不是os.walk 。
这就是您想要的：

def dir_struct(rootdir):
    a = {}
    rootdir = rootdir.rstrip(os.sep)
    a["name"] = rootdir.split(os.sep)[-1]
    a["sub"] = []
    for dir in os.listdir(rootdir):
        b = dir_struct(rootdir + os.sep + dir)
        a["sub"].append(b)
    return a