models/continent.py

import pandas as pd

def read_country_data(path_world):
    """Reads CSV data into a Pandas DataFrame"""
    return pd.read_csv(path_world)

def get_continents(data_frame):
    continents = data_frame["Continent"].unique()
    return continents

def get_continent_stats(data_frame, attribute, stat_type):
    continents_stats = data_frame.groupby("Continent")[attribute].agg(total_attribute = 'sum')
    if stat_type == "highest":
        continent = continents_stats.idxmax().item()
        value = continents_stats.max().item()
        return {continent: value}
    elif stat_type == "lowest":
        continent = continents_stats.idxmin().item()
        value = continents_stats.min().item()
        return {continent: value}