def main(skip_download=False): if not skip_download: print("\nAttempting to download latest CSV files...") download_csv() df_merged = _load_merged() if check_data_correctness(df_merged): print("Data correctness check %s.\n" % colored("passed", "green")) else: print_err("Data correctness check %s.\n" % colored("failed", "red")) sys.exit(1) if export(df_merged): print("Successfully exported CSVs to %s\n" % colored(os.path.abspath(OUTPUT_PATH), "magenta")) else: print_err("JHU export failed.\n") sys.exit(1) print("Generating megafile…") megafile.generate_megafile() print("Megafile is ready.") send_success(channel="corona-data-updates", title="Updated JHU GitHub exports")
def main(filename=None, skip_download=False): import inquirer from glob import glob if not skip_download: print("\nAttempting to download latest report...") download_xlsx() download_csv() if filename is None: print( """\n[Note] If you don't see the latest report in the options below, please download the Excel file from: https://www.ecdc.europa.eu/en/publications-data/download-todays-data-geographic-distribution-covid-19-cases-worldwide Then move it to the folder %s\n""" % os.path.abspath(RELEASES_PATH)) filenames = glob(os.path.join(RELEASES_PATH, '*.xlsx')) filenames.extend(glob(os.path.join(RELEASES_PATH, '*.xls'))) filenames.extend(glob(os.path.join(RELEASES_PATH, '*.csv'))) filenames = list( filter( lambda name: not name.startswith("~"), map(os.path.basename, sorted(filenames, reverse=True)) ) ) answers = inquirer.prompt([ inquirer.List('filename', message='Which release to use?', choices=filenames, default=0) ]) filename = answers['filename'] if check_data_correctness(filename): print("Data correctness check %s.\n" % colored("passed", 'green')) else: print_err("Data correctness check %s.\n" % colored("failed", 'red')) sys.exit(1) if export(filename): print("Successfully exported CSVs to %s\n" % colored(os.path.abspath(OUTPUT_PATH), 'magenta')) else: print_err("ECDC Export failed.\n") sys.exit(1) print("Generating megafile…") megafile.generate_megafile() print("Megafile is ready.") send_success( channel='corona-data-updates', title='Updated ECDC GitHub exports' )
filenames.extend(glob(os.path.join(RELEASES_PATH, '*.csv'))) filenames = list( filter(lambda name: not name.startswith("~"), map(os.path.basename, sorted(filenames, reverse=True)))) answers = inquirer.prompt([ inquirer.List('filename', message='Which release to use?', choices=filenames, default=0) ]) filename = answers['filename'] if check_data_correctness(filename): print("Data correctness check %s.\n" % colored("passed", 'green')) else: print("Data correctness check %s.\n" % colored("failed", 'red')) sys.exit(1) if export(filename): print("Successfully exported CSVs to %s\n" % colored(os.path.abspath(OUTPUT_PATH), 'magenta')) else: print("ECDC Export failed.\n") sys.exit(1) print("Generating megafile…") megafile.generate_megafile() print("Megafile is ready.")