展示目录和文件直接的结构关系,方式有很多,下面这中式最直接明了的,代码是网上学习,所以如下这段代码并非原创,只是不知道来自哪里。
展示的文件结构如下:
这种展示也是大家所期待的,因为层级关系更清晰。
实现代码如下:
import os
import argparse
import time
class FileTreeMaker(object):
def _recurse(self, parent_path, file_list, prefix, output_buf, level):
if len(file_list) ==0 or (self.max_level != -1 and self.max_level <= level):
return
else:
file_list.sort(key=lambda f: os.path.isfile(os.path.join(parent_path, f)))
for idx, sub_pathin enumerate(file_list):
if any(exclude_namein sub_pathfor exclude_namein self.exn):
continue
full_path = os.path.join(parent_path, sub_path)
idc =“┣━”
if idx ==len(file_list) -1:
idc =“┗━”
if os.path.isdir(full_path)and sub_pathnot in self.exf:
output_buf.append("%s%s[%s]" % (prefix, idc, sub_path))
if len(file_list) >1 and idx !=len(file_list) -1:
tmp_prefix = prefix +"┃ "
else:
tmp_prefix = prefix +" "
self._recurse(full_path, os.listdir(full_path), tmp_prefix, output_buf, level +1)
elif os.path.isfile(full_path):
output_buf.append("%s%s%s" % (prefix, idc, sub_path))
def make(self, args):
self.root = args.root
self.exf = args.exclude_folder
self.exn = args.exclude_name
self.max_level = args.max_level
print(“root:%s” %self.root)
buf = []
path_parts =self.root.rsplit(os.path.sep, 1)
buf.append("[%s]" % (path_parts[-1],))
self._recurse(self.root, os.listdir(self.root), “”, buf, 0)
output_str ="\n".join(buf)
if len(args.output) !=0:
with open(args.output, ‘w’)as of:
of.write(output_str)
return output_str
def Show_Folder_File_Struct(theFolderPath):
parser = argparse.ArgumentParser()
# parser.add_argument("-r", “–root”, help=“root of file tree”, default=".")
parser.add_argument("-r", “–root”, help=“root of file tree”, default=theFolderPath)
parser.add_argument("-o", “–output”, help=“output file name”, default="")
parser.add_argument("-xf", “–exclude_folder”, nargs=’*’, help=“exclude folder”, default=[])
parser.add_argument("-xn", “–exclude_name”, nargs=’*’, help=“exclude name”, default=[])
parser.add_argument("-m", “–max_level”, help=“max level”, type=int, default=-1)
args = parser.parse_args()
print(FileTreeMaker().make(args))
测试验证:
Show_Folder_File_Struct(r’D:\test’)
除了展示,有时需要将文件的结构存储到Log日志,如下代码演示存储Log。
def Storage_Fodler_Struct(theFolder):
with open(‘logs.txt’, ‘w’, encoding=“utf8”)as f:
for root, dirs, filesin os.walk(theFolder):
f.write(“目录:” + root +’\n’)
if dirs !="":
for din dirs:
f.write(“子目录:” + d +’\n’)
if files !="":
for filein files:
f.write(“文件:” + file +’\n’)
f.write(’*’ *50 +"\n")
执行上述函数验证:
Storage_Fodler_Struct(r’D:\Test’)
在当前目录存储了一个log.txt文件,文件内容是结构信息:
目录:D:\Test
子目录:Test1
子目录:Test2
文件:新建文本文档- 副本(2).txt
文件:新建文本文档- 副本(3).txt
文件:新建文本文档- 副本.txt
文件:新建文本文档.txt
**************************************************
目录:D:\Test\Test1
文件:S1新建文本文档- 副本(2).txt
文件:S1新建文本文档- 副本.txt
文件:S1新建文本文档.txt
**************************************************
目录:D:\Test\Test2
子目录:Test2_1
文件:S2新建文本文档- 副本(2).txt
文件:S2新建文本文档- 副本.txt
文件:S2新建文本文档.txt
**************************************************
目录:D:\Test\Test2\Test2_1
文件:子目录2下的子目录的文件- 副本.txt
文件:子目录2下的子目录的文件.txt
**************************************************
下一篇利用1,2的知识,做文件名的遍历修改,扩展名的遍历修改,文件夹的备份等操作。