Как сгенерировать список каталогов html с помощью Python

У меня возникают проблемы с использованием Python для создания html-документа. Я пытаюсь создать список HTML дерева каталогов. Это то, что у меня есть до сих пор:

def list_files(startpath):
    for root, dirs, files in os.walk(startpath):
        level = root.replace(startpath, '').count(os.sep)
        if level <= 1:
            print('<li>{}<ul>'.format(os.path.basename(root)))
        else:
            print('<li>{}'.format(os.path.basename(root)))
        for f in files:
            last_file = len(files)-1
            if f == files[last_file]:
                print('<li>{}</li></ul>'.format(f))
            elif f == files[0] and level-1 > 0:
                print('<ul><li>{}</li>'.format(f))
            else:
                print('<li>{}</li>'.format(f))
    print('</li></ul>')

Кажется, что это хорошо работает, если есть только корневой каталог, один уровень подкаталогов и файлов. Однако добавление другого уровня подкаталогов вызывает проблемы (поскольку тег закрытия не вводится достаточно времени в конце, я думаю). Но я с трудом обнимаю его.

Если это невозможно сделать, есть ли более простой способ сделать это? Я использую Flask, но я очень неопытен с шаблонами, поэтому, возможно, я что-то пропустил.

Ответы

Ответ 1

Вы можете разделить создание дерева каталогов и его рендеринг как html.

Чтобы сгенерировать дерево, вы можете использовать простую рекурсивную функцию:

def make_tree(path):
    tree = dict(name=os.path.basename(path), children=[])
    try: lst = os.listdir(path)
    except OSError:
        pass #ignore errors
    else:
        for name in lst:
            fn = os.path.join(path, name)
            if os.path.isdir(fn):
                tree['children'].append(make_tree(fn))
            else:
                tree['children'].append(dict(name=name))
    return tree

Чтобы отобразить его как html, вы можете использовать функцию jinja2 loop recursive:

<!doctype html>
<title>Path: {{ tree.name }}</title>
<h1>{{ tree.name }}</h1>
<ul>
{%- for item in tree.children recursive %}
    <li>{{ item.name }}
    {%- if item.children -%}
        <ul>{{ loop(item.children) }}</ul>
    {%- endif %}</li>
{%- endfor %}
</ul>

Поместите html в файл templates/dirtree.html. Чтобы проверить его, запустите следующий код и зайдите в http://localhost:8888/:

import os
from flask import Flask, render_template

app = Flask(__name__)

@app.route('/')
def dirtree():
    path = os.path.expanduser(u'~')
    return render_template('dirtree.html', tree=make_tree(path))

if __name__=="__main__":
    app.run(host='localhost', port=8888, debug=True)