def lambda_handler()

in workflow1_endpointbuilder/sam-app/functions/function1_createtableandfunction3trigger/index.py [0:0]


def lambda_handler(event, context):
    folder_uri = event["folder_uri"]
    bucket_name = urlparse(folder_uri).hostname
    datetime_id = datetime.datetime.now().strftime("%Y%m%d%H%M%S")

    create_dataset_table(datetime_id, bucket_name)

    dynamodb_client = boto3.client('dynamodb')
    while True:
        response = dynamodb_client.describe_table(
            TableName=f'DatasetCSVTable_{datetime_id}_{bucket_name}_'
        )
        status = response["Table"]["TableStatus"]
        if status == "ACTIVE":
            stream_arn = response['Table']['LatestStreamArn']
            break

    create_function3_trigger(stream_arn)

    return {
        'datetime_id': datetime_id,
        'folder_uri': folder_uri
    }