format scripts

This commit is contained in:
TimHasert 2023-08-20 22:55:22 +02:00
parent 3c141f870c
commit 536b45db0d
2 changed files with 10 additions and 4 deletions

View File

@ -1,5 +1,6 @@
import json import json
def reorder_resistances(raw: list) -> list: def reorder_resistances(raw: list) -> list:
return [ return [
raw[0], raw[0],
@ -9,6 +10,7 @@ def reorder_resistances(raw: list) -> list:
raw[4], raw[4],
] ]
def map_skills_map(raw: dict, skill_ids: dict) -> dict: def map_skills_map(raw: dict, skill_ids: dict) -> dict:
skills = {} skills = {}
for s in raw: for s in raw:
@ -17,11 +19,12 @@ def map_skills_map(raw: dict, skill_ids: dict) -> dict:
try: try:
v = int(s["q"]) v = int(s["q"])
except: except:
v = 0 # torso up v = 0 # torso up
skills[id] = v skills[id] = v
return skills return skills
def map_deco(raw: dict, skill_ids: dict) -> dict: def map_deco(raw: dict, skill_ids: dict) -> dict:
return { return {
"name": raw["name"], "name": raw["name"],
@ -54,7 +57,8 @@ if __name__ == "__main__":
for piece in armor_category["armor"]: for piece in armor_category["armor"]:
# get attributes # get attributes
name = piece["name"] name = piece["name"]
if name == "None": continue if name == "None":
continue
defe = piece["defense"] defe = piece["defense"]
# get type # get type
@ -78,8 +82,8 @@ if __name__ == "__main__":
"skills": map_skills_map(skills, skill_ids), "skills": map_skills_map(skills, skill_ids),
"slots": piece["slots"][-1], "slots": piece["slots"][-1],
"defense": { "defense": {
"base": defe[0], "base": defe[0],
"max": defe[-1], "max": defe[-1],
}, },
"type": type, "type": type,
"resistance": reorder_resistances(piece["resistances"]), "resistance": reorder_resistances(piece["resistances"]),

View File

@ -2,9 +2,11 @@ import json
CATEGORIES = [] CATEGORIES = []
def clean_name(s: str) -> str: def clean_name(s: str) -> str:
return s.split("<")[0] return s.split("<")[0]
if __name__ == "__main__": if __name__ == "__main__":
# read input # read input
with open("./raw-skills.json") as f: with open("./raw-skills.json") as f: