def lambda_handler()

in lambda/build/lambda_pipeline_change.py [0:0]


def lambda_handler(event, context):
    try:
        logger.debug(json.dumps(event))
        if (
            event.get("source") == "aws.sagemaker"
            and event.get("detail-type")
            == "SageMaker Model Building Pipeline Execution Status Change"
        ):
            status_code, result = update_pipeline_rules(event)
        else:
            raise Exception("Expect SageMaker Model Package State Change")
        logger.debug(json.dumps(result))
        return {"statusCode": status_code, "body": json.dumps(result)}
    except ClientError as e:
        error_message = e.response["Error"]["Message"]
        logger.error(error_message)
        raise Exception(error_message)
    except Exception as e:
        logger.error(e)
        raise e