diff --git a/app/services/location/nyt.py b/app/services/location/nyt.py index 1f25ec34..94b92bcf 100644 --- a/app/services/location/nyt.py +++ b/app/services/location/nyt.py @@ -12,7 +12,7 @@ from ...models import Timeline from ...utils import httputils from . import LocationService - +from ...utils.baseurls import BaseUrl LOGGER = logging.getLogger("services.location.nyt") @@ -36,7 +36,7 @@ async def get(self, loc_id): # pylint: disable=arguments-differ # Base URL for fetching category. -BASE_URL = "https://raw.githubusercontent.com/nytimes/covid-19-data/master/us-counties.csv" +BASE_URL = BaseUrl.NYT def get_grouped_locations_dict(data): diff --git a/app/utils/baseurls.py b/app/utils/baseurls.py new file mode 100644 index 00000000..ccebf743 --- /dev/null +++ b/app/utils/baseurls.py @@ -0,0 +1,10 @@ +import enum + +class BaseUrl(str, enum.Enum): + """ + A base url available for retrieving data. + """ + + JHU = "https://raw.githubusercontent.com/CSSEGISandData/2019-nCoV/master/csse_covid_19_data/csse_covid_19_time_series/" + CSBS = "https://facts.csbs.org/covid-19/covid19_county.csv" + NYT = "https://raw.githubusercontent.com/nytimes/covid-19-data/master/us-counties.csv"