AWS Lambda AMI Cleaner

cleaner.py

from amicleaner.cli import App as Cleaner
from amicleaner.utils import parse_args


# AWS Lambda AMI Cleaner
class AmiCleaner:

    # Initialize the AMI Cleaner
    def __init__(self, cleaner=None):

        # Initialize the AMI Cleaner
        self.cleaner = cleaner or Cleaner(parse_args([
            '--keep-previous', '3',
            '--mapping-key', 'tags',
            '--mapping-values', 'Name'
        ]))

    # Clean AMIs
    def clean(self):

        # Resolve AMIs
        amis = self.cleaner.prepare_candidates()

        # Delete AMIs
        if amis:
            self.cleaner.prepare_delete_amis(amis)

handler.py

from cleaner import AmiCleaner


# AWS Lambda AMI Cleaner Handler
def handle(event, context, cleaner=None):

    # Clean AMIs
    AmiCleaner(cleaner).clean()

requirements.txt

aws-amicleaner==0.2.2
future==0.16.0