awsRestore.py

#!/usr/bin/env python3
import boto3
from botocore.exceptions import ClientError
import sys
import argparse

try:
    #add parser for command line arguments. User adds prefix/folder-name, bucket name, and Days restored file available for download
    parser = argparse.ArgumentParser(description = 'add parameters to restore')
    parser.add_argument('--prefix', action = 'store', help="the object/file path prefix or folders to restore")
    parser.add_argument('--days', action = 'store', help="days to keep file restored before going back to archive")
    parser.add_argument('--bucket', action = 'store', help="bucket to restore from")
    parser.add_argument('--dryrun', action='store_true')
    args = parser.parse_args()
    Days = int(args.days)
    bucketname = args.bucket
    #get prefix of the folder you would like to restore from glacier (input on command line)
    prefix = args.prefix
    dryrun = args.dryrun
except Exception as e:
    print(e)
    sys.exit()

try:
    s3 = boto3.resource('s3')
    bucket = s3.Bucket(bucketname)
except ClientError as e:
    print(e.response)
    sys.exit()

try:
    s3Client = boto3.client('s3')
    #add paginator to allow for larger than 1000 objects restore
    paginator = s3Client.get_paginator('list_objects')
    page_iterator = paginator.paginate(Bucket=bucketname, Prefix=prefix, PaginationConfig={'PageSize':1000})
except ClientError as e:
    print(e.response)
    sys.exit()

try:
    #loop through pages to restore
    for page in page_iterator:
            #loop through bucket objects to restore
            for obj in page['Contents']:
                    #if the folder/prefix name is in the object key continue to restore
                    if prefix in obj['Key']:
                        #variable for current object
                        obj = s3.Object(bucketname, obj['Key'])
                        #check to see if object is in glacier deep archive
                        if obj.storage_class == 'DEEP_ARCHIVE':
                            #if restore status is blank continue to request restore
                            if obj.restore is None:
                                    print('Submitting restoration request: %s' % obj.key)
                                    #restore object with Days parameter for the number of days restore is available for downloading
                                    if dryrun:
                                       print('\tDryrun: would have restored: %s' %obj.key)
                                    else:
                                       obj.restore_object(RestoreRequest={'Days': Days, 'GlacierJobParameters': {'Tier': 'Bulk'}})
                                       print('\tRestoring: %s' %obj.key)
                            #if object is being restored print status
                            elif 'ongoing-request="true"' in obj.restore:
                                print('Restoration in-progress: %s' % obj.key)
                            #if object is done being restored print status
                            elif 'ongoing-request="false"' in obj.restore:
                                print('Restoration complete: %s' % obj.key)
except ClientError as e:
    print(e.response)
    sys.exit()