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)