import boto3
import cfnresponse
def handler(event, context):
    # Init ...
    '''
    This function Creates required directory structure inside S3 bucket
    '''
    the_event = event['RequestType']
    print("The event is: ", str(the_event))
    response_data = {}
    s_3 = boto3.client('s3')
    s3_resource = boto3.resource('s3')
    # Retrieve parameters
    the_bucket = event['ResourceProperties']['the_bucket']
    dirs_to_create = event['ResourceProperties']['dirs_to_create']
    file_content = event['ResourceProperties']['file_content']
    file_prefix = event['ResourceProperties']['file_prefix']
    print("file_content is : " + file_content)
    print("file_prefix is : " + file_prefix)
    try:
        if the_event in ('Create', 'Update'):
            print("Requested folders: ", str(dirs_to_create))
            for dir_name in dirs_to_create.split(","):
                print("Creating: ", str(dir_name))
                s_3.put_object(Bucket=the_bucket,
                                Key=(dir_name
                                    + '/'))
            # if  file_prefix == "" or  file_content == "" :          
            s3_resource.Object(the_bucket,file_prefix).put(Body=file_content)
            print("file created")
            # else:
            #     print("No file created")
        elif the_event == 'Delete':
            print("Deleting S3 content...")
            #s3_resource.Bucket(str(the_bucket)).objects.all().delete()
            bucket = s3_resource.Bucket(the_bucket)
            bucket.objects.all().delete()
            print("All object deleted")
            bucket.object_versions.delete()
            print("All object version deleted")
            s_3.delete_bucket(Bucket=str(the_bucket))
            print("Bucket deleted")
        # Everything OK... send the signal back
        print("Execution succesfull!")
        cfnresponse.send(event,
                        context,
                        cfnresponse.SUCCESS,
                        response_data)
    except Exception as e:
        print("Execution failed...")
        print(str(e))
        response_data['Data'] = str(e)
        cfnresponse.send(event,
                        context,
                        cfnresponse.FAILED,
                        response_data)