From 536b45db0d07674782aa6420d87c5c199f0d09ca Mon Sep 17 00:00:00 2001 From: TimHasert <27998143+TimH96@users.noreply.github.com> Date: Sun, 20 Aug 2023 22:55:22 +0200 Subject: [PATCH] format scripts --- scripts/get-equipment-decos.py | 12 ++++++++---- scripts/get-skills.py | 2 ++ 2 files changed, 10 insertions(+), 4 deletions(-) diff --git a/scripts/get-equipment-decos.py b/scripts/get-equipment-decos.py index 54a2784..a60a457 100644 --- a/scripts/get-equipment-decos.py +++ b/scripts/get-equipment-decos.py @@ -1,5 +1,6 @@ import json + def reorder_resistances(raw: list) -> list: return [ raw[0], @@ -9,6 +10,7 @@ def reorder_resistances(raw: list) -> list: raw[4], ] + def map_skills_map(raw: dict, skill_ids: dict) -> dict: skills = {} for s in raw: @@ -17,11 +19,12 @@ def map_skills_map(raw: dict, skill_ids: dict) -> dict: try: v = int(s["q"]) except: - v = 0 # torso up + v = 0 # torso up skills[id] = v return skills + def map_deco(raw: dict, skill_ids: dict) -> dict: return { "name": raw["name"], @@ -54,7 +57,8 @@ if __name__ == "__main__": for piece in armor_category["armor"]: # get attributes name = piece["name"] - if name == "None": continue + if name == "None": + continue defe = piece["defense"] # get type @@ -78,8 +82,8 @@ if __name__ == "__main__": "skills": map_skills_map(skills, skill_ids), "slots": piece["slots"][-1], "defense": { - "base": defe[0], - "max": defe[-1], + "base": defe[0], + "max": defe[-1], }, "type": type, "resistance": reorder_resistances(piece["resistances"]), diff --git a/scripts/get-skills.py b/scripts/get-skills.py index 9a605d3..76c1dff 100644 --- a/scripts/get-skills.py +++ b/scripts/get-skills.py @@ -2,9 +2,11 @@ import json CATEGORIES = [] + def clean_name(s: str) -> str: return s.split("<")[0] + if __name__ == "__main__": # read input with open("./raw-skills.json") as f: