build_adapter.py

import argparse, docker
import system_testing
import os
from common import ccall

if __name__ == "__main__":
    # Parsing flags
    parser = argparse.ArgumentParser(description='Build local docker image of an adapter.')
    parser.add_argument('-d', '--dockerfile', type=str, help="Choose Dockerfile you want to build", default="adapters/Dockerfile.fenics-adapter")
    parser.add_argument('-b', '--branch', help="preCICE branch to use", default="develop")
    parser.add_argument('-ab', '--adapter-branch', help="Adapter branch to use", default="develop")
    parser.add_argument('-u', '--docker-username', help="docker username", default=os.environ["DOCKER_USERNAME"] if "DOCKER_USERNAME" in os.environ else "precice")
    parser.add_argument('-os', '--operating-system', help="Operating system used by preCICE base image", default="ubuntu1804")
    parser.add_argument('-i', '--precice-installation', help="Installation mode used for preCICE", default="package")
    parser.add_argument('-p', '--petsc', help="set 'yes', if you want to use a preCICE base image that was built with PETSc.", default="no", choices={'yes', 'no'})
    parser.add_argument('-f', '--force-rebuild', action='store_true', help="Force rebuild of docker image.")
    parser.add_argument('-bs', '--base-solver', type=str, help="Specify the docker image of the base solver used.", default=None)

    args = parser.parse_args()

    dockerfile = os.path.basename(args.dockerfile)
    assert(dockerfile.split(".")[0] == "Dockerfile")  # We have the convention that our Dockerfiles always start with the term "Dockerfile"

    adapter_name = dockerfile.split(".")[1]  # Extract adapter name from filename

    # converting features provided via command-line to dictionary
    features = dict()
    assert(args.operating_system)
    if args.operating_system:  # first feature is mandatory always describes the os
        features["os"] = args.operating_system
    if args.precice_installation:  # second feature is optional if it exists it describes the preCICE installation
        features["installation"] = args.precice_installation
    if args.petsc == "yes":
        features["petsc"] = "yes"

    tag = system_testing.compose_tag(args.docker_username, adapter_name, features, args.branch)

    # converting features provided via command-line to dictionary
    precice_base_features = dict()
    assert(args.operating_system)
    if args.operating_system:  # first feature is mandatory always describes the os
        precice_base_features["os"] = args.operating_system
    if args.precice_installation:  # second feature is optional if it exists it describes the preCICE installation
        precice_base_features["installation"] = args.precice_installation
    if args.petsc == "yes":
        precice_base_features["petsc"] = "yes"

    precice_base_tag = system_testing.compose_tag(args.docker_username, "precice", precice_base_features, args.branch)

    print("Building {} image with the following features: {}".format(adapter_name, features))

    build_args={"branch" : args.branch,
                "adapter_branch" : args.adapter_branch,
                "from" : precice_base_tag}

    # Check in order to preserve default value in dockerfile unless base_solver is deliberately set
    if args.base_solver:
        build_args["base_solver"] = args.base_solver

    docker.build_image(tag=tag,
                       dockerfile=args.dockerfile,
                       build_args=build_args,
                       force_rebuild=args.force_rebuild,
                       namespace="")