Reshape the JHU data files

reshape_jhu_data.py

import csv
import sys
import traceback

FIELD_NAMES = ['Province_State', 'Country_Region',
               'Lat', 'Long', 'Date', 'Count']


if __name__ == '__main__':
    count = 0
    try:
        with open(sys.argv[1], 'r') as input_file:
            reader = csv.DictReader(input_file)
            with open(sys.argv[2], 'w') as output_file:
                writer = csv.DictWriter(
                    output_file, FIELD_NAMES, extrasaction='ignore', quoting=csv.QUOTE_MINIMAL)
                writer.writeheader()
                for row_in in reader:
                    row_out = dict()
                    row_out['Province_State'] = row_in['Province/State']
                    row_out['Country_Region'] = row_in['Country/Region']
                    row_out['Lat'] = row_in['Lat']
                    row_out['Long'] = row_in['Long']
                    for key in reader.fieldnames[4:]:
                        row_out['Date'] = key
                        row_out['Count'] = row_in[key]
                        writer.writerow(row_out)
                        count += 1
    except Exception:
        traceback.print_exc()
    if count:
        print(f'Generated {count} rows')