DiskWalk 遍历目录树工具类

import osclass DiskWalk(object):    '''    API for getting directory walking collections    '''    def __init__(self, path):        if path is None or len(path) == 0:            print 'Error: Please enter a valid path!'            return False        self.path = path    def enumeratePaths(self):        '''        Returns the path to all the files in a directory as a list        '''        path_collection = []        for dirpath, dirnames, filenames in os.walk(self.path):            for file in filenames:                fullpath = os.path.join(dirpath, file)                path_collection.append(fullpath)        return path_collection    def enumerateFiles(self):        '''        Returns all the files in a directory as a list        '''        path_collection = []        for dirpath, dirnames, filenames in os.walk(self.path):            for file in filenames:                path_collection.append(file)        return path_collection    def enumerateDir(self):        '''        Returns all the directories in a directory as a list        '''        path_collection = []        for dirpath, dirnames, filenames in os.walk(self.path):            for dir in dirnames:                path_collection.append(dir)        return path_collection