+
−# bootstrapping setuptools
+
−import ez_setup
+
−ez_setup.use_setuptools()
+
−
+
−import os
+
−import sys
+
−import textwrap
+
−from distutils.errors import *
+
−from distutils.command.clean import clean as _clean
+
−from distutils.cmd import Command
+
−from setuptools import setup
+
−from distutils import log
+
−
+
−from distutils.core import setup
+
−
+
−
+
−class clean(_clean):
+
−    """Also cleanup local temp files."""
+
−
+
−    def run(self):
+
−        _clean.run(self)
+
−
+
−        import fnmatch
+
−        
+
−        # kill temporary files
+
−        patterns = [
+
−            # generic tempfiles
+
−            '*~', '*.bak', '*.pyc',
+
−
+
−            # tempfiles generated by ANTLR runs
+
−            't[0-9]*Lexer.py', 't[0-9]*Parser.py',
+
−            '*.tokens', '*__.g',
+
−            ]
+
−            
+
−        for path in ('antlr3', 'unittests', 'tests'):
+
−            path = os.path.join(os.path.dirname(__file__), path)
+
−            if os.path.isdir(path):
+
−                for root, dirs, files in os.walk(path, topdown=True):
+
−                    graveyard = []                    
+
−                    for pat in patterns:
+
−                        graveyard.extend(fnmatch.filter(files, pat))
+
−
+
−                    for name in graveyard:
+
−                        filePath = os.path.join(root, name)
+
−
+
−                        try:
+
−                            log.info("removing '%s'", filePath)
+
−                            os.unlink(filePath)
+
−                        except OSError, exc:
+
−                            log.warn(
+
−                                "Failed to delete '%s': %s",
+
−                                filePath, exc
+
−                                )
+
−
+
−            
+
−class TestError(DistutilsError):
+
−    pass
+
−
+
−
+
−# grml.. the class name appears in the --help output:
+
−# ...
+
−# Options for 'CmdUnitTest' command
+
−# ...
+
−# so I have to use a rather ugly name...
+
−class unittest(Command):
+
−    """Run unit tests for package"""
+
−
+
−    description = "run unit tests for package"
+
−
+
−    user_options = [
+
−        ]
+
−    boolean_options = []
+
−
+
−    def initialize_options(self):
+
−        pass
+
−    
+
−    def finalize_options(self):
+
−        pass
+
−    
+
−    def run(self):
+
−        testDir = os.path.join(os.path.dirname(__file__), 'unittests')
+
−        if not os.path.isdir(testDir):
+
−            raise DistutilsFileError(
+
−                "There is not 'unittests' directory. Did you fetch the "
+
−                "development version?",
+
−                )
+
−
+
−        import glob
+
−        import imp
+
−        import unittest
+
−        import traceback
+
−        import StringIO
+
−        
+
−        suite = unittest.TestSuite()
+
−        loadFailures = []
+
−        
+
−        # collect tests from all unittests/test*.py files
+
−        testFiles = []
+
−        for testPath in glob.glob(os.path.join(testDir, 'test*.py')):
+
−            testFiles.append(testPath)
+
−
+
−        testFiles.sort()
+
−        for testPath in testFiles:
+
−            testID = os.path.basename(testPath)[:-3]
+
−
+
−            try:
+
−                modFile, modPathname, modDescription \
+
−                         = imp.find_module(testID, [testDir])
+
−
+
−                testMod = imp.load_module(
+
−                    testID, modFile, modPathname, modDescription
+
−                    )
+
−                
+
−                suite.addTests(
+
−                    unittest.defaultTestLoader.loadTestsFromModule(testMod)
+
−                    )
+
−                
+
−            except Exception:
+
−                buf = StringIO.StringIO()
+
−                traceback.print_exc(file=buf)
+
−                
+
−                loadFailures.append(
+
−                    (os.path.basename(testPath), buf.getvalue())
+
−                    )              
+
−                
+
−            
+
−        runner = unittest.TextTestRunner(verbosity=2)
+
−        result = runner.run(suite)
+
−
+
−        for testName, error in loadFailures:
+
−            sys.stderr.write('\n' + '='*70 + '\n')
+
−            sys.stderr.write(
+
−                "Failed to load test module %s\n" % testName
+
−                )
+
−            sys.stderr.write(error)
+
−            sys.stderr.write('\n')
+
−            
+
−        if not result.wasSuccessful() or loadFailures:
+
−            raise TestError(
+
−                "Unit test suite failed!",
+
−                )
+
−            
+
−
+
−class functest(Command):
+
−    """Run functional tests for package"""
+
−
+
−    description = "run functional tests for package"
+
−
+
−    user_options = [
+
−        ('testcase=', None,
+
−         "testcase to run [default: run all]"),
+
−        ('antlr-version=', None,
+
−         "ANTLR version to use [default: HEAD (in ../../build)]"),
+
−        ]
+
−    
+
−    boolean_options = []
+
−
+
−    def initialize_options(self):
+
−        self.testcase = None
+
−        self.antlr_version = 'HEAD'
+
−
+
−    
+
−    def finalize_options(self):
+
−        pass
+
−
+
−    
+
−    def run(self):
+
−        import glob
+
−        import imp
+
−        import unittest
+
−        import traceback
+
−        import StringIO
+
−        
+
−        testDir = os.path.join(os.path.dirname(__file__), 'tests')
+
−        if not os.path.isdir(testDir):
+
−            raise DistutilsFileError(
+
−                "There is not 'tests' directory. Did you fetch the "
+
−                "development version?",
+
−                )
+
−
+
−        # make sure, relative imports from testcases work
+
−        sys.path.insert(0, testDir)
+
−
+
−        rootDir = os.path.abspath(
+
−            os.path.join(os.path.dirname(__file__), '..', '..'))
+
−
+
−        if self.antlr_version == 'HEAD':
+
−            classpath = [
+
−                os.path.join(rootDir, 'build', 'classes'),
+
−                os.path.join(rootDir, 'build', 'rtclasses')
+
−                ]
+
−        else:
+
−            classpath = [
+
−                os.path.join(rootDir, 'archive',
+
−                             'antlr-%s.jar' % self.antlr_version)
+
−                ]
+
−
+
−        classpath.extend([
+
−            os.path.join(rootDir, 'lib', 'antlr-2.7.7.jar'),
+
−            os.path.join(rootDir, 'lib', 'stringtemplate-3.2.jar'),
+
−            os.path.join(rootDir, 'lib', 'junit-4.2.jar')
+
−            ])
+
−        os.environ['CLASSPATH'] = ':'.join(classpath)
+
−
+
−        os.environ['ANTLRVERSION'] = self.antlr_version
+
−
+
−        suite = unittest.TestSuite()
+
−        loadFailures = []
+
−        
+
−        # collect tests from all tests/t*.py files
+
−        testFiles = []
+
−        for testPath in glob.glob(os.path.join(testDir, 't*.py')):
+
−            if (testPath.endswith('Lexer.py')
+
−                or testPath.endswith('Parser.py')
+
−                ):
+
−                continue
+
−
+
−            # if a single testcase has been selected, filter out all other
+
−            # tests
+
−            if (self.testcase is not None
+
−                and os.path.basename(testPath)[:-3] != self.testcase
+
−                ):
+
−                continue
+
−            
+
−            testFiles.append(testPath)
+
−
+
−        testFiles.sort()
+
−        for testPath in testFiles:
+
−            testID = os.path.basename(testPath)[:-3]
+
−
+
−            try:
+
−                modFile, modPathname, modDescription \
+
−                         = imp.find_module(testID, [testDir])
+
−
+
−                testMod = imp.load_module(
+
−                    testID, modFile, modPathname, modDescription
+
−                    )
+
−                
+
−                suite.addTests(
+
−                    unittest.defaultTestLoader.loadTestsFromModule(testMod)
+
−                    )
+
−                
+
−            except Exception:
+
−                buf = StringIO.StringIO()
+
−                traceback.print_exc(file=buf)
+
−                
+
−                loadFailures.append(
+
−                    (os.path.basename(testPath), buf.getvalue())
+
−                    )              
+
−                
+
−            
+
−        runner = unittest.TextTestRunner(verbosity=2)
+
−        result = runner.run(suite)
+
−
+
−        for testName, error in loadFailures:
+
−            sys.stderr.write('\n' + '='*70 + '\n')
+
−            sys.stderr.write(
+
−                "Failed to load test module %s\n" % testName
+
−                )
+
−            sys.stderr.write(error)
+
−            sys.stderr.write('\n')
+
−            
+
−        if not result.wasSuccessful() or loadFailures:
+
−            raise TestError(
+
−                "Functional test suite failed!",
+
−                )
+
−            
+
−
+
−setup(name='antlr_python_runtime',
+
−      version='3.1.1',
+
−      packages=['antlr3'],
+
−
+
−      author="Benjamin Niemann",
+
−      author_email="pink@odahoda.de",
+
−      url="http://www.antlr.org/",
+
−      download_url="http://www.antlr.org/download.html",
+
−      license="BSD",
+
−      description="Runtime package for ANTLR3",
+
−      long_description=textwrap.dedent('''\
+
−      This is the runtime package for ANTLR3, which is required to use parsers
+
−      generated by ANTLR3.
+
−      '''),
+
−      
+
−      
+
−      cmdclass={'unittest': unittest,
+
−                'functest': functest,
+
−                'clean': clean
+
−                },
+
−      )