有时候需要罗列下U盘等移动设备或一个程序下面的目录结构的需求。基于这样的需求个人整理了一个使用Python的小工具,期望对有这方面需求的朋友有所帮助。以下为具体代码:
如果你所有要求的文件目录不需要完整的文件路径的话,直接更换下面的注释代码即可~
# -*- coding:utf-8 -*-
import os
def list_files(startPath):
fileSave = open(\'list.txt\',\'w\')
for root, dirs, files in os.walk(startPath):
level = root.replace(startPath, \'\').count(os.sep)
indent = \' \' * 1 * level
#fileSave.write(\'{}{}/\'.format(indent, os.path.basename(root)) + \'\\n\')
fileSave.write(\'{}{}\\\\\'.format(indent, os.path.abspath(root)) + \'\\n\')
subIndent = \' \' * 1 * (level + 1)
for f in files:
#fileSave.write(\'{}{}\'.format(subIndent, f) + \'\\n\')
fileSave.write(\'{}{}{}\'.format(subIndent, os.path.abspath(root), f) + \'\\n\')
fileSave.close()
dir = raw_input(\'please input the path:\')
list_files(dir)