feat: Avatar Items

pages/Avatar_Items.py

@@ -1,11 +1,12 @@
 import streamlit as st
 from utils.wolvesville import Wolvesville
-from utils.models import AvatarItems
+from utils.models import AvatarItems, AvatarItemSets
 from typing import List
 
 api = Wolvesville()
 
 items: List[AvatarItems] = api.getItems()
+itemSets : List[AvatarItemSets] = api.getItemAsSets()
 
 st.title("Avatar Items")
 
@@ -22,3 +23,20 @@ for item in items[:5]:
     st.button(f"{item.costInGold} 🪙", key=item.id)
 
     st.divider()
+
+st.title("Item Collections")
+for item in itemSets[:5]:
+    st.markdown(
+        f'<img src="{item.promoImageUrl}" style="height: 75%; width:75%;"/>',
+        unsafe_allow_html=True,
+    )
+
+    if item.ids != []:
+        with st.expander("See item sets"):
+            for id in item.ids:
+                st.markdown(
+                    f'<img src="{id}" style="height: 25%; width:25%;"/>',
+                    unsafe_allow_html=True,
+                )
+
+    st.divider()

pages/Emojis.py

@@ -18,7 +18,7 @@ for emoji in emojis[:5]:
         unsafe_allow_html=True,
     )
     st.markdown(f"Event: {emoji.event.title() if emoji.event else 'None'}")
-    
+
 st.header("Emoji Collections")
 
 for item in collections[:5]:
@@ -40,7 +40,7 @@ for item in collections[:5]:
                     f'<img src="{emoji.urlPreview}" style="height: 100px; width:100px;"/>',
                     unsafe_allow_html=True,
                 )
-                eventName = emoji.event.title() if emoji.event else 'None'
+                eventName = emoji.event.title() if emoji.event else "None"
                 eventName = " ".join(eventName.split("_"))
                 st.markdown(f"Event: {eventName}")
     st.divider()

utils/models.py

@@ -68,6 +68,7 @@ def _getEmojiObjects(emojis: List[Dict[str, Union[str, Dict]]]):
             print(emoji)
     return emojiList
 
+
 def _getEmojiCollectionObjects(emojis: List[Dict[str, Union[str, Dict]]]):
     emojCollectioniList = []
     for emoji in emojis:
@@ -79,6 +80,7 @@ def _getEmojiCollectionObjects(emojis: List[Dict[str, Union[str, Dict]]]):
             print(emoji)
     return emojCollectioniList
 
+
 def _mapEmojiCollections(emojis, collections):
     for item in collections:
         emojiIds = item.emojiIds
@@ -86,6 +88,7 @@ def _mapEmojiCollections(emojis, collections):
         item.emojis = emojis
     return collections
 
+
 def _getRoleIconObjects(icons: List[Dict[str, Union[str, Dict]]]):
     roleIconList = []
     for icon in icons:
@@ -134,6 +137,14 @@ class AvatarItems(BaseModel):
     costInGold: int = 0
 
 
+class AvatarItemSets(BaseModel):
+    id: str
+    promoImageUrl: str
+    avatarItemIds: List[str]
+    promoImagePrimaryColor: str = None
+    ids: List[AvatarItems] = []
+
+
 def _getAvatarObjects(items: List[Dict[str, Union[str, Dict]]]):
     itemList = []
     for item in items:
@@ -146,6 +157,24 @@ def _getAvatarObjects(items: List[Dict[str, Union[str, Dict]]]):
     return itemList
 
 
+def _getAvatarItemSetObjects(items: List[Dict[str, Union[str, Dict]]]):
+    itemSetList = []
+    for item in items:
+        try:
+            entry = AvatarItemSets(**item)
+            itemSetList.append(entry)
+        except Exception as e:
+            print(e)
+            print(item)
+    return itemSetList
+
+def _mapItemItemSets(sets, items):
+    for setItem in sets:
+        avatarIds = setItem.avatarItemIds
+        ids = [item.imageUrl for item in items if item.id in avatarIds]
+        setItem.ids = ids
+    return sets
+
 class ProfileIcons(BaseModel):
     id: str
     name: str

utils/wolvesville.py

@@ -7,7 +7,9 @@ from utils.models import (
     _getAvatarObjects,
     _getProfileIconObjects,
     _getEmojiCollectionObjects,
-    _mapEmojiCollections
+    _mapEmojiCollections,
+    _getAvatarItemSetObjects,
+    _mapItemItemSets
 )
 from dotenv import load_dotenv
 from os import getenv
@@ -46,9 +48,11 @@ class Wolvesville:
         emojis = requests.get(f"{self.url}/items/emojis", headers=self.headers).json()
         resp = _getEmojiObjects(emojis=emojis)
         return resp
-    
+
     def getEmojiCollections(self):
-        emojis = requests.get(f"{self.url}/items/emojiCollections", headers=self.headers).json()
+        emojis = requests.get(
+            f"{self.url}/items/emojiCollections", headers=self.headers
+        ).json()
         resp = _getEmojiCollectionObjects(emojis=emojis)
         return resp
 
@@ -58,6 +62,7 @@ class Wolvesville:
         mapping = _mapEmojiCollections(emojis=emojis, collections=collections)
         return mapping
     
+
     def getScreens(self):
         screens = requests.get(
             f"{self.url}/items/loadingScreens", headers=self.headers
@@ -66,12 +71,25 @@ class Wolvesville:
         return resp
 
     def getItems(self):
-        screens = requests.get(
+        items = requests.get(
             f"{self.url}/items/avatarItems", headers=self.headers
         ).json()
-        resp = _getAvatarObjects(items=screens)
+        resp = _getAvatarObjects(items=items)
         return resp
 
+    def getItemSets(self):
+        itemSets = requests.get(
+            f"{self.url}/items/avatarItemSets", headers=self.headers
+        ).json()
+        resp = _getAvatarItemSetObjects(items=itemSets)
+        return resp
+    
+    def getItemAsSets(self):
+        items = self.getItems()
+        sets = self.getItemSets()
+        mapping = _mapItemItemSets(sets=sets, items=items)
+        return mapping
+
     def getProfileIcons(self):
         icons = requests.get(
             f"{self.url}/items/profileIcons", headers=self.headers