Move loading a module by filename into a helper function

[mass-build.git] / mass_build.py

diff --git a/mass_build.py b/mass_build.py
index cdf511402caad26525c0acc6d46d581b55333b06..e327b7ee2e91c23ea86b6226f949813de281c5ae 100755 (executable)
--- a/mass_build.py
+++ b/mass_build.py
@@ -1,58 +1,54 @@
-#!/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
 
 import argparse, os, sys, subprocess
 from collections import OrderedDict
 

-# read command-line arguments
-parser = argparse.ArgumentParser(description='Update and build a bunch of stuff')
-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("--reset-source",
-                    action="store_true", dest="reset_source",
-                    help="Reset sourcecode to the given version (removes local changes!)")
-parser.add_argument("--no-update",
-                    action="store_false", dest="update",
-                    help="Do not update modules before compilation")
-parser.add_argument("--resume-from", metavar='MODULE',
-                    dest="resume_from",
-                    help="Resume building from the given repository")
-parser.add_argument("modules",  metavar='MODULE', nargs='*',
-                    help="Manually specify modules to be built")
-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")
-
-# load config
-config = imp.load_source('config', args.config).__dict__
-os.remove(args.config+'c') # remove compiled python file
-projects = OrderedDict() # all projects
-workProjects = [] # projects we work on
+# helper funcrions
+def load_module(name, path):
+       import importlib.machinery
+       old_val = sys.dont_write_bytecode
+       sys.dont_write_bytecode = True
+       module = importlib.machinery.SourceFileLoader(name, path).load_module()
+       sys.dont_write_bytecode = old_val
+       return module

 
 # an entire Project
 class Project:
 
 # an entire Project
 class Project:

-       def __init__(self, config, folder, module):
+       def __init__(self, folder, config):

                self.folder = folder
                self.folder = folder

-               self.name = module['name']
+               self.name = config['name']

                # VCS
                # VCS

-               vcsName = module['vcs']
+               vcsName = config['vcs']

                if vcsName == 'git':
                if vcsName == 'git':

-                       self.vcs = vcs.Git(self.sourceFolder(), module['url'], module['version'])
+                       self.vcs = vcs.Git(self.sourceFolder(), config)

                elif vcsName == 'svn':
                elif vcsName == 'svn':

-                       self.vcs = vcs.SVN(self.sourceFolder(), module['url'], module.get('versionName'))
+                       self.vcs = vcs.SVN(self.sourceFolder(), config['url'])

                else:
                        raise Exception("Unknown VCS type "+vcsName)
                # build system
                if config.get('buildDeb', False):
                else:
                        raise Exception("Unknown VCS type "+vcsName)
                # build system
                if config.get('buildDeb', False):

-                       self.buildSystem = build_system.AutoDebuild(self.sourceFolder(), self.buildFolder(), module, self.vcs, config)
+                       self.buildSystem = build_system.AutoDebuild(self.sourceFolder(), self.buildFolder(), config, self.vcs)

                else:
                else:

-                       buildSystemName = module['buildSystem']
+                       buildSystemName = config['buildSystem']

                        if buildSystemName == 'cmake':
                        if buildSystemName == 'cmake':

-                               self.buildSystem = build_system.CMake(self.sourceFolder(), self.buildFolder(), module, config)
+                               self.buildSystem = build_system.CMake(self.sourceFolder(), self.buildFolder(), config)

                        else:
                                raise Exception("Unknown build system type "+buildSystemName)
        
                        else:
                                raise Exception("Unknown build system type "+buildSystemName)
        


@@ -62,56 +58,116 @@ class Project:
        def buildFolder(self):
                return os.path.join(config['buildDir'], self.sourceFolder())
 
        def buildFolder(self):
                return os.path.join(config['buildDir'], self.sourceFolder())
 

-# return the position of the given item in the list
-def findInList(list, item):
-       for i in xrange(len(list)):
-               if list[i] == item:
-                       return i
-       raise Exception("%s not found in list" % str(item))
+# read command-line arguments
+parser = argparse.ArgumentParser(description='Update and build a bunch of stuff')
+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!)")
+parser.add_argument("--no-update",
+                    action="store_false", dest="update",
+                    help="Do not update projects before compilation")
+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()
+if args.reset_source and not args.update:
+       raise Exception("Can not reset sources without doing an update")
+
+# load config as dictionary
+config = load_module('config', args.config).__dict__
+
+# 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]

 
 

-# populate list of projects
-def loadProjects(config, modules, folder=''):
-       for module in modules:
-               if 'folder' in module: # a subpath
-                       loadProjects(config, module['modules'], os.path.join(folder, module['folder']))
+# 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
                else: # a proper project

-                       if module['name'] in projects:
-                               raise Exception("Duplicate module name "+module['name'])
-                       projects[module['name']] = Project(config, folder, module)
+                       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

 
 

-# now check what we have to do
-loadProjects(config, config['modules'])
-if args.modules:
-       if args.resume_from is not None:
-               raise Exception("Can not use --resume-from and manually specify modules")
-       for module in args.modules:
-               if not module in projects:
-                       raise Exception("Project %s does not exist" % module)
-               workProjects.append(projects[module])
-elif args.resume_from is None:
-       workProjects = projects.values() # all the projects
+# 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)

 else:
 else:

-       if not args.resume_from in projects:
-               raise Exception("Project %s does not exist" % args.resume_from)
-       startWith = projects[args.resume_from]
-       startIndex = findInList(projects.values(), startWith)
-       workProjects = projects.values()[startIndex:]
+       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:]

 
 # and do it!
 for project in workProjects:
        try:
 
 # and do it!
 for project in workProjects:
        try:

-               if args.update:
-                       print "Updating module",project.sourceFolder()
-                       project.vcs.update(forceVersion=args.reset_source)
-               print "Building module",project.sourceFolder()
-               project.buildSystem.build(reconfigure=args.reconfigure)
-               print
-       except (subprocess.CalledProcessError, KeyboardInterrupt) as e:
-               print >> sys.stderr
-               print >> sys.stderr
+               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
                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:
                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)
                sys.exit(1)

+print("All operations successfully completed")