Source code of COVID-19: The Italian case on Medium

data.py

import pandas as pd

_provinces = pd.read_csv('./data/dati-province/dpc-covid19-ita-province.csv')
_regions = pd.read_csv('./data/dati-regioni/dpc-covid19-ita-regioni.csv')


def region(name: str = False):
    if(name):
        return _regions[_regions.denominazione_regione == name]
    else:
        return _regions


def all_regions_except(region: str):
    return _regions[_regions.denominazione_regione != region]


def province(name: str):
    prov = _provinces[_provinces.denominazione_provincia == name]
    prov.data = pd.to_datetime(prov.data)
    return prov


def provinces(region: str = False):
    if(region):
        provs = _provinces[_provinces.denominazione_regione == region]
        provs.data = pd.to_datetime(provs.data)
        return provs
    else:
        provs = _provinces
        provs.data = pd.to_datetime(provs.data)
        return provs


def all_provinces_except(region: str, name: str):
    return _provinces[_provinces.denominazione_regione == region][_provinces.denominazione_provincia != name]