make load_module compatible with python 3.2
[mass-build.git] / mass_build.py
index 096d9a5834377a090fd5fb5c34fac5dada8b5654..79bc17a9db3f41edfe6c0c74eff1f2a570e2569b 100755 (executable)
@@ -1,8 +1,40 @@
-#!/usr/bin/python
-import vcs, build_system, imp
+#!/usr/bin/python3
+# mass-build - Easily Build Software Involving a Large Amount of Source Repositories
+# Copyright (C) 2012-2013 Ralf Jung <post@ralfj.de>
+#
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation; either version 2 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# 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
 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):
@@ -11,7 +43,7 @@ class Project:
                # VCS
                vcsName = config['vcs']
                if vcsName == 'git':
-                       self.vcs = vcs.Git(self.sourceFolder(), config['url'], config['version'])
+                       self.vcs = vcs.Git(self.sourceFolder(), config)
                elif vcsName == 'svn':
                        self.vcs = vcs.SVN(self.sourceFolder(), config['url'])
                else:
@@ -34,12 +66,15 @@ class Project:
 
 # 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",
                     action="store_true", dest="reconfigure",
                     help="Force configuration to be run")
+parser.add_argument("-w", "--wait-after-config",
+                    action="store_true", dest="wait_after_config",
+                    help="Wait for user confirmation after configuration is finished")
 parser.add_argument("--reset-source",
                     action="store_true", dest="reset_source",
                     help="Reset sourcecode to the given version (removes local changes!)")
@@ -49,16 +84,19 @@ parser.add_argument("--no-update",
 parser.add_argument("--resume-from", metavar='PROJECT',
                     dest="resume_from",
                     help="From the projects specified, continue building with this one (i.e., remove all projects before this one from the list - this never adds new projects)")
+parser.add_argument("--check-versions",
+                    action="store_true", dest="version_check",
+                    help="Check the repositories for newer tags, if possible (does not perform any building steps)")
 parser.add_argument("projects",  metavar='PROJECT', nargs='*',
                     help="Manually specify projects or folders to be built (project names take precedence)")
 args = parser.parse_args()
-# sanitize
 if args.reset_source and not args.update:
-       raise Exception("When no update is performed, no reset to the given version can be done either")
+       raise Exception("Can not reset sources without doing an update")
 
-# load config
-config = imp.load_source('config', args.config).__dict__
-os.remove(args.config+'c') # remove compiled python file
+# load config as dictionary
+config = vars(load_module('config', args.config))
+
+# initialise variables holding the configuration
 allProjects = OrderedDict() # all projects
 allFolders = {} # all folders
 workProjects = [] # projects we work on
@@ -101,7 +139,7 @@ if args.projects:
                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
@@ -119,18 +157,23 @@ if args.resume_from is not None:
 # and do it!
 for project in workProjects:
        try:
-               if args.update:
-                       print "Updating project",project.sourceFolder()
-                       project.vcs.update(forceVersion=args.reset_source)
-               print "Building project",project.sourceFolder()
-               project.buildSystem.build(reconfigure=args.reconfigure)
-               print
+               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
+               print(file=sys.stderr)
+               print(file=sys.stderr)
                if isinstance(e, KeyboardInterrupt): # str(e) would be the empty string
-                       print >> sys.stderr, "Interruped by user while processing %s" % (project.name)
+                       print("Interruped by user while processing %s" % (project.name), file=sys.stderr)
                else:
-                       print >> sys.stderr, "Error while processing %s: %s" % (project.name, str(e))
-               print >> sys.stderr
+                       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")