import csv
import json
import os
import sys

path = sys.argv[1] if len(sys.argv) >= 2 else "."

category = "Base"
affixes = {"prefixes": {}, "suffixes": {}}
with open(os.path.join(path, "magicprefix.txt")) as f:
    dr = csv.DictReader(f, delimiter="\t")
    index = 0
    for row in dr:
        index += 1
        if len(row["Name"]) == 0:
            continue
        affixes["prefixes"][index] = {
            "name": row["Name"],
            "req_lvl": row["levelreq"],
            "req_class": None if len(row["class"]) == 0 else row["class"],
        }

with open(os.path.join(path, "magicsuffix.txt")) as f:
    dr = csv.DictReader(f, delimiter="\t")
    index = 0
    for row in dr:
        index += 1
        if len(row["Name"]) == 0:
            continue
        affixes["suffixes"][index] = {
            "name": row["Name"],
            "req_lvl": row["levelreq"],
            "req_class": None if len(row["class"]) == 0 else row["class"],
        }


with open("affixes.json", "w", newline="\n") as f:
    json.dump(affixes, f, indent=4)
    f.write("\n")