forked from omicron/d2warehouse
60 lines
1.6 KiB
Python
60 lines
1.6 KiB
Python
import csv
|
|
import json
|
|
import os
|
|
import sys
|
|
|
|
path = sys.argv[1] if len(sys.argv) >= 2 else "."
|
|
|
|
category = "Base"
|
|
setitems = {}
|
|
with open(os.path.join(path, "setitems.txt")) as f:
|
|
dr = csv.DictReader(f, delimiter="\t")
|
|
for row in dr:
|
|
if row["index"] == "Expansion":
|
|
category = row["index"]
|
|
continue
|
|
setitems[row["*ID"]] = {
|
|
"name": row["index"],
|
|
"set": row["set"],
|
|
"itembase": row["item"],
|
|
"req_lvl": int(row["lvl req"]),
|
|
"ilvl": int(row["lvl"]),
|
|
"rarity": int(row["rarity"]),
|
|
"category": category,
|
|
}
|
|
|
|
category = "Base"
|
|
uniqueitems = {}
|
|
with open(os.path.join(path, "uniqueitems.txt")) as f:
|
|
dr = csv.DictReader(f, delimiter="\t")
|
|
for row in dr:
|
|
if row["index"] in [
|
|
"Expansion",
|
|
"Armor",
|
|
"Elite Uniques",
|
|
"Rings",
|
|
"Class Specific",
|
|
]:
|
|
category = row["index"]
|
|
continue
|
|
if row["*ID"] == "288":
|
|
category = "Patch 1.10+"
|
|
if len(row["lvl req"]) == 0:
|
|
continue # deleted uniques
|
|
uniqueitems[row["*ID"]] = {
|
|
"name": row["index"],
|
|
"itembase": row["code"],
|
|
"req_lvl": int(row["lvl req"]),
|
|
"ilvl": int(row["lvl"]),
|
|
"rarity": int(row["rarity"]),
|
|
"category": category,
|
|
}
|
|
|
|
with open("uniques.json", "w", newline="\n") as f:
|
|
json.dump(uniqueitems, f, indent=4)
|
|
f.write("\n")
|
|
|
|
with open("sets.json", "w", newline="\n") as f:
|
|
json.dump(setitems, f, indent=4)
|
|
f.write("\n")
|