AWS RDS – Lambdaを用いたRDSのスナップショット取得＆世代管理

import boto3
import time
from botocore.client import ClientError
from datetime import datetime, timedelta, tzinfo

rds = boto3.client('rds')

class JST(tzinfo):
    def utcoffset(self, dt):
        return timedelta(hours=9)
    def dst(self, dt):
        return timedelta(0)
    def tzname(self, dt):
        return 'JST'

def create_snapshot(prefix, instanceid):
    snapshotid = "-".join([prefix, datetime.now().strftime("%Y-%m-%d-%H-%M")])
    
    for i in range(0, 5):
        try:
            snapshot = rds.create_db_snapshot(
                DBSnapshotIdentifier=snapshotid,
                DBInstanceIdentifier=instanceid
            )
            return
        except ClientError as e:
                print(str(e))
        time.sleep(1)

def delete_snapshots(prefix, days):
    snapshots = rds.describe_db_snapshots()
    now = datetime.utcnow().replace(tzinfo=None)
    for snapshot in snapshots['DBSnapshots']:
        # creating snapshot
        if not snapshot.has_key('SnapshotCreateTime'):
            continue
        
        delta = now - snapshot['SnapshotCreateTime'].replace(tzinfo=None)
        if snapshot['DBSnapshotIdentifier'].startswith(prefix) and delta.days >= days:
            rds.delete_db_snapshot(DBSnapshotIdentifier=snapshot['DBSnapshotIdentifier'])

def lambda_handler(event, context):
    delete_days = 60
    snapshot_prefix = "rds-INSTANCENAME"
    instances = ["INSTANCENAME"]
    
    for instance in instances:
        create_snapshot(snapshot_prefix, instance)
        
    delete_snapshots(snapshot_prefix, delete_days)