あるディレクトリ以下のファイルツリーをJSONで再帰的に出力する

やりたいこと

Bootstrap Tree View を使って、見栄えよくあるディレクトリ以下のファイル一覧を表示させたい。

誰か作っているだろうと思ったのですが、探しても見つからなかったので書きました。

#!/usr/bin/env python

import os
import json

def json_tree(path):
    target_json = []

    for item in os.listdir(path):
        new_hash = {}
        new_hash['text'] = item

        full_path = os.path.join(path,item)
        if os.path.isdir(full_path):
            child_dir = json_tree(full_path)
            new_hash['nodes'] = append(child_dir)
        else:
            new_hash['href'] = full_path.replace(os.getcwd(),'')

        target_json.append(new_hash)
    return target_json

if __name__ == '__main__':
    path = os.getcwd()+'/files'
    print(json.dumps(json_tree(path)))

ソースはこちらにも置いておきました。