def lambda_handler()

in ml_ops/visualization_blog/lambdas/createdatasetgroup/datasetgroup.py [0:0]

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

def lambda_handler(event, context):
    dataset_group = event['params']['DatasetGroup']
    status = None
    event['DatasetGroupArn'] = ARN.format(
        account=event['AccountID'],
        name=dataset_group['DatasetGroupName'],
        region=environ['AWS_REGION']
    )
    try:
        status = LOADER.forecast_cli.describe_dataset_group(
            DatasetGroupArn=event['DatasetGroupArn']
        )

    except LOADER.forecast_cli.exceptions.ResourceNotFoundException:
        LOADER.logger.info(
            'Dataset Group not found! Will follow to create Dataset Group.'
        )
        LOADER.forecast_cli.create_dataset_group(
            **dataset_group, DatasetArns=[event['DatasetArn']]
        )
        status = LOADER.forecast_cli.describe_dataset_group(
            DatasetGroupArn=event['DatasetGroupArn']
        )

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