add py/EBS-Snapgots.py.

py/EBS-Snapgots.py

@@ -0,0 +1,73 @@
+import boto3
+import os,time
+from botocore.exceptions import ClientError
+from datetime import datetime, timedelta, timezone
+
+AK = "AKIAR62ETM5M"
+SK = "B52Y/Rv0h2nOYjoE6"
+
+
+client = boto3.client('ec2',region_name='cn-northwest-1',aws_access_key_id=AK,
+    aws_secret_access_key=SK)
+ec2 = boto3.resource('ec2',region_name='cn-northwest-1',aws_access_key_id=AK,
+    aws_secret_access_key=SK)
+ddb = boto3.resource('dynamodb',region_name='cn-northwest-1',aws_access_key_id=AK,
+    aws_secret_access_key=SK)
+
+# 'EBS-Snapshot-Lifecycle' is the table you store snapshot metadata, modify it to your own table name
+table = ddb.Table('EBS-Snapshot-Lifecycle')
+
+
+DAYS_TO_RETAIN = 7
+
+def lambda_handler(event, context):
+
+    os.environ['TZ'] = 'Asia/Shanghai'
+    time.tzset()
+    i=time.strftime('%X %x %Z')
+    
+    volume_id_list = ['vol-0cb469d952fcf6108']
+
+    # set snapshot
+    for volume_id in volume_id_list:
+        volume = ec2.Volume(volume_id)
+    
+        for tags in volume.tags:
+            if(tags.get('Key') == 'Name'):
+                volume_name = tags.get('Value')
+        description = volume_name + ' volume snapshot is created at ' + i
+        
+        try:
+            response = client.create_snapshot(
+                Description=description,
+                # Description=i + ' -- ',
+                VolumeId=volume_id)
+        except Exception as e:
+            print('Create Snapshot occured error, Volume id is ' + volume_id)
+        else:
+            print('Snapshot is created succeed, Snapshot id is ' + response.get('SnapshotId'))
+        
+        # write into DynanoDB table
+        '''
+        snapshot_id = response.get('SnapshotId')
+        start_time = response.get('StartTime')
+        ttl_time = int(start_time.timestamp()) + 86400 * DAYS_TO_RETAIN
+    
+        try:
+            table.put_item(
+                Item={
+                    'snapshot_id': snapshot_id,
+                    'create_time': start_time.astimezone(timezone(timedelta(hours=8))).strftime('%X %x %Z'),
+                    'description': description,
+                    'timestamp': int(start_time.timestamp()),
+                    'ttl': ttl_time,
+                }
+            )
+        except:
+            print('Write to DynanoDB Table occured error, snapshot id is ' + snapshot_id)
+        else:
+            print('Write to DynanoDB Table succeed, snapshot id is ' + snapshot_id)
+
+        '''
+if __name__ == '__main__':
+    lambda_handler(None,None)