Как вывести дерево каталогов в формате HTML? - PullRequest
Новая
       39

Как вывести дерево каталогов в формате HTML?

3 голосов
/ 13 сентября 2011

Вот что я получил до сих пор: 

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>')

Это дает мне список каталогов и всех файлов под ними, но я все еще не могу понять, как разделить путь к каталогу, чтобы он тоже был вложенным.

Ответы [ 4 ]

6 голосов
/ 13 сентября 2011

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)
1 голос
/ 13 сентября 2011 
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

Однако, как уже упоминалось, рекурсивное решение ручной работы, которое вызывается для каждой (под) папки, может упростить вашу задачу.

1 голос
/ 13 сентября 2011

Вам нужна рекурсия. В качестве отправной точки: 

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(".")
0 голосов
/ 14 сентября 2011 
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>')
...