Python 扫面文件夹中的文件

#coding=utf-8
import os,sys
reload(sys)
sys.setdefaultencoding("utf-8")
def scan_files_sub_dir(directory,prefix=None,postfix=None):
    files_list=[]
    for root, sub_dirs, files in os.walk(directory):
        for special_file in files:
            if postfix:
                if special_file.endswith(postfix):
                    files_list.append(os.path.join(root,special_file))
            elif prefix:
                if special_file.startswith(prefix):
                    files_list.append(os.path.join(root,special_file))
            else:
                files_list.append(os.path.join(root,special_file)) 
    return files_list
files  = scan_files_sub_dir(".",None,".txt")
print files

def scan_files(directory,prefix=None,postfix=None):
    files_list=[]
    for file in os.listdir(directory):
        if postfix:
                if file.endswith(postfix):
                    files_list.append(file)
        elif prefix:
            if file.startswith(prefix):
                files_list.append(file)
        else:
            files_list.append(file) 
    return files_list
files  = scan_files(".",None,".txt")
print files

 另一种方法:

import os
def get_file_lists(init_dir,postfix=None):
    file_lists = []
    if not os.path.exists(init_dir):
        print('%s not exist'%init_dir)
        return file_lists
    if not os.path.isdir(init_dir):
        print('%s not dir'%init_dir)
        return file_lists
    for file in os.listdir(init_dir):
        path = "%s\%s"%(init_dir,file)
        if postfix and not file.endswith(postfix) and os.path.isfile(path):
            continue
        if not os.path.isdir(path):
            file_lists.append(path)
        if os.path.isdir(path):
            file_lists.extend(get_file_lists(path,postfix))
    return file_lists

file_lists = get_file_lists(r'D:Program FilesJava','java')

file_writer = open('list.txt','w',encoding='utf-8')
for file in file_lists:
    file_writer.write("%s
"%file)
file_writer.close()