platform_methods.py 2.85 KB
Newer Older
1 2 3 4 5 6 7 8 9
import os
import sys
import json
import uuid
import functools
import subprocess

# NOTE: The multiprocessing module is not compatible with SCons due to conflict on cPickle

10 11 12 13 14
if sys.version_info[0] < 3:
    JSON_SERIALIZABLE_TYPES = (bool, int, long, float, basestring)
else:
    JSON_SERIALIZABLE_TYPES = (bool, int, float, str)

15 16 17 18 19 20 21 22 23 24

def run_in_subprocess(builder_function):

    @functools.wraps(builder_function)
    def wrapper(target, source, env):

        # Convert SCons Node instances to absolute paths
        target = [node.srcnode().abspath for node in target]
        source = [node.srcnode().abspath for node in source]

25 26 27 28
        # Short circuit on non-Windows platforms, no need to run in subprocess
        if sys.platform not in ('win32', 'cygwin'):
            return builder_function(target, source, env)

29 30 31 32 33 34 35 36 37 38 39
        # Identify module
        module_name = builder_function.__module__
        function_name = builder_function.__name__
        module_path = sys.modules[module_name].__file__
        if module_path.endswith('.pyc') or module_path.endswith('.pyo'):
            module_path = module_path[:-1]

        # Subprocess environment
        subprocess_env = os.environ.copy()
        subprocess_env['PYTHONPATH'] = os.pathsep.join([os.getcwd()] + sys.path)

40 41 42 43 44 45 46
        # Keep only JSON serializable environment items
        filtered_env = dict(
            (key, value)
            for key, value in env.items()
            if isinstance(value, JSON_SERIALIZABLE_TYPES)
        )

47
        # Save parameters
48
        args = (target, source, filtered_env)
49 50 51 52
        data = dict(fn=function_name, args=args)
        json_path = os.path.join(os.environ['TMP'], uuid.uuid4().hex + '.json')
        with open(json_path, 'wt') as json_file:
            json.dump(data, json_file, indent=2)
53 54 55 56 57
        json_file_size = os.stat(json_path).st_size

        print('Executing builder function in subprocess: '
              'module_path=%r, parameter_file=%r, parameter_file_size=%r, target=%r, source=%r' % (
               module_path, json_path, json_file_size, target, source))
58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82
        try:
            exit_code = subprocess.call([sys.executable, module_path, json_path], env=subprocess_env)
        finally:
            try:
                os.remove(json_path)
            except (OSError, IOError) as e:
                # Do not fail the entire build if it cannot delete a temporary file
                print('WARNING: Could not delete temporary file: path=%r; [%s] %s' %
                      (json_path, e.__class__.__name__, e))

        # Must succeed
        if exit_code:
            raise RuntimeError(
                'Failed to run builder function in subprocess: module_path=%r; data=%r' % (module_path, data))

    return wrapper


def subprocess_main(namespace):

    with open(sys.argv[1]) as json_file:
        data = json.load(json_file)

    fn = namespace[data['fn']]
    fn(*data['args'])