Python遍历文件目录并生成JSON结构树

最新推荐文章于 2024-09-30 19:32:29 发布

ZTao-z

最新推荐文章于 2024-09-30 19:32:29 发布

阅读量4.3k

点赞数 2

分类专栏：工具和插件文章标签： python

本文链接：https://blog.csdn.net/think_A_lot/article/details/101979494

版权

工具和插件专栏收录该内容

12 篇文章 0 订阅

订阅专栏

利用pathlib库进行路径读取，然后递归进行搜索

import os, re, json, copy
from pathlib import Path, PurePath

def searchAllFiles(self, path, fileTree={}, initial=True):
	if initial:
	    root = path if isinstance(path, str) else str(path)
	    fileTree[root] = {}
	    initial = False
	    return searchAllFiles(root, fileTree[root], initial)
	p = Path(path if isinstance(path, str) else str(path))
	dirs = [x for x in p.iterdir() if x.is_dir()]
	files = [x for x in p.iterdir() if x.is_file()]
	for file in files:
	    fileTree[file.name] = [file.suffix, file.as_posix()]
	for dir in dirs:
	    fileTree[str(dir.parts[-1])] = {}
	    searchAllFiles(dir, fileTree[str(dir.parts[-1])], initial)

if __name__ == '__main__':
	sampleDir = './test'
	fileTree = {}
	searchAllFiles(sampleDir, fileTree)
	print(fileTree)