18

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

ルートディレクトリ、1レベルのサブディレクトリ、およびファイルしかない場合は、うまく機能するようです。ただし、別のレベルのサブディレクトリを追加すると、問題が発生します（最後に、closeタグが十分な回数入力されていないためです）。しかし、私は頭を悩ませるのに苦労しています。

この方法で実行できない場合、より簡単な方法はありますか？私はFlaskを使用していますが、テンプレートの経験が非常に少ないため、何かが足りない可能性があります。

4

2 に答える 2

42

ディレクトリ ツリーの生成とその 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

recursiveHTML としてレンダリングするには、jinja2 のループ機能を使用できます。

<!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)
于 2012-06-09T14:54:49.277 に答える