diff --git a/.github/workflows/release.yml b/.github/workflows/release.yml index 263499a..7890bec 100644 --- a/.github/workflows/release.yml +++ b/.github/workflows/release.yml @@ -57,6 +57,7 @@ jobs: run: | protoc --python_out=. esf.proto python convert.py sde/fsd + python list_shiptypes.py sde/fsd - name: Fetch icons run: | diff --git a/list_shiptypes.py b/list_shiptypes.py new file mode 100644 index 0000000..d7a7fc4 --- /dev/null +++ b/list_shiptypes.py @@ -0,0 +1,35 @@ +import json +import os +import sys +import yaml + +if len(sys.argv) < 2: + print("Usage: python3 convert.py ") + exit(1) + +path = sys.argv[1] + +os.makedirs("dist", exist_ok=True) + +with open(f"{path}/groupIDs.yaml") as fp: + groupIDs = yaml.load(fp, Loader=yaml.CSafeLoader) + +with open(f"{path}/typeIDs.yaml") as fp: + typeIDs = yaml.load(fp, Loader=yaml.CSafeLoader) + +ships = [] + +for id, entry in typeIDs.items(): + group = groupIDs[entry["groupID"]] + + if group["categoryID"] == 6 and entry["published"]: + ships.append( + { + "id": id, + "name": entry["name"]["en"], + "group": group["name"]["en"], + } + ) + +with open("dist/shiptypes.json", "w") as fp: + json.dump(ships, fp)