don't hard-code the path to the python interpreter
[mass-build.git] / build_system.py
index 4658f4a5fd08fc6a2fe7b352a32e8cb0d0d41da0..5dc4d98def5eb2721a8eb1c9c2d609d2ec6dbdb4 100644 (file)
@@ -1,5 +1,5 @@
 # mass-build - Easily Build Software Involving a Large Amount of Source Repositories
-# Copyright (C) 2012 Ralf Jung <post@ralfj.de>
+# 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
 # along with this program; if not, write to the Free Software
 # Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
 
-import os, shutil, subprocess
+import os, shutil, subprocess, multiprocessing
 
 '''A build system must have a "build" method with parameters "reconfigure" and "waitAfterConfig".'''
 
 # Compile, build, and install cmake projects:
 class CMake:
-       def __init__(self, sourceFolder, buildFolder, config):
-               self.sourceFolder = os.path.abspath(sourceFolder)
-               self.buildFolder = os.path.abspath(buildFolder)
-               self.config = config
-       
-       def setEnv(self, name, val):
-               '''Set the given environment variable, return old value'''
-               oldVal = os.getenv(name)
-               os.putenv(name, val)
-               return oldVal
-       
-       def prependDirToEnv(self, name, dir, default):
-               '''Prepends the given directory to the environment variable. If the variable is empty, dir isprepended to the default.
-                  Returns the old value.'''
-               oldVal = os.getenv(name)
-               oldPaths = default if oldVal is None else oldVal
-               os.putenv(name, dir+':'+oldPaths)
-               return oldVal
-       
-       def restoreEnv(self, name, oldVal):
-               '''Restore environment variable to previous value'''
-               if oldVal is None:
-                       os.unsetenv(name)
-               else:
-                       os.putenv(name, oldVal)
-       
-       def build(self, reconfigure, waitAfterConfig):
-               # Make sure we have a build directory
-               if not os.path.exists(self.buildFolder): os.makedirs(self.buildFolder)
-               os.chdir(self.buildFolder)
-               # In case of reconfiguration, delete cache file if it exists
-               cacheFile = 'CMakeCache.txt'
-               if os.path.exists(cacheFile) and reconfigure: os.remove(cacheFile)
-               # Run cmake, in the proper environment, then restore old environment
-               oldPKGConfigPath = self.setEnv('PKG_CONFIG_PATH', os.path.join(self.config['installDir'], 'lib', 'pkgconfig'))
-               oldCMakePrefixPath = self.setEnv('CMAKE_PREFIX_PATH', self.config['installDir'])
-               oldXDGDataDirs = self.prependDirToEnv('XDG_DATA_DIRS', os.path.join(self.config['installDir'], 'share'), '/usr/share')
-               oldXDGConfigDirs = self.prependDirToEnv('XDG_CONFIG_DIRS', os.path.join(self.config['installDir'], 'etc', 'xdg'), '/etc/xdg')
-               subprocess.check_call(['cmake', self.sourceFolder, '-DCMAKE_BUILD_TYPE='+self.config['buildType'],
-                       '-DCMAKE_INSTALL_PREFIX='+self.config['installDir']]+self.config.get('cmakeParameters', []))
-               self.restoreEnv('PKG_CONFIG_PATH', oldPKGConfigPath)
-               self.restoreEnv('CMAKE_PREFIX_PATH', oldCMakePrefixPath)
-               self.restoreEnv('XDG_DATA_DIRS', oldXDGDataDirs)
-               self.restoreEnv('XDG_CONFIG_DIRS', oldXDGConfigDirs)
-               # if asked to do so, wait
-               if waitAfterConfig:
-                       raw_input('Configuration done. Hit "Enter" to build the project. ')
-               # run compilation
-               subprocess.check_call(self.config.get('buildCmdPrefix', []) + ['make', '-j'+str(self.config['jobs'])])
-               # run installation
-               subprocess.check_call(self.config.get('installCmdPrefix', []) + ['make', 'install'])
+    def __init__(self, sourceFolder, buildFolder, config):
+        self.sourceFolder = os.path.abspath(sourceFolder)
+        self.buildFolder = os.path.abspath(buildFolder)
+        self.config = config
+    
+    def setEnv(self, name, val):
+        '''Set the given environment variable, return old value'''
+        oldVal = os.getenv(name)
+        os.putenv(name, val)
+        return oldVal
+    
+    def prependDirToEnv(self, name, dir, default):
+        '''Prepends the given directory to the environment variable. If the variable is empty, dir isprepended to the default.
+        Returns the old value.'''
+        oldVal = os.getenv(name)
+        oldPaths = default if oldVal is None else oldVal
+        os.putenv(name, dir+':'+oldPaths)
+        return oldVal
+    
+    def restoreEnv(self, name, oldVal):
+        '''Restore environment variable to previous value'''
+        if oldVal is None:
+            os.unsetenv(name)
+        else:
+            os.putenv(name, oldVal)
+    
+    def build(self, reconfigure, waitAfterConfig):
+        # Make sure we have a build directory
+        if not os.path.exists(self.buildFolder): os.makedirs(self.buildFolder)
+        os.chdir(self.buildFolder)
+        # In case of reconfiguration, delete cache file if it exists
+        cacheFile = 'CMakeCache.txt'
+        if os.path.exists(cacheFile) and reconfigure: os.remove(cacheFile)
+        # Run cmake, in the proper environment, then restore old environment
+        oldPKGConfigPath = self.setEnv('PKG_CONFIG_PATH', os.path.join(self.config['installDir'], 'lib', 'pkgconfig'))
+        oldCMakePrefixPath = self.setEnv('CMAKE_PREFIX_PATH', self.config['installDir'])
+        oldXDGDataDirs = self.prependDirToEnv('XDG_DATA_DIRS', os.path.join(self.config['installDir'], 'share'), '/usr/share')
+        oldXDGConfigDirs = self.prependDirToEnv('XDG_CONFIG_DIRS', os.path.join(self.config['installDir'], 'etc', 'xdg'), '/etc/xdg')
+        subprocess.check_call(['cmake', self.sourceFolder, '-DCMAKE_BUILD_TYPE='+self.config['buildType'],
+            '-DCMAKE_INSTALL_PREFIX='+self.config['installDir']]+self.config.get('cmakeParameters', []))
+        self.restoreEnv('PKG_CONFIG_PATH', oldPKGConfigPath)
+        self.restoreEnv('CMAKE_PREFIX_PATH', oldCMakePrefixPath)
+        self.restoreEnv('XDG_DATA_DIRS', oldXDGDataDirs)
+        self.restoreEnv('XDG_CONFIG_DIRS', oldXDGConfigDirs)
+        # if asked to do so, wait
+        if waitAfterConfig:
+            input('Configuration done. Hit "Enter" to build the project. ')
+        # run compilation
+        jobs = multiprocessing.cpu_count()+1
+        subprocess.check_call(self.config.get('buildCmdPrefix', []) + ['make', '-j'+str(jobs)])
+        # run installation
+        subprocess.check_call(self.config.get('installCmdPrefix', []) + ['make', 'install', '-j'+str((jobs+1)//2)]) # jobs/2, rounded up
 
 # if auto-debuild is available, provide a wrapper for it
 try:
-       import auto_debuild
-       class AutoDebuild:
-               def __init__(self, sourceFolder, buildFolder, config, vcs):
-                       self.sourceFolder = os.path.abspath(sourceFolder)
-                       self.buildFolder = os.path.abspath(buildFolder)
-                       self.debFolder = os.path.abspath(config['debDir'])
-                       self.config = config
-                       self.vcs = vcs
+    import auto_debuild
+    class AutoDebuild:
+        def __init__(self, sourceFolder, buildFolder, config, vcs):
+            self.sourceFolder = os.path.abspath(sourceFolder)
+            self.buildFolder = os.path.abspath(buildFolder)
+            self.debFolder = os.path.abspath(config['debDir'])
+            self.config = config
+            self.vcs = vcs
 
-               def build(self, reconfigure, waitAfterConfig): # reconfigure is ignored (we always do a reconfiguration)
-                       # get version name
-                       versionName = self.config['versionName'] if 'versionName' in self.config else self.vcs.version()
-                       if versionName is None:
-                               raise Exception("VCS did not provide us with a proper version number, please provide one manually")
-                       # create auto-debuild configuration
-                       autoDebuildConfig = {
-                               'sourceName': self.config['name'],
-                               'buildSystem': self.config['buildSystem'],
-                               'debDir': self.debFolder,
-                               'buildDir': self.buildFolder,
-                               'name': self.config['debName'],
-                               'email': self.config['debEMail'],
-                               'parallelJobs': self.config['jobs'],
-                               'version': versionName,
-                               'waitAfterConfig': waitAfterConfig,
-                       }
-                       # copy some more optional configuration
-                       for option in ('epoch', 'dbgPackage', 'section', 'withPython2', 'binarySkipFiles', 'binaryInstallFiles',
-                                       'buildDepends', 'binaryDepends', 'binaryShims', 'binaryRecommends', 'binaryProvides', 'binaryConflicts', 'binaryBreaks',
-                                       'binaryReplaces', 'binaryBreaksReplaces',
-                                       'alternatives', 'cmakeParameters', 'automakeParameters'):
-                               if option in self.config:
-                                       autoDebuildConfig[option] = self.config[option]
-                       # create Debian files
-                       os.chdir(self.sourceFolder)
-                       if os.path.isdir('debian'): # clean previous build attempts
-                               shutil.rmtree('debian')
-                       files = auto_debuild.createDebianFiles(autoDebuildConfig)
-                       # build package(s)
-                       auto_debuild.buildDebianPackage(autoDebuildConfig)
-                       # install package(s)
-                       if self.config.get('debInstall', True):
-                               subprocess.check_call(['sudo', 'dpkg', '--install'] + files)
+        def build(self, reconfigure, waitAfterConfig): # reconfigure is ignored (we always do a reconfiguration)
+            # get version name
+            versionName = self.config['versionName'] if 'versionName' in self.config else self.vcs.version()
+            if versionName is None:
+                raise Exception("VCS did not provide us with a proper version number, please provide one manually")
+            # create auto-debuild configuration
+            autoDebuildConfig = {
+                'sourceName': self.config['name'],
+                'buildSystem': self.config['buildSystem'],
+                'debDir': self.debFolder,
+                'buildDir': self.buildFolder,
+                'name': self.config['debName'],
+                'email': self.config['debEMail'],
+                'version': versionName + self.config.get('versionSuffix', ''),
+                'waitAfterConfig': waitAfterConfig,
+            }
+            # copy some more optional configuration
+            for option in ('epoch', 'dbgPackage', 'section', 'withPython2', 'withSIP', 'binarySkipFiles', 'binaryInstallFiles',
+                    'buildDepends', 'binaryDepends', 'binaryShims', 'binaryRecommends', 'binaryProvides', 'binaryConflicts', 'binaryBreaks',
+                    'binaryReplaces', 'binaryBreaksReplaces',
+                    'alternatives', 'cmakeParameters', 'automakeParameters', 'autogen'):
+                if option in self.config:
+                    autoDebuildConfig[option] = self.config[option]
+            # create Debian files
+            os.chdir(self.sourceFolder)
+            auto_debuild.deleteDebianFolder()
+            files = auto_debuild.createDebianFiles(autoDebuildConfig)
+            # build package(s)
+            auto_debuild.buildDebianPackage(autoDebuildConfig)
+            # install package(s)
+            if self.config.get('debInstall', True):
+                subprocess.check_call(['sudo', 'dpkg', '--install'] + files)
 
 except ImportError:
-       #print "auto_debuild not found, disabling auto-debuild system"
-       pass
+    #print "auto_debuild not found, disabling auto-debuild system"
+    pass