PollToPush_OpenConnection.py

import boto3
import os
import json
client = boto3.client('stepfunctions')

def lambda_handler(event, context):
    print(event)
    messageBody = json.loads(event['body'])
    task_token = messageBody['task_token']
    executionArn = messageBody['executionArn']

    describe_response = client.describe_execution(
        executionArn=executionArn
    )
    if describe_response == '':
        return {
            'statusCode': 502,
            'body': 'Invalid Execution ARN'
        }
                    # send task success
    activity_response = client.send_task_success(
        taskToken=task_token,
        output="{\"socket\":\"opened\", \"connectionId\":\"" + event['requestContext']['connectionId'] + "\"}"
    )
                    
    return {
        'statusCode': 200,
        'body': event['requestContext']['connectionId']
    }