simple recurse class (Python recipe) by alexander baker

ActiveState Code (http://code.activestate.com/recipes/576449/)

# Class to encapsualte building a file
# from a path
 
import sys, os, marshal, os.path, unittest
 
import unittest
 
class recursiveException(Exception):
    'Error to raise for any recursive problem.'
class recurse:
    ''' Encapulation of utility functions for directory
    recursion or other scripts'''
 
    def __init__ (self):
        # set any global state
        self.results = {'start_path': [], 'path_list': [], 'bad_paths': [], 'file_list': [] }
        self.stop_list = []
 
    def __init__ (self, stop_list):
        # set any global state
        self.results = {'start_path': [], 'path_list': [], 'bad_paths': [], 'file_list': [] }
        self.stop_list = stop_list
 
    def recurseDirs(self):
        try:
            self.results.get('path_list').append(os.getcwd())
            self.results.get('path_list').append(os.getcwd())
        except:
          raise recursiveException('failed to get any directories')
 
    def recurseDirsPath(self, path):
        try:
            self.results.get('start_path').append(path)
            self.__internalRecursePath(self.results.get('start_path')[0])
        except:
          raise recursiveException('failed to get any directories')
 

    def __internalRecursePath(self, path):
        parent = []
        # recursive call
        parent = os.listdir(path)
        #print parent
        for child in parent:
            if os.path.isfile(os.path.join(path, child)) and child[0] != '.':
               if not len(self.stop_list) == 0:
                       stop_child = True
                       for stop in self.stop_list:
                                if stop in child:
                                        stop_child = False
                       if not stop_child:
                                self.results.get('file_list').append(os.path.join(path, child)) # is file add to file list
               else:
                       if not '.jar' in child:
                                if not '.class' in child:
                                        self.results.get('file_list').append(os.path.join(path, child)) # is file add to file list
            if os.path.isdir(os.path.join(path, child)) and child[0] != '.' and child != 'RCS':
                #print os.path.join(path, child)
                self.results.get('path_list').append(os.path.join(path, child)) # else recurse the directories checking for more files
                self.__internalRecursePath(os.path.join(path, child))
 
    def somethingElse(self):
        print self.results
 
    def getAllPaths(self):
        return self.results
 
    def getAllFiles(self):
        return self.results.get('file_list')

class TestCase(unittest.TestCase):
    def setUp(self):
        pass
    def tearDown(self):
        pass
    def testRecurrsion(self):
        p = recurse([])
        p.recurseDirs()
    def testSomethingElse(self):
        p = recurse([])
        p.somethingElse()
    def testReturn(self):
        p = recurse([])
        p.recurseDirs()
        print p.getAllPaths()
    def testPath(self):
        p = recurse([])
        p.recurseDirsPath('/home/blogsj/simple')
        print p.getAllPaths()
    def testFile(self):
        p = recurse(['.zip', '.txt', '.html', '.pyc', '.pyo', '~'])
        p.recurseDirsPath('/home/blogsj/simple')
        for f in p.getAllFiles():
                print f
if __name__ == '__main__':
        widgetTestSuite = unittest.TestSuite()
        widgetTestSuite.addTest(TestCase("testRecurrsion"))
        widgetTestSuite.addTest(TestCase("testSomethingElse"))
        widgetTestSuite.addTest(TestCase("testReturn"))
        widgetTestSuite.addTest(TestCase("testPath"))
        widgetTestSuite.addTest(TestCase("testFile"))
        runner = unittest.TextTestRunner()
        runner.run(widgetTestSuite)