narenas
/
opengnsys_ipxe
mirror of https://github.com/ipxe/ipxe.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

[cloud] Add ability to overwrite existing AMI images 

AMI names must be unique within a region.  Add a --overwrite option
that allows an existing AMI of the same name to be deregistered (and
its underlying snapshot deleted).

Signed-off-by: Michael Brown <mcb30@ipxe.org>
pull/1082/head
Michael Brown 2023-11-07 15:54:59 +00:00
parent 595b1796f6
commit d8f9c221ed
1 changed files with 13 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
contrib/cloud/aws-import
View File

@ -46,11 +46,19 @@ def create_snapshot(region, description, image):
return snapshot_id return snapshot_id
def import_image(region, name, architecture, image, public): def import_image(region, name, architecture, image, public, overwrite):
"""Import an AMI image""" """Import an AMI image"""
client = boto3.client('ec2', region_name=region) client = boto3.client('ec2', region_name=region)
resource = boto3.resource('ec2', region_name=region) resource = boto3.resource('ec2', region_name=region)
description = '%s (%s)' % (name, architecture) description = '%s (%s)' % (name, architecture)
images = client.describe_images(Filters=[{'Name': 'name',
'Values': [description]}])
if overwrite and images['Images']:
images = images['Images'][0]
image_id = images['ImageId']
snapshot_id = images['BlockDeviceMappings'][0]['Ebs']['SnapshotId']
resource.Image(image_id).deregister()
resource.Snapshot(snapshot_id).delete()
snapshot_id = create_snapshot(region=region, description=description, snapshot_id = create_snapshot(region=region, description=description,
image=image) image=image)
client.get_waiter('snapshot_completed').wait(SnapshotIds=[snapshot_id]) client.get_waiter('snapshot_completed').wait(SnapshotIds=[snapshot_id])
@ -88,6 +96,8 @@ parser.add_argument('--name', '-n',
help="Image name") help="Image name")
parser.add_argument('--public', '-p', action='store_true', parser.add_argument('--public', '-p', action='store_true',
help="Make image public") help="Make image public")
parser.add_argument('--overwrite', action='store_true',
help="Overwrite any existing image with same name")
parser.add_argument('--region', '-r', action='append', parser.add_argument('--region', '-r', action='append',
help="AWS region(s)") help="AWS region(s)")
parser.add_argument('--wiki', '-w', metavar='FILE', parser.add_argument('--wiki', '-w', metavar='FILE',
@ -115,7 +125,8 @@ with ThreadPoolExecutor(max_workers=len(imports)) as executor:
name=args.name, name=args.name,
architecture=architectures[image], architecture=architectures[image],
image=image, image=image,
public=args.public): (region, image) public=args.public,
overwrite=args.overwrite): (region, image)
for region, image in imports} for region, image in imports}
results = {futures[future]: future.result() results = {futures[future]: future.result()
for future in as_completed(futures)} for future in as_completed(futures)}