buildroot/support/testing/infra/builder.py

import os
import shutil
import subprocess

import infra

class Builder(object):
    def __init__(self, config, builddir, logtofile):
        self.config = config
        self.builddir = builddir
        self.logtofile = logtofile

    def build(self):
        if not os.path.isdir(self.builddir):
            os.makedirs(self.builddir)

        log = "{}-build.log".format(self.builddir)
        if not self.logtofile:
            log = None

        config_file = os.path.join(self.builddir, ".config")
        with open(config_file, "w+") as cf:
            cf.write(self.config)

        cmd = ["make",
               "O={}".format(self.builddir),
               "olddefconfig"]
        with infra.smart_open(log) as log_fh:
            ret = subprocess.call(cmd, stdout=log_fh, stderr=log_fh)
        if ret != 0:
            raise SystemError("Cannot olddefconfig")

        cmd = ["make", "-C", self.builddir]
        with infra.smart_open(log) as log_fh:
            ret = subprocess.call(cmd, stdout=log_fh, stderr=log_fh)
        if ret != 0:
            raise SystemError("Build failed")

        open(self.stamp_path(), 'a').close()

    def stamp_path(self):
        return os.path.join(self.builddir, "build-done")

    def is_finished(self):
        return os.path.exists(self.stamp_path())

    def delete(self):
        if os.path.exists(self.builddir):
            shutil.rmtree(self.builddir)