53 lines
2.3 KiB
Python
53 lines
2.3 KiB
Python
|
from subprocess import run
|
||
|
|
from typing import Literal, get_args
|
||
|
|
|
||
|
|
Platforms = Literal["linux/amd64", "linux/amd64/v2", "linux/amd64/v3", "linux/arm64", "linux/riscv64", "linux/ppc64le", "linux/s390x", "linux/386", "linux/mips64le", "linux/mips64", "linux/arm/v7", "linux/arm/v6"]
|
||
|
|
supported_platforms: list[Platforms] = list(get_args(Platforms))
|
||
|
|
|
||
|
|
Progress = Literal["auto", "plain", "tty"]
|
||
|
|
|
||
|
|
def semver_tags(version: str):
|
||
|
|
parts = version.split('.')
|
||
|
|
return list('.'.join(parts[:i]) for i in range(1, len(parts) + 1))
|
||
|
|
|
||
|
|
def pull(repository: str, tag: str):
|
||
|
|
label = f"{repository}:{tag}"
|
||
|
|
run(["docker", "pull", label], check=True)
|
||
|
|
return label
|
||
|
|
|
||
|
|
def push_image(repository: str, tag: str):
|
||
|
|
label = f"{repository}:{tag}"
|
||
|
|
run(["docker", "push", label], check=True)
|
||
|
|
return label
|
||
|
|
|
||
|
|
def create_manifest(repository: str, manifest_tag: str, image_tags: list[str], push: bool = True, write_command: bool = False):
|
||
|
|
raise Exception("Creation of manifests is not yet supported!")
|
||
|
|
manifest = f"{repository}:{manifest_tag}"
|
||
|
|
images = [f"{repository}:{tag}" for tag in image_tags]
|
||
|
|
for image_tag in image_tags:
|
||
|
|
pull(repository, image_tag)
|
||
|
|
command = ["docker", "manifest", "create", manifest, *images]
|
||
|
|
if write_command:
|
||
|
|
print(" ".join(command))
|
||
|
|
run(command, check=True)
|
||
|
|
if push:
|
||
|
|
return push_image(repository, manifest_tag)
|
||
|
|
return manifest
|
||
|
|
|
||
|
|
def buildx(repository: str, tags: list[str], build_platforms: list[Platforms], dockerfile: str = "Dockerfile", build_args: dict[str, str] | None = None, directory: str = ".", push: bool = True, pull: bool = False, progress: Progress = "auto", write_command: bool = False):
|
||
|
|
if build_args is None:
|
||
|
|
build_args = dict()
|
||
|
|
labels = [f"{repository}:{tag}" for tag in tags]
|
||
|
|
command = list(filter(None, ["docker", "buildx", "build",
|
||
|
|
"--platform", ",".join(build_platforms),
|
||
|
|
*[t for (key, value) in build_args.items() for t in ("--build-arg", f"{key}={value}")],
|
||
|
|
"--file", dockerfile,
|
||
|
|
*[t for label in labels for t in ("--tag", label)],
|
||
|
|
f"--progress={progress}",
|
||
|
|
"--pull" if pull else None,
|
||
|
|
"--push" if push else None,
|
||
|
|
directory]))
|
||
|
|
if write_command:
|
||
|
|
print(" ".join(command))
|
||
|
|
run(command, check=True)
|