python 搜索

用 walk 函数 遍历全部文件

查找出 后缀为 jpg的文件

import os,shutil
path = "D:\\"
pattern = "*.jpg"

#Print files that match to file extensions
def printFiles(dirList, spaceCount, typeList):
for file in dirList:
for ext in typeList:
if file.endswith(ext):
print "/".rjust(spaceCount+1) + file

#shutil.copyfile(src,dst)#并移动到F盘

break

#Print each sub-directory
def printDirectory(dirEntry, typeList):
print dirEntry[0] + "/"
printFiles(dirEntry[2], len(dirEntry[0]),
typeList)

#Convert pattern string to list of file extensions
extList = []
for ext in pattern.split(";"):
extList.append(ext.lstrip("*"))

#Walk the tree to print files
for directory in os.walk(path):
printDirectory(directory, extL