python读取指定目录下所有文件

import os,glob

def get_file_list(dir_path, extension_list):

'''

fuction: get_file_list(dir_path,extension_list)

parms:

dir_path - a string of directory full path. eg. 'D:user'

extension_list - a list of file extension. eg. ['zip']

return: a list of file full path. eg. ['D:user1.zip', 'D:user2.zip']

''' 

os.chdir(dir_path)

file_list = []

for extension in extension_list:

extension = '*.' + extension

file_list += [os.path.realpath(e) for e in glob.glob(extension) ] 

return file_list

dir_path = 'F:downloadspic'

extension_list = ['jpg']

a = get_file_list(dir_path, extension_list)

print('extension is 'jpg'')

for file in a:

print(file)

print('n') 

extension_list = ['txt']

a = get_file_list(dir_path, extension_list)

print('extension is 'txt'')

for file in a:

print(file)

print('n') 

extension_list = ['jpg','txt']

a = get_file_list(dir_path, extension_list)

print('extension is 'jpg' and 'txt'')

for file in a:

print(file)

#!/usr/bin/python
# -*- coding:utf8 -*-

import os
allFileNum = 0
def printPath(level, path):
    global allFileNum
    '''
    打印一个目录下的所有文件夹和文件
    '''
    # 所有文件夹，第一个字段是次目录的级别
    dirList = []
    # 所有文件
    fileList = []
    # 返回一个列表，其中包含在目录条目的名称(google翻译)
    files = os.listdir(path)
    # 先添加目录级别
    dirList.append(str(level))
    for f in files:
        if(os.path.isdir(path + '/' + f)):
            # 排除隐藏文件夹。因为隐藏文件夹过多
            if(f[0] == '.'):
                pass
            else:
                # 添加非隐藏文件夹
                dirList.append(f)
        if(os.path.isfile(path + '/' + f)):
            # 添加文件
            fileList.append(f)
    # 当一个标志使用，文件夹列表第一个级别不打印
    i_dl = 0
    for dl in dirList:
        if(i_dl == 0):
            i_dl = i_dl + 1
        else:
            # 打印至控制台，不是第一个的目录
            print '-' * (int(dirList[0])), dl
            # 打印目录下的所有文件夹和文件，目录级别+1
            printPath((int(dirList[0]) + 1), path + '/' + dl)
    for fl in fileList:
        # 打印文件
        print '-' * (int(dirList[0])), fl
        # 随便计算一下有多少个文件
        allFileNum = allFileNum + 1

if __name__ == '__main__':
    printPath(1, '/home/lizheng')
    print '总文件数 =', allFileNum