don't hard-code the path to the python interpreter
[mass-build.git] / mass_build.py
index 814fa3efa397901af2452cc3d8c64660359d3f65..5b7a9dc0e27f58636c694408622b6d0cf54d4bea 100755 (executable)
@@ -1,4 +1,4 @@
-#!/usr/bin/python
+#!/usr/bin/env python3
 # mass-build - Easily Build Software Involving a Large Amount of Source Repositories
 # Copyright (C) 2012-2013 Ralf Jung <post@ralfj.de>
 #
 # along with this program; if not, write to the Free Software
 # Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
 
-import vcs, build_system, imp
+import vcs, build_system
 import argparse, os, sys, subprocess
 from collections import OrderedDict
 
+# helper functions
+def load_module(name, path, write_bytecode = False):
+    old_val = sys.dont_write_bytecode
+    sys.dont_write_bytecode = not write_bytecode
+    module = None
+    try:
+        from importlib.machinery import SourceFileLoader
+        module = SourceFileLoader(name, path).load_module()
+    except ImportError:
+        import imp
+        module = imp.load_source(name, path)
+    finally:
+        sys.dont_write_bytecode = old_val
+    return module
+
 # an entire Project
 class Project:
-       def __init__(self, folder, config):
-               self.folder = folder
-               self.name = config['name']
-               # VCS
-               vcsName = config['vcs']
-               if vcsName == 'git':
-                       self.vcs = vcs.Git(self.sourceFolder(), config)
-               elif vcsName == 'svn':
-                       self.vcs = vcs.SVN(self.sourceFolder(), config['url'])
-               else:
-                       raise Exception("Unknown VCS type "+vcsName)
-               # build system
-               if config.get('buildDeb', False):
-                       self.buildSystem = build_system.AutoDebuild(self.sourceFolder(), self.buildFolder(), config, self.vcs)
-               else:
-                       buildSystemName = config['buildSystem']
-                       if buildSystemName == 'cmake':
-                               self.buildSystem = build_system.CMake(self.sourceFolder(), self.buildFolder(), config)
-                       else:
-                               raise Exception("Unknown build system type "+buildSystemName)
-       
-       def sourceFolder(self):
-               return os.path.join(self.folder, self.name)
-       
-       def buildFolder(self):
-               return os.path.join(config['buildDir'], self.sourceFolder())
+    def __init__(self, folder, config):
+        self.folder = folder
+        self.name = config['name']
+        # VCS
+        vcsName = config['vcs']
+        if vcsName == 'git':
+            self.vcs = vcs.Git(self.sourceFolder(), config)
+        elif vcsName == 'svn':
+            self.vcs = vcs.SVN(self.sourceFolder(), config['url'])
+        else:
+            raise Exception("Unknown VCS type "+vcsName)
+        # build system
+        if config.get('buildDeb', False):
+            self.buildSystem = build_system.AutoDebuild(self.sourceFolder(), self.buildFolder(), config, self.vcs)
+        else:
+            buildSystemName = config['buildSystem']
+            if buildSystemName == 'cmake':
+                self.buildSystem = build_system.CMake(self.sourceFolder(), self.buildFolder(), config)
+            else:
+                raise Exception("Unknown build system type "+buildSystemName)
+    
+    def sourceFolder(self):
+        return os.path.join(self.folder, self.name)
+    
+    def buildFolder(self):
+        return os.path.join(config['buildDir'], self.sourceFolder())
 
 # read command-line arguments
 parser = argparse.ArgumentParser(description='Update and build a bunch of stuff')
