Example of Proxy Model InfectionRateVisualization

infection_rate_visualization.py

from visualizations.models import Visualization

class InfectionRateVisualization(Visualization):

    class Meta:
        proxy = True

    def from_data(csvfile):
        department_instances = []
        infection_type = []
        
        with open csvfile as file:
          if file["department"] and file["type"]:
            department_instances = np.array(file["department"])
            infection_type = np.array(file["type"])
          else:
            return IncompleteDataException("Please include a department and a type column in your CSV.")
        
        map = HeatMap(list(zip(department_instances, infection_type)))
        return map.to_svg()