Question

这是我到目前为止所得到的：

project_dir = '/my/project/dir'
project_depth = len(project_dir.split(os.path.sep))

xml_files = []
for dirpath, dirnames, filenames in os.walk(project_dir):
    for filename in fnmatch.filter(filenames, '*.xml'):
        dirs = dirpath.split(os.path.sep)[project_depth:]
        print(dirs)
        xml_files.append(os.path.join(dirpath,filename))

基本上我想要做的是吐出我的项目目录结构，将所有XML文件作为HTML树（使用<ul>）。我可以通过这种方式获取所有文件，但我似乎无法弄清楚如何将它们组织到树中。

通过这个os.walk的工作方式，我不知道我什么时候进入更深层，或者我还在遍历同一个目录。

for dirpath, dirnames, filenames in os.walk(project_dir):
    xml_files = fnmatch.filter(filenames, '*.xml')
    if len(xml_files) > 0:
        out.write('<li>{0}<ul>'.format(dirpath))
        for f in xml_files:
            out.write('<li>{0}</li>'.format(f))
        out.write('</ul></li>')
out.write('</ul>')

这给了我一个目录列表以及它们下面的所有文件，但我仍然无法弄清楚如何拆分目录路径以便它也嵌套。

Answer 1

如果您关心层次结构，

os.walk可能不是最佳解决方案。一个更简单的解决方案可能就是使用os.listdir和os.path.isdir递归遍历树。

import os

def traverse(dir):
    print '<ul>'
    for item in os.listdir(dir):
        print '<li>%s</li>' % item
        fullpath = os.path.join(dir, item)
        if os.path.isdir(fullpath):
            traverse(fullpath)
    print '</ul>'

projectdir = '.'
traverse(projectdir)

Answer 2

你需要递归。作为一个起点：

import os

def walk(d, ident=""):
    print "<ul>"
    for p in os.listdir(d):
        fullpath = os.path.join(d, p)
        print ident, "<li>",p,"</li>"
        if os.path.isdir(fullpath):
            walk(fullpath, ident+"   ")
    print "</ul>"

walk(".")

Answer 3

import os

tmpold=[]
for (f, fol, fil) in os.walk("./mydir"):
    tmp = f.split("/")
    if len(tmp)>len(tmpold):
        print "<ul>\n<li>" + tmp[-1] + "</li>"
    elif len(tmp)==len(tmpold):
        print "<li>" + tmp[-1] + "</li>"
    else:
        print "</ul>\n"*(1+len(tmpold)-len(tmp)) + "<ul>\n<li>" + tmp[-1] + "</li>"
    tmpold = tmp

但是，正如其他人所提到的，为每个（子）文件夹调用的手工递归解决方案可以简化您的任务。

Answer 4

def buildtree(dir):
    tree = []
    for item in os.listdir(dir):
        path = os.path.join(dir, item)
        if os.path.isdir(path):
            subtree = buildtree(path)
            if len(subtree) > 0:
                tree.append((item,subtree))
        elif item.endswith('.xml'):
            tree.append(item)
    return tree

def writetree(tree, fp):
    fp.write('<ul>')
    for n in tree:
        fp.write('<li>')
        if isinstance(n,tuple):
            fp.write(n[0])
            writetree(n[1],fp)
        else:
            fp.write(n)
        fp.write('</li>')
    fp.write('</ul>')

如何将目录树输出为HTML？

4 个答案: