From d792ca8df56ceeb819d02da4c4c6a95cc9255936 Mon Sep 17 00:00:00 2001
From: Dave North <dnorth@signiant.com>
Date: Fri, 20 Oct 2017 13:39:30 -0400
Subject: [PATCH] Pretty well done

---
 src/aws-target-group-cleanup.py | 78 +++++++++++++++++++++++++++++++++
 src/requirements.txt            |  1 +
 2 files changed, 79 insertions(+)
 create mode 100644 src/aws-target-group-cleanup.py
 create mode 100644 src/requirements.txt

diff --git a/src/aws-target-group-cleanup.py b/src/aws-target-group-cleanup.py
new file mode 100644
index 0000000..0b35d93
--- /dev/null
+++ b/src/aws-target-group-cleanup.py
@@ -0,0 +1,78 @@
+import sys
+import boto3
+import argparse
+import pprint
+
+
+def remove_target_group(arn, elb_client):
+    request_id = None
+
+    try:
+        response = elb_client.delete_target_group(
+            TargetGroupArn=arn,
+        )
+    except Exception as e:
+        print("Error removing target group " + arn + " (" + str(e) + ")")
+
+    if response:
+        if response['ResponseMetadata']['HTTPStatusCode'] == 200:
+            request_id = response['ResponseMetadata']['RequestId']
+
+    return request_id
+
+
+def main(argv):
+    plugin_results = dict()
+
+    parser = argparse.ArgumentParser(description='Remove ALB target groups not assigned to load balancers')
+    parser.add_argument('-f','--force', help='Perform the actual deletes (will run in dryrun mode by default)', action='store_true')
+    parser.add_argument('-r','--region', help='AWS region to process', required=True)
+    parser.add_argument('-p','--prefix', help='Only process target groups with this prefix in their name', required=False)
+
+    args = parser.parse_args()
+
+    if args.prefix:
+        print("Prefix specified - only evaluating target groups beginning with " + args.prefix)
+
+    if args.force:
+        print("Force flag specified - doing deletion")
+
+    session = boto3.session.Session(region_name=args.region)
+    elb_client = session.client("elbv2")
+
+    tg_paginator = elb_client.get_paginator('describe_target_groups')
+    tg_iterator = tg_paginator.paginate()
+    for response in tg_iterator:
+        # pprint.pprint(response)
+
+        for target_group in response['TargetGroups']:
+            punt = False
+
+            target_group_name = target_group['TargetGroupName']
+            target_group_load_balancers = target_group['LoadBalancerArns']
+            target_group_arn = target_group['TargetGroupArn']
+
+            if args.prefix:
+                if target_group_name.startswith(args.prefix):
+                    if not target_group_load_balancers:
+                        punt = True
+            else:
+                if not target_group_load_balancers:
+                    punt = True
+
+            if punt:
+                sys.stdout.write("Target group " + target_group_name + " has no load balancers - removing....")
+
+                if args.force:
+                    request_id = remove_target_group(target_group_arn, elb_client)
+                    if request_id:
+                        print("REMOVED (request id " + request_id + ")")
+                    else:
+                        print("ERROR")
+                else:
+                    print("DRYRUN,no deletion took place")
+
+    print "Complete"
+
+if __name__ == "__main__":
+   main(sys.argv[1:])
diff --git a/src/requirements.txt b/src/requirements.txt
new file mode 100644
index 0000000..30ddf82
--- /dev/null
+++ b/src/requirements.txt
@@ -0,0 +1 @@
+boto3