bootstrap.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-

"""

Bootstrap helps you to test pyFAI scripts without installing them
by patching your PYTHONPATH on the fly

example: ./bootstrap.py pyFAI-integrate test/testimages/Pilatus1M.edf

"""

__authors__ = ["Frédéric-Emmanuel Picca", "Jérôme Kieffer"]
__contact__ = "jerome.kieffer@esrf.eu"
__license__ = "GPLv3+"
__date__ = "21/11/2015"


import sys
import os
import shutil
import distutils.util
import subprocess


TARGET = "imagizer"

def _copy(infile, outfile):
    "link or copy file according to the OS. Nota those are HARD_LINKS"
    if "link" in dir(os):
        os.link(infile, outfile)
    else:
        shutil.copy(infile, outfile)


def _distutils_dir_name(dname="lib"):
    """
    Returns the name of a distutils build directory
    """
    platform = distutils.util.get_platform()
    architecture = "%s.%s-%i.%i" % (dname, platform,
                                    sys.version_info[0], sys.version_info[1])
    return architecture


def _distutils_scripts_name():
    """Return the name of the distrutils scripts sirectory"""
    f = "scripts-{version[0]}.{version[1]}"
    return f.format(version=sys.version_info)


def _get_available_scripts(path):
    res = []
    try:
        res = " ".join([s.rstrip('.py') for s in os.listdir(path)])
    except OSError:
        res = ["no script available, did you ran "
               "'python setup.py build' before bootstrapping ?"]
    return res


def _copy_files(source, dest, extn):
    """
    copy all files with a given extension from source to destination
    """
    if not os.path.isdir(dest):
        os.makedirs(dest)
    full_src = os.path.join(os.path.dirname(__file__), source)
    for clf in os.listdir(full_src):
        if clf.endswith(extn) and clf not in os.listdir(dest):
            _copy(os.path.join(full_src, clf), os.path.join(dest, clf))


def runfile(fname):
    try:
        execfile(fname)
    except SyntaxError:
        env = os.environ.copy()
        env.update({"PYTHONPATH": LIBPATH + os.pathsep + os.environ.get("PYTHONPATH", ""),
                    "PATH": SCRIPTSPATH + os.pathsep + os.environ.get("PATH", "")})
        run = subprocess.Popen(sys.argv, shell=False, env=env)
        run.wait()

home = os.path.dirname(os.path.abspath(__file__))
SCRIPTSPATH = os.path.join(home,
                           'build', _distutils_scripts_name())
LIBPATH = (os.path.join(home,
                       'build', _distutils_dir_name('lib')))

if (not os.path.isdir(SCRIPTSPATH)) or (not os.path.isdir(LIBPATH)):
    build = subprocess.Popen([sys.executable, "setup.py", "build"],
                     shell=False, cwd=os.path.dirname(__file__))
    print("Build process ended with rc= %s" % build.wait())
# _copy_files("openCL", os.path.join(LIBPATH, TARGET, "openCL"), ".cl")
_copy_files("gui", os.path.join(LIBPATH, TARGET, "gui"), ".ui")
# _copy_files("calibration", os.path.join(LIBPATH, TARGET, "calibration"), ".D")

if __name__ == "__main__":
    if len(sys.argv) < 2:
        print("usage: ./bootstrap.py <script>\n")
        print("Available scripts : %s\n" %
              _get_available_scripts(SCRIPTSPATH))
        sys.exit(1)
    os.system("cd %s;python setup.py build; cd -" % home)
    print("Executing %s from source checkout" % (sys.argv[1]))

    sys.path.insert(0, LIBPATH)
    print("01. Patched sys.path with %s" % LIBPATH)

    sys.path.insert(0, SCRIPTSPATH)
    print("02. Patched sys.path with %s" % SCRIPTSPATH)

    script = sys.argv[1]
    sys.argv = sys.argv[1:]
    print("03. patch the sys.argv : ", sys.argv)

    print("04. Executing %s.main()" % (script,))
    fullpath = os.path.join(SCRIPTSPATH, script)
    if os.path.exists(fullpath):
        runfile(fullpath)
    else:
        if os.path.exists(script):
            runfile(script)
        else:
            for dirname in os.environ.get("PATH", "").split(os.pathsep):
                fullpath = os.path.join(dirname, script)
                if os.path.exists(fullpath):
                    runfile(fullpath)
                    break