Python 读取指定目录下的所有子目录及所有文件

借鉴的   http://blog.csdn.net/pipisorry/article/details/46972171    但是他的代码有些问题。  在此更改后发出

扫描指定目录下的文件，或者匹配指定后缀和前缀的文件。

如果要扫描指定目录下的文件，包括子目录，调用scan_files("/export/home/test/")<
4000
/p>
如果要扫描指定目录下的特定后缀的文件（比如jar包），包括子目录，调用scan_files("/export/home/test/", postfix=".jar")

如果要扫描指定目录下的特定前缀的文件（比如test_xxx.py），包括子目录，调用scan_files("/export/home/test/", prefix="test_")

# -*- coding:utf8 -*-
import os
class ScanFile(object):
def __init__(self, directory, prefix=None, postfix=None):
self.directory = directory
self.prefix = prefix
self.postfix = postfix

def scan_files(self):
files_list = []
for dirpath, dirnames, filenames in os.walk(self.directory):
'''''
dirpath is a string, the path to the directory.
dirnames is a list of the names of the subdirectories in dirpath (excluding '.' and '..').
filenames is a list of the names of the non-directory files in dirpath.
'''
for special_file in filenames:
if self.postfix:
if special_file.endswith(self.postfix): # 判断一个字符的结尾是否是某字符   Python 内置函数 endswith（）
files_list.append(os.path.join(dirpath, special_file))
elif self.prefix:
if special_file.startswith(self.prefix): # 判断一个字符的开始是否是某字符   Python 内置函数 endswith（）
files_list.append(os.path.join(dirpath, special_file))
else:
files_list.append(os.path.join(dirpath, special_file))

return files_list

def scan_subdir(self):
subdir_list = []
for dirpath, dirnames, files in os.walk(self.directory):
subdir_list.append(dirpath)
return subdir_list

if __name__ == "__main__":
dir = r"/home/djl/project/s" #需要扫描的文件路径
scan = ScanFile(dir, postfix=".py")
subdirs = scan.scan_subdir()
files = scan.scan_files()

print u"扫描的子目录是:"
for subdir in subdirs:
print subdir

print u"扫描的文件是:"
for file in files:
print file

print '--------------------------'
print files
print '--------------------------'