data¶

def import_data(dataset_path: str = dish_dataset_path) -> pandas.core.frame.DataFrame:
    """Imports dataset.
    Args:
      dataset_path (str): Path of dataset file.
    Returns:
      dataset (dataset values): Datas loaded from csv file.
    """

    dish_dataset = pandas.read_csv(dataset_path)
    return dish_dataset

def list_of_dishes(length: int) -> int:
    """List the dishes with given length.
    Args:
      length (int): Length to display dishes.
    Returns:
      length (int): Validated length of dishes to display.
    """

    # Check whether the input length is valid
    if validate_length(length):
        data = import_data()
        dataset = split_data(data)
        # Display the movie list
        for i in range(length):
            print_dish(name=dataset['names'][i],
                      dish_id=i,
                      color=yellow)
        return length
    return 0

def search_dish(dish_id: int) -> str:
    """Prints the single dish name with given id.
    Args:
      dish_id (int): Dish id.
    Returns:
      Dish name or None.
    """

    # Check whether the dish id is valid
    if validate_dish_id(dish_id):
        # Search dish title by id
        data = import_data()
        dataset = split_data(data)
        # Display the search result
        print_dish(name=dataset['names'][dish_id],
                   dish_id=dish_id,
                   color=yellow)
        return dataset['names'][dish_id]
    return None

def split_data(dish_dataset: Dict) -> Dict:
    """Splits dataset.
    Args:
      dish_dataset (Dict): Dataset.
    Returns:
      names (dictionary): Dish names in a dictionary.
    """

    # Get dish names
    dish_names = dish_dataset[['name']].values.flatten().tolist()
    # Pack and return the split data
    return {'names': dish_names}

def validate_dish_id(dish_id: int) -> bool:
    """Validates dish id.
    Args:
      dish_id (int): Dish id.
    Returns:
      is_valid (bool): True / False.
    """

    # Check whether the id is valid
    if not (isinstance(dish_id, int) and dish_id >= 0 and dish_id <= 424508):
        print_red('Input is not a valid integer between [0, 424508]')
        return False
    return True

def validate_length(length: int) -> bool:
    """Validates given length.
    Args:
      length (int): Length to display dishes.
    Returns:
      is_valid (bool): True / False.
    """

    # Check whether the list length is valid
    if (isinstance(length, int) and length >= 1 and length <= 424508):
        return True
    print_red('Invalid value for list length: "' + str(length) + '"')
    print_red('Input is not a valid integer between [1, 424508]')
    return False

def validate_max_recommendation(recommendation_count: int) -> bool:
    """Validates dish id.
    Args:
      recommendation_count (int): Max recommendation count given by user.
    Returns:
      is_valid (bool): True / False.
    """

    # Check whether the recommendations count is valid
    if not (isinstance(recommendation_count, int)
            and recommendation_count >= 1 and recommendation_count <= 30):
        print_red('Invalid value for recommendations number: "' +
                 str(recommendation_count) + '"')
        print_red('Input is not a valid integer between [1, 30]')
        return False
    return True