def get_countries():
print('Getting available countries...')
df = github.get()
df = df[df.confirmed > 0]
countries = sorted(list(set(df.country.values)))
for a,b,c in zip(countries[::3],countries[1::3],countries[2::3]):
print('{:<30}{:<30}{:<}'.format(a,b,c))
def get_countries():
print('Getting available countries...')
df = github.get()
df = df[df.confirmed > 0]
countries = sorted(list(set(df.country.values)))
for a,b,c in zip(countries[::3],countries[1::3],countries[2::3]):
print('{:<30}{:<30}{:<}'.format(a,b,c))
print('\n\033[1;31mNUMBER OF COUNTRIES/AREAS INFECTED:\033[0;0m', len(countries))
args = docopt.docopt(__doc__)
out = args['--output_folder']
country = args['--country']
source = args['--source']
############ DATA SELECTION ############
if '_' in country:
country = replace_arg_score(country)
if country == 'Global':
country = None
if source == 'JHU':
df = github.get()
elif source == 'wiki':
print('Apologies, the wikipedia source is not ready yet - getting github data')
df = github.get()
############ COUNTRY SELECTION ############
def get_similar_countries(c, country_list):
pos_countries = get_close_matches(c, country_list)
if len(pos_countries) > 0:
print('\033[1;31m'+c, 'was not listed. did you mean', pos_countries[0].capitalize() + '?\033[0;0m')
