def lambda_handler()

in ml_ops/visualization_blog/lambdas/createdataset/dataset.py [0:0]

• 22 lines of code
• 3 McCabe index (conditional complexity)

def lambda_handler(event, context):
    datasets = event['params']['Datasets']
    status = None
    event['DatasetArn'] = ARN.format(
        account=ACCOUNTID,
        name=datasets[0]['DatasetName'],
        region=environ['AWS_REGION']
    )
    event['AccountID'] = ACCOUNTID
    try:
        status = LOADER.forecast_cli.describe_dataset(
            DatasetArn=event['DatasetArn']
        )
    except LOADER.forecast_cli.exceptions.ResourceNotFoundException:
        LOADER.logger.info('Dataset not found! Will follow to create dataset.')
        for dataset in datasets:
            LOADER.forecast_cli.create_dataset(**dataset)
        status = LOADER.forecast_cli.describe_dataset(
            DatasetArn=event['DatasetArn']
        )

    actions.take_action(status['Status'])
    return event