免费工具--遍历目标文件夹下所有文件

免费工具--遍历目标文件夹下所有文件

# -*- coding: cp936 -*-


#文件:BASE.py
#用途:遍历目的文件夹所有文件,并根据过滤标志返回属于过滤条件返回的文件绝对地址
#作者:刘华飞
#版本: 0.01
#时间:2007年5月11日
#授权:本程序可以免费使用,转载修改必修附上原作者信息




import os

#全局变量设置歌曲预定格式
Const_Song_Format=["mp3","wma","ogg"]


class BASE:
        #类变量,设置文件列表
        fileList=[""]
        #类变量,设置文件计算
        counter=0
        def __init__(self):
                pass
        def RecusWalkDir(self,dir,filtrate=0):
                """本方法递归遍历目的文件夹中所有文件,获取指定格式的文件绝对地址,利用类变量fileList存储地址"""
                global Const_Song_Format
                for s in os.listdir(dir):
                        newDir=dir+"/"+s
                        if os.path.isdir(newDir):
                                self.RecusWalkDir(newDir)
                        else:
                                if os.path.isfile(newDir):
                                        if filtrate:
                                                if newDir and (self.GetFileFormat(newDir) in Const_Song_Format):
                                                        self.__class__.fileList.append(newDir)
                                                        self.__class__.counter+=1
                                        else:
                                                self.__class__.fileList.append(newDir)
                                                self.__class__.counter+=1




        def CycWalkDir(self,dir,filtrate=0):
                """本方法循环遍历文件夹中所有文件,获取指定格式的文件绝对地址,返回歌曲列表fileList"""
                global Const_Song_Format
                fileList=[""]
                for s in os.listdir(dir):
                        newDir=dir+"/"+s
                        if os.path.isfile(newDir):
                                if filtrate:
                                        if newDir and (self.GetFileFormat(newDir) in Const_Song_Format):
                                                fileList.append(newDir)
                                else:
                                        fileList.append(newDir)
                        else:
                                newDir=dir+"/"+s
                        while os.path.isdir(newDir):
                                        for s in os.listdir(dir):
                                                newDir=dir+"/"+s
                                                if os.path.isfile(newDir):
                                                        if filtrate:
                                                                if newDir and (self.GetFileFormat(newDir) in Const_Song_Format):
                                                                        fileList.append(newDir)
                                                        else:
                                                                fileList.append(newDir)
                                                else:
                                                        newDir=dir+"/"+s
                return fileList


        def GetFileFormat(self,fileName):
                """返回文件格式"""
                if fileName:
                        BaseName=os.path.basename(fileName)
                        str=BaseName.split(".")
                        return str[-1]
                else:
                        return fileName

if __name__=="__main__":
        b=BASE()
        b.RecusWalkDir(dir="E:/音乐无限")
        print (b.counter)
        for k in  b.fileList:
                        print k
为什么不用os.walk()?
呵呵,OS.WALK要处理两次列表,你看看OS 的源码就知道了
def walk(top, topdown=True, onerror=None):
   
    from os.path import join, isdir, islink

    # We may not have read permission for top, in which case we can't
    # get a list of the files the directory contains.  os.path.walk
    # always suppressed the exception then, rather than blow up for a
    # minor reason when (say) a thousand readable directories are still
    # left to visit.  That logic is copied here.
    try:
        # Note that listdir and error are globals in this module due
        # to earlier import-*.
        names = listdir(top)
    except error, err:
        if onerror is not None:
            onerror(err)
        return

    dirs, nondirs = [], []
    for name in names:
        if isdir(join(top, name)):
            dirs.append(name)
        else:
            nondirs.append(name)

    if topdown:
        yield top, dirs, nondirs
    for name in dirs:
        path = join(top, name)
        if not islink(path):
            for x in walk(path, topdown, onerror):
                yield x
    if not topdown:
        yield top, dirs, nondirs

__all__.append("walk")

# Make sure os.environ exists, at least
try:
    environ
except NameError:
    environ = {}