1# Quick scipt to apply categorized items to the 2# base commitlist . Useful if you are refactoring any code 3# but want to keep the previous data on categories 4 5import csv 6 7import commitlist 8 9 10category_csv = "results/category_data.csv" 11commitlist_csv = "results/commitlist.csv" 12 13with open(category_csv) as category_data: 14 reader = csv.DictReader(category_data, commitlist.commit_fields) 15 rows = list(reader) 16 category_map = {row["commit_hash"]: row["category"] for row in rows} 17 18with open(commitlist_csv) as commitlist_data: 19 reader = csv.DictReader(commitlist_data, commitlist.commit_fields) 20 commitlist_rows = list(reader) 21 22for row in commitlist_rows: 23 hash = row["commit_hash"] 24 if hash in category_map and category_map[hash] != "Uncategorized": 25 row["category"] = category_map[hash] 26 27with open(commitlist_csv, "w") as commitlist_write: 28 writer = csv.DictWriter(commitlist_write, commitlist.commit_fields) 29 writer.writeheader() 30 writer.writerows(commitlist_rows) 31