有時候需要羅列下U盤等移動設備或一個程序下面的目錄結構的需求。基于這樣的需求個人整理了一個使用Python的小工具,期望對有這方面需求的朋友有所幫助。以下為具體代碼:
如果你所有要求的文件目錄不需要完整的文件路徑的話,直接更換下面的注釋代碼即可~
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
|
# -*- coding:utf-8 -*- import os def list_files(startPath): fileSave = open (& #039;list.txt','w') for root, dirs, files in os.walk(startPath): level = root.replace(startPath, & #039;').count(os.sep) indent = & #039; ' * 1 * level #fileSave.write('{}{}/'.format(indent, os.path.basename(root)) + '\n') fileSave.write(& #039;{}{}\\'.format(indent, os.path.abspath(root)) + '\n') subIndent = & #039; ' * 1 * (level + 1) for f in files: #fileSave.write('{}{}'.format(subIndent, f) + '\n') fileSave.write(& #039;{}{}{}'.format(subIndent, os.path.abspath(root), f) + '\n') fileSave.close() dir = raw_input (& #039;please input the path:') list_files( dir ) |