-parser.add_argument("-c--config",
+parser.add_argument("-c", "--config",
                     dest="config", default="mass-build.conf",
                     help="mass-build config file")
 parser.add_argument("--reconfigure",
@@ -76,88 +91,89 @@ parser.add_argument("projects",  metavar='PROJECT', nargs='*',
                     help="Manually specify projects or folders to be built (project names take precedence)")
 args = parser.parse_args()
 if args.reset_source and not args.update:
-       raise Exception("Can not reset sources without doing an update")
+    raise Exception("Can not reset sources without doing an update")
+
+# load config as dictionary
+config = vars(load_module('config', args.config))
 
-# load config
-config = imp.load_source('config', args.config).__dict__
-os.remove(args.config+'c') # remove compiled python file
+# initialise variables holding the configuration
 allProjects = OrderedDict() # all projects
 allFolders = {} # all folders
 workProjects = [] # projects we work on
 
 # copy all items which don't exist below, except for those in the exclude list
 def inherit(subConfig, superConfig, exclude = ('name', 'projects')):
-       for name in superConfig.keys():
-               if (not name in subConfig) and (not name in exclude):
-                       subConfig[name] = superConfig[name]
+    for name in superConfig.keys():
+        if (not name in subConfig) and (not name in exclude):
+            subConfig[name] = superConfig[name]
 
 # populate list of projects, return list of projects in that folder
 def loadProjects(config, folder=''):
-       folderProjects = []
-       for projectConfig in config['projects']:
-               assert 'name' in projectConfig # everything must have a name
-               inherit(projectConfig, config)
-               if 'projects' in projectConfig: # a subpath
-                       folderProjects += loadProjects(projectConfig, os.path.join(folder, projectConfig['name']))
-               else: # a proper project
-                       if projectConfig['name'] in allProjects:
-                               raise Exception("Duplicate project name "+projectConfig['name'])
-                       project = Project(folder, projectConfig)
-                       allProjects[projectConfig['name']] = project
-                       folderProjects.append(project)
-       # store projects of this folder
-       if folder in allFolders:
-               raise Exception("Duplicate folder name "+folder)
-       allFolders[folder] = folderProjects
-       return folderProjects
+    folderProjects = []
+    for projectConfig in config['projects']:
+        assert 'name' in projectConfig # everything must have a name
+        inherit(projectConfig, config)
+        if 'projects' in projectConfig: # a subpath
+            folderProjects += loadProjects(projectConfig, os.path.join(folder, projectConfig['name']))
+        else: # a proper project
+            if projectConfig['name'] in allProjects:
+                raise Exception("Duplicate project name "+projectConfig['name'])
+            project = Project(folder, projectConfig)
+            allProjects[projectConfig['name']] = project
+            folderProjects.append(project)
+    # store projects of this folder
+    if folder in allFolders:
+        raise Exception("Duplicate folder name "+folder)
+    allFolders[folder] = folderProjects
+    return folderProjects
 
 # load available projects
 loadProjects(config)
 # get base set og projects to process
 if args.projects:
-       for name in args.projects:
-               if name in allProjects:
-                       workProjects.append(allProjects[name])
-               elif name in allFolders:
-                       workProjects += allFolders[name]
-               else:
-                       raise Exception("Project or folder %s does not exist" % name)
+    for name in args.projects:
+        if name in allProjects:
+            workProjects.append(allProjects[name])
+        elif name in allFolders:
+            workProjects += allFolders[name]
+        else:
+            raise Exception("Project or folder %s does not exist" % name)
 else:
-       workProjects = allProjects.values() # all the projects
+    workProjects = list(allProjects.values()) # all the projects
 # apply the "resume from"
 if args.resume_from is not None:
-       # find project index
-       startIndex = 0
-       while startIndex < len(workProjects):
-               if workProjects[startIndex].name == args.resume_from:
-                       break # we found it
-               else:
-                       startIndex += 1
-       if startIndex >= len(workProjects): # project not found
-               raise Exception("%s not found in list of projects to work on" % args.resume_from)
-       # start here
-       workProjects = workProjects[startIndex:]
+    # find project index
+    startIndex = 0
+    while startIndex < len(workProjects):
+        if workProjects[startIndex].name == args.resume_from:
+            break # we found it
+        else:
+            startIndex += 1
+    if startIndex >= len(workProjects): # project not found
+        raise Exception("%s not found in list of projects to work on" % args.resume_from)
+    # start here
+    workProjects = workProjects[startIndex:]
 
 # and do it!
 for project in workProjects:
-       try:
-               if args.version_check:
-                       print "Checking project",project.sourceFolder()
-                       project.vcs.checkVersions()
-               else:
-                       if args.update:
-                               print "Updating project",project.sourceFolder()
-                               project.vcs.update(mode = vcs.MODE_RESET if args.reset_source else vcs.MODE_REBASE)
-                       print "Building project",project.sourceFolder()
-                       project.buildSystem.build(reconfigure=args.reconfigure, waitAfterConfig=args.wait_after_config)
-               print
-       except (subprocess.CalledProcessError, KeyboardInterrupt) as e: # for some exceptions, a stackrace is usually pointless
-               print >> sys.stderr
-               print >> sys.stderr
-               if isinstance(e, KeyboardInterrupt): # str(e) would be the empty string
-                       print >> sys.stderr, "Interruped by user while processing %s" % (project.name)
-               else:
-                       print >> sys.stderr, "Error while processing %s: %s" % (project.name, str(e))
-               print >> sys.stderr
-               sys.exit(1)
-print "All operations successfully completed"
+    try:
+        if args.version_check:
+            print("Checking project",project.sourceFolder())
+            project.vcs.checkVersions()
+        else:
+            if args.update:
+                print("Updating project",project.sourceFolder())
+                project.vcs.update(mode = vcs.MODE_RESET if args.reset_source else vcs.MODE_REBASE)
+            print("Building project",project.sourceFolder())
+            project.buildSystem.build(reconfigure=args.reconfigure, waitAfterConfig=args.wait_after_config)
+        print()
+    except (subprocess.CalledProcessError, KeyboardInterrupt) as e: # for some exceptions, a stackrace is usually pointless
+        print(file=sys.stderr)
+        print(file=sys.stderr)
+        if isinstance(e, KeyboardInterrupt): # str(e) would be the empty string
+            print("Interruped by user while processing %s" % (project.name), file=sys.stderr)
+        else:
+            print("Error while processing %s: %s" % (project.name, str(e)), file=sys.stderr)
+        print(file=sys.stderr)
+        sys.exit(1)
+print("All operations successfully completed")