manual_cpp

#!/usr/bin/env python3
from __future__ import print_function
import os
import sys
import subprocess


def cpp_arg(arg):
    return \
        arg.startswith("-I") or \
        arg.startswith("-D") or \
        arg.startswith("-U")


def check_call(args, **kwargs):
    if os.getenv("V") == "1":
        print(" ".join(args))
    return subprocess.check_call(args, **kwargs)


def check_call_redirect(args, filename=None, **kwargs):
    if os.getenv("V") == "1":
        print(" ".join(args), ">", filename)
    with open(filename, "wb") as fd:
        try:
            return subprocess.check_call(args, stdout=fd, **kwargs)
        except subprocess.CalledProcessError as e:
            os.remove(filename)
            raise SystemExit(e.returncode)


args = sys.argv[1:]
cpp_args = list(filter(cpp_arg, args))
files = list(filter(lambda q: q.endswith(".F90"), args))
args = list(filter(lambda q: not q.endswith(".F90"), args))

if len(files) > 1:
    raise Exception("Specify exactly one .F90 file")
elif len(files) == 0:
    # No .F90 file specified, execute program as-is
    try:
        os.execvp(args[0], args[0:])
    except OSError as e:
        print("Error executing '{0}': {1}".format(args[0], e.args[1]))
        raise SystemExit(1)
elif len(files) == 1:
    file, = files

tmp_filename = "manually_preprocessed_" + file.replace("/", "_")

try:
    output = args.index("-o")
    outputname = args[output + 1]
    tmp_filename += "-" + outputname.replace("/", "_") + ".F90"
except ValueError:
    pass

tmp_filename = tmp_filename[-250:]

# preprocess
check_call_redirect(["cpp", "-P", "-traditional", "-Wall", "-Werror"] + cpp_args + [file], filename=tmp_filename)

# compile
check_call(args + [tmp_filename])

# cleanup (may be commented out for better debuggability
os.remove(tmp_filename)