あるディレクトリ以下のファイルツリーを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)))

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