def list_datasets()

in src/lookoutequipment/dataset.py [0:0]

• 17 lines of code
• 8 McCabe index (conditional complexity)

def list_datasets(dataset_name_prefix=None, max_results=50):
    """
    List all the Lookout for Equipment datasets available in this account.
    
    Parameters:
        dataset_name_prefix (string):
            prefix to filter out all the datasets which names starts by 
            this prefix. Defaults to None to list all datasets.
            
        max_results (integer):
            Max number of datasets to return (default: 50)
            
    Returns:
        list of strings:
            A list with all the dataset names found in the current region
    """
    # Initialization:
    dataset_list = []
    has_more_records = True
    client = boto3.client('lookoutequipment')
    
    # Building the request:
    kargs = {"MaxResults": max_results}
    if dataset_name_prefix is not None:
        kargs["DatasetNameBeginsWith"] = dataset_name_prefix
    
    # We query for the list of datasets, until there are none left to fetch:
    while has_more_records:
        # Query for the list of L4E datasets available for this AWS account:
        list_datasets_response = client.list_datasets(**kargs)
        if "NextToken" in list_datasets_response:
            kargs["NextToken"] = list_datasets_response["NextToken"]
        else:
            has_more_records = False
        
        # Add the dataset names to the list:
        dataset_summaries = list_datasets_response["DatasetSummaries"]
        for dataset_summary in dataset_summaries:
            dataset_list.append(dataset_summary['DatasetName'])
    
    return dataset_list