| #!/usr/bin/env python |
| |
| # Licensed to the Apache Software Foundation (ASF) under one or more |
| # contributor license agreements. See the NOTICE file distributed with |
| # this work for additional information regarding copyright ownership. |
| # The ASF licenses this file to You under the Apache License, Version 2.0 |
| # (the "License"); you may not use this file except in compliance with |
| # the License. You may obtain a copy of the License at |
| # |
| # http://www.apache.org/licenses/LICENSE-2.0 |
| # |
| # Unless required by applicable law or agreed to in writing, software |
| # distributed under the License is distributed on an "AS IS" BASIS, |
| # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| # See the License for the specific language governing permissions and |
| # limitations under the License. |
| |
| """ |
| Python script to build and push a multiarch docker image |
| This script is used to prepare and publish docker release candidate |
| |
| Pre requisites: |
| Ensure that you are logged in the docker registry and you have access to push to that registry. |
| Ensure that docker buildx is enabled for you. |
| |
| Usage: |
| docker_release.py --help |
| Get detailed description of argument |
| |
| Example command:- |
| docker_release <image> --kafka-url <kafka_url> --image-type <type> |
| |
| This command will build the multiarch image of type <type> (jvm by default), |
| named <image> using <kafka_url> to download kafka and push it to the docker image name <image> provided. |
| Make sure image is in the format of <registry>/<namespace>/<image_name>:<image_tag>. |
| """ |
| |
| from datetime import date |
| import argparse |
| |
| from common import execute, build_docker_image_runner |
| |
| def build_push(image, kafka_url, image_type): |
| try: |
| create_builder() |
| build_docker_image_runner(f"docker buildx build -f $DOCKER_FILE --build-arg kafka_url={kafka_url} --build-arg build_date={date.today()} --push \ |
| --platform linux/amd64,linux/arm64 --tag {image} $DOCKER_DIR", image_type) |
| except: |
| raise SystemError("Docker image push failed") |
| finally: |
| remove_builder() |
| |
| def create_builder(): |
| execute(["docker", "buildx", "create", "--name", "kafka-builder", "--use"]) |
| |
| def remove_builder(): |
| execute(["docker", "buildx", "rm", "kafka-builder"]) |
| |
| if __name__ == "__main__": |
| print("\ |
| This script will build and push docker images of apache kafka.\n \ |
| Please ensure that image has been sanity tested before pushing the image. \n \ |
| Please ensure you are logged in the docker registry that you are trying to push to.") |
| parser = argparse.ArgumentParser() |
| parser.add_argument("image", help="Dockerhub image that you want to push to (in the format <registry>/<namespace>/<image_name>:<image_tag>)") |
| parser.add_argument("--image-type", "-type", choices=["jvm", "native"], default="jvm", dest="image_type", help="Image type you want to build") |
| parser.add_argument("--kafka-url", "-u", dest="kafka_url", help="Kafka url to be used to download kafka binary tarball in the docker image") |
| args = parser.parse_args() |
| |
| print(f"Docker image of type {args.image_type} containing kafka downloaded from {args.kafka_url} will be pushed to {args.image}") |
| |
| print("Building and pushing the image") |
| build_push(args.image, args.kafka_url, args.image_type) |
| print(f"Image has been pushed to {args.image}") |