Example script to generate a callback script to run

exampleStepFunctionCallbackScript.py

def generateScriptStepFunctions(programName, taskToken, operationName):
        secretName = 'exampleSecretsManager'
        regionName = 'us-east-1'
        secretRetrieved = brevitycore.core.get_secret(secretName,regionName)
        secretjson = json.loads(secretRetrieved)
        awsAccessKeyId = secretjson['AWS_ACCESS_KEY_ID']
        awsSecretKey = secretjson['AWS_SECRET_ACCESS_KEY']
   
        stateInput = '{"program":"' + programName + '","operation":"' + operationName + '","statusCode":200}'
        fileBuffer = io.StringIO()
    fileContents = f"""#!/bin/bash

export AWS_ACCESS_KEY_ID={awsAccessKeyId}
export AWS_SECRET_ACCESS_KEY={awsSecretKey}
export AWS_DEFAULT_REGION=us-east-1

aws stepfunctions send-task-success --task-token {taskToken} --task-output '{stateInput}'"""

    fileBuffer.write(fileContents)
    objectBuffer = io.BytesIO(fileBuffer.getvalue().encode())
    # Upload file to S3
    object_name = 'stepfunctions-' + programName + '.sh'
    object_path = 'run/' + programName + '/' + object_name
    status = brevitycore.core.upload_object(objectBuffer,inputBucketName,object_path)
    fileBuffer.close()
    objectBuffer.close()
    return status