From ce226f73524c2723d3b0fe122c25448fd2531e45 Mon Sep 17 00:00:00 2001
From: Hazem <79111320+7azeemm@users.noreply.github.com>
Date: Sat, 30 Nov 2024 16:08:19 +0100
Subject: Museum: fix hypixel api data (#1464)

* museum changes

* ..
---
 .github/scripts/updateMuseum.py | 56 +++++++++++++++++++++++++++++++++++++++--
 1 file changed, 54 insertions(+), 2 deletions(-)

(limited to '.github/scripts')

diff --git a/.github/scripts/updateMuseum.py b/.github/scripts/updateMuseum.py
index 0e3f7806..d6a48f89 100644
--- a/.github/scripts/updateMuseum.py
+++ b/.github/scripts/updateMuseum.py
@@ -42,6 +42,9 @@ def processMuseumData(internalName, data):
         for armorSet in donationXpInfo:
             itemToXp[armorSet] = donationXpInfo[armorSet]
             armor.add(armorSet)
+            if armorSet in setOverride:
+                addPieceToSet(setOverride[armorSet], armorSet)
+                return
             addPieceToSet(internalName, armorSet)
     else:
         donationXp = data.get('donation_xp', 0)
@@ -56,8 +59,11 @@ def processMuseumData(internalName, data):
 def addPieceToSet(piece, setName):
     if setName not in armorSets:
         armorSets[setName] = set()
-    armorSets[setName].add(piece)
-
+    if isinstance(piece, list):
+        for p in piece:
+            armorSets[setName].add(p)
+    else:
+        armorSets[setName].add(piece)
 
 priorityExceptions = {
     "PERFECT_TIER_12": "PERFECT_HELMET_12",
@@ -75,6 +81,52 @@ setPriorityList = [
     "CLOAK",
 ]
 
+setOverride = {
+    "BLAZE": [
+      "BLAZE_BOOTS",
+      "BLAZE_CHESTPLATE",
+      "BLAZE_HELMET",
+      "BLAZE_LEGGINGS"
+    ],
+    "CRIMSON_HUNTER": [
+      "BLAZE_BELT",
+      "GHAST_CLOAK",
+      "GLOWSTONE_GAUNTLET",
+      "MAGMA_NECKLACE"
+    ],
+    "END": [
+      "ENDER_BELT",
+      "ENDER_CLOAK",
+      "ENDER_GAUNTLET",
+      "ENDER_NECKLACE"
+      "END_BOOTS",
+      "END_CHESTPLATE",
+      "END_HELMET",
+      "END_LEGGINGS",
+    ],
+    "MONSTER_RAIDER": [
+      "CREEPER_LEGGINGS",
+      "GUARDIAN_CHESTPLATE",
+      "SKELETON_HELMET",
+      "TARANTULA_BOOTS"
+    ],
+    "SNOW_SUIT": [
+      "SNOW_SUIT_BOOTS",
+      "SNOW_SUIT_CHESTPLATE",
+      "SNOW_SUIT_HELMET",
+      "SNOW_SUIT_LEGGINGS",
+      "SNOW_BELT",
+      "SNOW_CLOAK",
+      "SNOW_GLOVES",
+      "SNOW_NECKLACE"
+    ],
+    "SPONGE": [
+      "SPONGE_BOOTS",
+      "SPONGE_CHESTPLATE",
+      "SPONGE_HELMET",
+      "SPONGE_LEGGINGS"
+    ],
+}
 
 def findAppropriateId(setName):
     if setName in priorityExceptions:
-- 
cgit