diff --git a/src/covidify/covidify_data.py b/src/covidify/covidify_data.py new file mode 100644 index 0000000..4eb7251 --- /dev/null +++ b/src/covidify/covidify_data.py @@ -0,0 +1,11 @@ +from convidify.sources import github, wiki + +""" +global values +""" +class DataStoreBridge(): + def jhu_sources(self): + return github.get() + + def wiki_sources(self): + return github.get() \ No newline at end of file diff --git a/src/covidify/data_prep.py b/src/covidify/covidify_prep.py similarity index 97% rename from src/covidify/data_prep.py rename to src/covidify/covidify_prep.py index 6153d44..477884c 100644 --- a/src/covidify/data_prep.py +++ b/src/covidify/covidify_prep.py @@ -3,7 +3,6 @@ Usage: data_prep.py [options] data_prep.py -h | --help - Options: -h --help Show this message. --output_folder=OUT Output folder for the data and reports to be saved. @@ -21,7 +20,7 @@ from difflib import get_close_matches from datetime import datetime, date, time -from covidify.sources import github, wiki +from covidify.covidify_data import DataStore from covidify.config import REPO, TMP_FOLDER, TMP_GIT, DATA from covidify.utils.utils import replace_arg_score @@ -42,11 +41,13 @@ country = None if source == 'JHU': - df = github.get() + datastore = DataStore() + df = datastore.jhu_sources() elif source == 'wiki': print('Apologies, the wikipedia source is not ready yet - getting github data') - df = github.get() + datastore = DataStore() + df = datastore.wiki_sources() @@ -151,7 +152,6 @@ def get_exp_moving_average(tmp, col): Calculate the number of people that are ACTUALLY infected on a given day currently infected = sum of people date - (recovored + died) ex: 5 = 10 - (4 - 1) - ''' current_infected = pd.DataFrame([]) current_infected['currently_infected'] = (df.groupby('file_date').confirmed.sum() - (df.groupby('file_date').deaths.sum() + df.groupby('file_date').recovered.sum())) diff --git a/src/covidify/list_countries.py b/src/covidify/list_countries.py index 8637489..82c34c4 100644 --- a/src/covidify/list_countries.py +++ b/src/covidify/list_countries.py @@ -9,12 +9,13 @@ import click import covidify import numpy as np -from covidify.sources import github -from covidify.config import SCRIPT +from covidify.covidify_data import DataStore +from config import SCRIPT def get_countries(): print('Getting available countries...') - df = github.get() + datastore = DataStore() + df = datastore.jhu_sources() df = df[df.confirmed > 0] countries = sorted(list(set(df.country.values))) diff --git a/src/covidify/sources/github.py b/src/covidify/sources/github.py index 38c7a71..fc1ec9d 100644 --- a/src/covidify/sources/github.py +++ b/src/covidify/sources/github.py @@ -157,7 +157,6 @@ def get(): sys.exit(1) sheets = os.listdir(DATA) - # Clean the result to the sheet tabs we want print('Getting sheets...') cleaned_sheets = clean_sheet_names(sheets)