From f27121bd0c10d49d0cfe8bc42678713b6f4e9094 Mon Sep 17 00:00:00 2001
From: amar-kamat <amar-kamat@users.noreply.github.com>
Date: Wed, 21 Sep 2022 00:36:06 -0700
Subject: [PATCH 1/2] Support maxage param (default 90 days)

Support `-maxage` or MAX_AGE env. Issue #41
---
 main.py | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/main.py b/main.py
index 458a1e4..6f891df 100644
--- a/main.py
+++ b/main.py
@@ -16,6 +16,7 @@
 import os
 import re
 import boto3
+from datetime import datetime, timezone, timedelta
 
 REGION = None
 DRYRUN = None
@@ -28,6 +29,7 @@ def initialize():
     global DRYRUN
     global IMAGES_TO_KEEP
     global IGNORE_TAGS_REGEX
+    global MAX_AGE
 
     REGION = os.environ.get('REGION', "None")
     DRYRUN = os.environ.get('DRYRUN', "false").lower()
@@ -37,6 +39,7 @@ def initialize():
         DRYRUN = True
     IMAGES_TO_KEEP = int(os.environ.get('IMAGES_TO_KEEP', 100))
     IGNORE_TAGS_REGEX = os.environ.get('IGNORE_TAGS_REGEX', "^$")
+    MAX_AGE = int(os.environ.get('MAX_AGE', 90))
 
 def handler(event, context):
     initialize()
@@ -99,6 +102,10 @@ def discover_delete_images(regionname):
                 registryId=repository['registryId'],
                 repositoryName=repository['repositoryName']):
             for image in response_describeimagepaginator['imageDetails']:
+                # check for age and print it
+                if image['imagePushedAt'] < (datetime.now(timezone.utc) - timedelta(days=MAX_AGE)):
+                    print("**** WARN **** image {}:{} is older than {} days.".format(repository['repositoryName'], image['imageDigest'], MAX_AGE))
+
                 if 'imageTags' in image:
                     tagged_images.append(image)
                 else:
@@ -194,6 +201,7 @@ def delete_images(ecr_client, deletesha, deletetag, repo_id, name):
                         dest='imagestokeep')
     PARSER.add_argument('-region', help='ECR/ECS region', default=None, action='store', dest='region')
     PARSER.add_argument('-ignoretagsregex', help='Regex of tag names to ignore', default="^$", action='store', dest='ignoretagsregex')
+    PARSER.add_argument('-maxage', help='Max allowed age of the image', default='90', action='store', dest='maxage')
 
     ARGS = PARSER.parse_args()
     if ARGS.region:
@@ -203,4 +211,5 @@ def delete_images(ecr_client, deletesha, deletetag, repo_id, name):
     os.environ["DRYRUN"] = ARGS.dryrun.lower()
     os.environ["IMAGES_TO_KEEP"] = ARGS.imagestokeep
     os.environ["IGNORE_TAGS_REGEX"] = ARGS.ignoretagsregex
+    os.environ["MAX_AGE"] = ARGS.maxage
     handler(REQUEST, None)

From c18a70f925bd684a8ae3109d94db805deb2fca20 Mon Sep 17 00:00:00 2001
From: amar-kamat <amar-kamat@users.noreply.github.com>
Date: Wed, 21 Sep 2022 00:38:57 -0700
Subject: [PATCH 2/2] Update main.py

---
 main.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/main.py b/main.py
index 6f891df..ba2a391 100644
--- a/main.py
+++ b/main.py
@@ -104,7 +104,7 @@ def discover_delete_images(regionname):
             for image in response_describeimagepaginator['imageDetails']:
                 # check for age and print it
                 if image['imagePushedAt'] < (datetime.now(timezone.utc) - timedelta(days=MAX_AGE)):
-                    print("**** WARN **** image {}:{} is older than {} days.".format(repository['repositoryName'], image['imageDigest'], MAX_AGE))
+                    print("**** WARN **** image {}@{} is older than {} days.".format(repository['repositoryName'], image['imageDigest'], MAX_AGE))
 
                 if 'imageTags' in image:
                     tagged_images.append(image)