def main():
fm = FileManager()
fe = Fetcher()
folder_dir = fm.get_folder_dir()
df = [fe.statistics(), fe.coutries_data(), fe.countries_historical_data()]
filenames = [
'statistics', 'all_country_data', 'all_country_historical_data'
]
for i in range(3):
filename = '{}/{}.csv'.format(folder_dir, filenames[i])
df[i].to_csv(filename, index=False)
print('{} is created successfully.'.format(filename))
if len(argv) > 1:
country = argv[1]
df = [fe.coutry_data(country), fe.country_historical_data(country)]
filenames = [
'{}__country_data'.format(country),
'{}__country_historical_data'.format(country)
]
for i in range(2):
filename = '{}/{}.csv'.format(folder_dir, filenames[i])
df[i].to_csv(filename, index=False)
print('{} is created successfully.'.format(filename))
