def main(inpath, outpath, gadmpath, hansen_path): d = merge_dicts(gadmpath, hansenpath) subnat = UnicodeDictReader(open(inpath, 'r')) fields = get_fieldnames(inpath) outcsv = UnicodeDictWriter(open(outpath, 'w'), fieldnames=fields) outcsv.writeheader() process_csv(subnat, outcsv, d) print 'Done' return
def main(natpath, isopath, outpath): nat = UnicodeDictReader(open(natpath, 'r')) iso_dict = gen_iso_dict(isopath) fields = get_fieldnames(natpath) nat_final = UnicodeDictWriter(open(outpath, 'w'), fieldnames=fields) nat_final.writeheader() for row in nat: new_row = copy.copy(row) new_row = postprocess.cleanup_names(new_row) try: new_row['iso'] = iso_dict[new_row['country']] except KeyError: new_row['iso'] = 'Null' print new_row['country'] nat_final.writerow(new_row) print 'Done'