X-Git-Url: https://git.ralfj.de/mass-build.git/blobdiff_plain/b8fb01b798441403be79fb586f2ece3a5578214e..a1dadd0687b2b5fc1178f83db50bf310bf473655:/mass_build.py

diff --git a/mass_build.py b/mass_build.py
index 93ae5a3..79bc17a 100755
--- a/mass_build.py
+++ b/mass_build.py
@@ -16,10 +16,25 @@
 # 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):
@@ -51,7 +66,7 @@ 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",
@@ -78,11 +93,8 @@ args = parser.parse_args()
 if args.reset_source and not args.update:
 	raise Exception("Can not reset sources without doing an update")
 
-# load config
-old_val = sys.dont_write_bytecode
-sys.dont_write_bytecode = True
-config = imp.load_source('config', args.config).__dict__
-sys.dont_write_bytecode = old_val
+# load config as dictionary
+config = vars(load_module('config', args.config))
 
 # initialise variables holding the configuration
 allProjects = OrderedDict() # all projects
@@ -91,7 +103,7 @@ 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 list(superConfig.keys()):
+	for name in superConfig.keys():
 		if (not name in subConfig) and (not name in exclude):
 			subConfig[name] = superConfig[name]