def lambda_handler()

in automation/ml_ops/lambdas/create-dataset/dataset.py [0:0]

• 27 lines of code
• 2 McCabe index (conditional complexity)

def lambda_handler(event, context):
    # return event
    dataset = event['datasets'][event['datasetType']]
    datasetArn = DATASET_ARN.format(
        region=environ['AWS_REGION'],
        account=LOADER.account_id,
        datasetGroupName=event['datasetGroupName'],
        type=str.upper(event['datasetType'])
    )
    try:
        status = LOADER.personalize_cli.describe_dataset(datasetArn=datasetArn
                                                        )['dataset']

    except LOADER.personalize_cli.exceptions.ResourceNotFoundException:
        LOADER.logger.info(
            'Dataset not found! Will follow to create schema and dataset.'
        )
        LOADER.personalize_cli.create_dataset(
            name=dataset['name'],
            schemaArn=create_schema(
                dataset['schema']['name'], dataset['schema']
            ),
            datasetGroupArn=event['datasetGroupArn'],
            datasetType=event['datasetType']
        )
        status = LOADER.personalize_cli.describe_dataset(datasetArn=datasetArn
                                                        )['dataset']

    actions.take_action(status['status'])
    return datasetArn