gen_anoints_sheet.py

import unrealsdk
import csv
from typing import Dict, Set

INVBAL_CLS_BLACKLIST = (
    "CustomizationInventoryBalanceData",
    "InventoryBalanceData_Generated",
    "VaultCardRewardBalanceData",
)
INVDATA_CLS_BLACKLIST = (
    "BPInv_Ammo_C",
    "BPInv_Eridian_Fabricator_C",
    "BPInv_LootOGramData_C",
    "BPInv_Money_C",
    "InventoryData",
)

all_anoints: Set[str] = set()
balance_map: Dict[str, Set[str]] = {}

for obj in unrealsdk.FindAll("InventoryBalanceData"):
    if obj.Class.Name in INVBAL_CLS_BLACKLIST:
        continue
    if obj.InventoryData is None or obj.InventoryData.Class.Name in INVDATA_CLS_BLACKLIST:
        continue

    local_anoints = set()
    if obj.RuntimeGenericPartList.bEnabled:
        for entry in obj.RuntimeGenericPartList.PartList:
            if entry.PartData is None:
                continue

            name = entry.PartData.Name
            all_anoints.add(name)
            local_anoints.add(name)

    obj_name = obj.Name
    if obj_name in balance_map:
        obj_name = str(obj).split(" ")[-1]
    balance_map[obj_name] = local_anoints


ordered_anoints = sorted(all_anoints)
with open("anoints.csv", "w", newline="") as file:
    writer = csv.writer(file)
    writer.writerow(["Balance"] + ordered_anoints)

    for balance in sorted(balance_map):
        row = [balance]
        for anoint in ordered_anoints:
            row.append("Y" if anoint in balance_map[balance] else "")
        writer.writerow(row)