#!/usr/bin/env python3

import boto3
import sys


repository_name = sys.argv[1]
client = boto3.client('ecr')
paginator = client.get_paginator('list_images')
page_iterator = paginator.paginate(
    repositoryName=repository_name,
    filter={
        'tagStatus': 'UNTAGGED'
    }
)

for page in page_iterator:
    response = client.batch_delete_image(
        repositoryName=repository_name,
        imageIds=[image for image in page['imageIds']]
    )