josjaf · March 19, 2018 14:46 · Dec 19, 2016 · Jan 7, 2016
diff --git a/lambdaAMIBackups.py b/lambdaAMIBackups.py
@@ -1,6 +1,6 @@
 # Automated AMI Backups
 #
-# @author Robert Kozora <[email protected]>
+# @author Robert Kozora <[email protected]>
 #
 # This script will search for all instances having a tag with "Backup" or "backup"
 # on it. As soon as we have the instances list, we loop through each instance
@@ -14,8 +14,11 @@
 import boto3
 import collections
 import datetime
+import sys
+import pprint
 
 ec = boto3.client('ec2')
+#image = ec.Image('id')
 
 def lambda_handler(event, context):
 
@@ -44,8 +47,7 @@ def lambda_handler(event, context):
                 if t['Key'] == 'Retention'][0]
         except IndexError:
             retention_days = 7
-
-        # Loop through instance's EBS volumes
+
         #for dev in instance['BlockDeviceMappings']:
         #    if dev.get('Ebs', None) is None:
         #        continue
@@ -60,22 +62,34 @@ def lambda_handler(event, context):
             #create_image(instance_id, name, description=None, no_reboot=False, block_device_mapping=None, dry_run=False)
             # DryRun, InstanceId, Name, Description, NoReboot, BlockDeviceMappings
             create_time = datetime.datetime.now()
-            create_fmt = create_time.strftime('%Y-%m-%d.%H.%M.%S')
+            create_fmt = create_time.strftime('%Y-%m-%d')
 
-            AMIid = ec.create_image(InstanceId=instance['InstanceId'], Name="Lambda - " + instance['InstanceId'] + " From " + create_fmt, Description="Lambda created AMI of instance " + instance['InstanceId'], NoReboot=True, DryRun=False)
-            to_tag[retention_days].append(AMIid['ImageId'])
+            AMIid = ec.create_image(InstanceId=instance['InstanceId'], Name="Lambda - " + instance['InstanceId'] + " from " + create_fmt, Description="Lambda created AMI of instance " + instance['InstanceId'] + " from " + create_fmt, NoReboot=True, DryRun=False)
 
+
+            pprint.pprint(instance)
+            #sys.exit()
+            #break
+
+            #to_tag[retention_days].append(AMIid)
+
+            to_tag[retention_days].append(AMIid['ImageId'])
+
             print "Retaining AMI %s of instance %s for %d days" % (
                 AMIid['ImageId'],
                 instance['InstanceId'],
                 retention_days,
             )
 
+    print to_tag.keys()
+
     for retention_days in to_tag.keys():
         delete_date = datetime.date.today() + datetime.timedelta(days=retention_days)
         delete_fmt = delete_date.strftime('%m-%d-%Y')
         print "Will delete %d AMIs on %s" % (len(to_tag[retention_days]), delete_fmt)
-
+
+        #break
+
         ec.create_tags(
             Resources=to_tag[retention_days],
             Tags=[

diff --git a/lambdaAMIBackups.py b/lambdaAMIBackups.py
@@ -0,0 +1,85 @@
+# Automated AMI Backups
+#
+# @author Robert Kozora <[email protected]>
+#
+# This script will search for all instances having a tag with "Backup" or "backup"
+# on it. As soon as we have the instances list, we loop through each instance
+# and create an AMI of it. Also, it will look for a "Retention" tag key which
+# will be used as a retention policy number in days. If there is no tag with
+# that name, it will use a 7 days default value for each AMI.
+#
+# After creating the AMI it creates a "DeleteOn" tag on the AMI indicating when
+# it will be deleted using the Retention value and another Lambda function 
+
+import boto3
+import collections
+import datetime
+
+ec = boto3.client('ec2')
+
+def lambda_handler(event, context):
+
+    reservations = ec.describe_instances(
+        Filters=[
+            {'Name': 'tag-key', 'Values': ['backup', 'Backup']},
+        ]
+    ).get(
+        'Reservations', []
+    )
+
+    instances = sum(
+        [
+            [i for i in r['Instances']]
+            for r in reservations
+        ], [])
+
+    print "Found %d instances that need backing up" % len(instances)
+
+    to_tag = collections.defaultdict(list)
+
+    for instance in instances:
+        try:
+            retention_days = [
+                int(t.get('Value')) for t in instance['Tags']
+                if t['Key'] == 'Retention'][0]
+        except IndexError:
+            retention_days = 7
+
+        # Loop through instance's EBS volumes
+        #for dev in instance['BlockDeviceMappings']:
+        #    if dev.get('Ebs', None) is None:
+        #        continue
+        #    vol_id = dev['Ebs']['VolumeId']
+        #    print "Found EBS volume %s on instance %s" % (
+        #        vol_id, instance['InstanceId'])
+
+            #snap = ec.create_snapshot(
+            #    VolumeId=vol_id,
+            #)
+
+            #create_image(instance_id, name, description=None, no_reboot=False, block_device_mapping=None, dry_run=False)
+            # DryRun, InstanceId, Name, Description, NoReboot, BlockDeviceMappings
+            create_time = datetime.datetime.now()
+            create_fmt = create_time.strftime('%Y-%m-%d.%H.%M.%S')
+
+            AMIid = ec.create_image(InstanceId=instance['InstanceId'], Name="Lambda - " + instance['InstanceId'] + " From " + create_fmt, Description="Lambda created AMI of instance " + instance['InstanceId'], NoReboot=True, DryRun=False)
+            to_tag[retention_days].append(AMIid['ImageId'])
+
+            print "Retaining AMI %s of instance %s for %d days" % (
+                AMIid['ImageId'],
+                instance['InstanceId'],
+                retention_days,
+            )
+
+    for retention_days in to_tag.keys():
+        delete_date = datetime.date.today() + datetime.timedelta(days=retention_days)
+        delete_fmt = delete_date.strftime('%m-%d-%Y')
+        print "Will delete %d AMIs on %s" % (len(to_tag[retention_days]), delete_fmt)
+
+        ec.create_tags(
+            Resources=to_tag[retention_days],
+            Tags=[
+                {'Key': 'DeleteOn', 'Value': delete_fmt},
+            ]
+        )
+