python遍历文件夹下所有文件并查找_[python] 读取一层文件目录下的文件，并遍历查找某行特定的信息...

#-*- coding:utf-8 -*-

#!/usr/local/bin/python

import os,fnmatch,pdb

def print_file_type(fileName):

with open(fileName) as f:

lines = (line for line in f)

for line in lines:

if line.startswith('\x02#!SILK_V3') is True:

print fileName[-4:] + " " + "SILK_V3"

if line.startswith('#!AMR') is True:

print fileName[-4:] + " " + "AMR"

break

def all_files(root,patterns='*' , single_level=False ,yieild_folders= False):

#将模式从字符串中取出放入列表中

patterns = patterns.split(';')

for path ,subdirs ,files in os.walk(root):

if yieild_folders :

files.extend(subdirs)

files.sort()

for name in files:

for pattern in patterns:

if fnmatch.fnmatch(name,pattern):

yield os.path.join(path,name)

break

if single_level:

break

if __name__ == "__main__":

#search_path = 'C:\Users\wenle\Desktop\ptt_20170222\\'

#+ os.pathsep + 'c:\\' #;on windows ,:on Unix

#find_file = search_file("[1~4]",search_path)

#print find_file

search_path = 'C:\Users\wenle\Desktop\ptt_20170222\\'

#只搜索一层目录，single_level=True

for path in all_files(search_path,'*',True):

print_file_type(path)

#print_file_type('C:\\Users\\wenle\\Desktop\\ptt_20170222\\1059')