faasmtools/docker.py

from subprocess import run

ACR_NAME = "faasm.azurecr.io"


def build_container(
    tag_name, dockerfile, cwd, nocache=False, push=False, build_args=None
):
    build_args = build_args if build_args else dict()

    if nocache:
        no_cache_str = "--no-cache"
    else:
        no_cache_str = ""

    build_cmd = [
        "docker build",
        no_cache_str,
        "-t {}".format(tag_name),
        "-f {}".format(dockerfile),
    ]

    for key, value in build_args.items():
        build_cmd.append("--build-arg {}={}".format(key, value))

    build_cmd.append(".")

    build_cmd = " ".join(build_cmd)

    print(build_cmd)
    run(
        build_cmd,
        shell=True,
        check=True,
        env={"DOCKER_BUILDKIT": "1"},
        cwd=cwd,
    )

    if push:
        push_container(tag_name)


def push_container(tag_name):
    cmd = "docker push {}".format(tag_name)

    print(cmd)
    run(
        cmd,
        shell=True,
        check=True,
    )