CopyOCPAMI.py - A utility to copy a RHEL CoreOS AMI Image to a different region/account.

CopyOCPAMI.py

#!/usr/bin/env python
"""CopyOCPAMI.py - A utility to copy OpenShift Amazon Machine Images

Copyright (C) 2020 Brian 'redbeard' Harrington

This program is free software: you can redistribute it and/or modify
it under the terms of the GNU Affero General Public License as
published by the Free Software Foundation, either version 3 of the
License, or (at your option) any later version.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU Affero General Public License for more details.

You should have received a copy of the GNU Affero General Public License
along with this program.  If not, see <https://www.gnu.org/licenses/>.
"""

import sys
import uuid

import boto3
import requests

RHCOS_BUILDS = "https://raw.githubusercontent.com/openshift/installer/master/data/data/rhcos.json"

try:
    DEST_REGION = sys.argv[1]
except IndexError:
    print("ERROR: You must supply the destination ",
          "region in which you wish to operate")
    sys.exit(1)

CLIENT = boto3.client('ec2', region_name=DEST_REGION)


def get_ocp_info(region="us-east-1"):
    """Retrieve OpenShift metadata from GitHub

    Args:
        region (str, optional): Region of the AMI ID to return. Defaults to "us-east-1".

    Returns:
        ami: AMI ID
    """
    builds = requests.get(RHCOS_BUILDS).json()
    return builds["amis"][region]


def copy_ami(ami, sregion="us-east-1"):
    """Copy an AMI to a different region

    Args:
        ami (str): Amazon Machine Image (AMI) Id that you wish to copy
        sregion (str, optional): Source Region. Defaults to "us-east-1".
    """
    # Get metadata on the AMI hosted in sregion
    image = boto3.resource('ec2', region_name=sregion).Image(ami)

    # Generate a token to ensure the operation is idempotent
    client_token = str(uuid.uuid4())

    # Use copy_image - https://boto3.amazonaws.com/
    # https://bit.ly/3jZy4JN
    print("Copying image {} from region {}".format(ami, sregion))
    response = ""
    try:
        response = CLIENT.copy_image(
            ClientToken=client_token,
            Name=image.name,
            SourceImageId=ami,
            SourceRegion=sregion
        )
    except Exception as err:
        print("ERROR: Unable to copy image - {}".format(err))

    print("Image copy executed")
    print("New AMI: {}\nRequestId: {}".format(
        response["ImageId"], response["ResponseMetadata"]["RequestId"]
    ))


def main():
    """Copy an AMI ID across Amazon accounts/regions
    """
    ami_id = get_ocp_info()["hvm"]

    copy_ami(ami=ami_id)


if __name__ == "__main__":
    try:
        main()
    except KeyboardInterrupt:
        sys.exit(0)