feat: Support backgrounds

pages/Backgrounds.py

@@ -0,0 +1,44 @@
+import streamlit as st
+from utils.wolvesville import Wolvesville
+from utils.models import Backgrounds
+from utils.helper import getTitle
+from typing import List
+
+api = Wolvesville()
+
+screens: List[Backgrounds] = api.getBackgrounds()
+
+st.markdown(getTitle("Backgrounds"), unsafe_allow_html=True)
+
+
+for screen in screens[:5]:
+    rarity = screen.rarity.title()
+    st.header(rarity)
+    dayCols = st.columns(3)
+    dayCols[0].markdown(
+        f'<img src="{screen.imageDay.url}" style="height:75%; width:75%;"/>',
+        unsafe_allow_html=True,
+    )
+    dayCols[1].markdown(
+        f'<img src="{screen.imageDaySmall.url}" style="height:75%; width:75%;"/>',
+        unsafe_allow_html=True,
+    )
+    dayCols[2].markdown(
+        f'<img src="{screen.imageDayWide.url}" style="height:75%; width:75%;"/>',
+        unsafe_allow_html=True,
+    )
+
+    nightCols = st.columns(3)
+    nightCols[0].markdown(
+        f'<img src="{screen.imageNight.url}" style="height:75%; width:75%;"/>',
+        unsafe_allow_html=True,
+    )
+    nightCols[1].markdown(
+        f'<img src="{screen.imageNightSmall.url}" style="height:75%; width:75%;"/>',
+        unsafe_allow_html=True,
+    )
+    nightCols[2].markdown(
+        f'<img src="{screen.imageNightWide.url}" style="height:75%; width:75%;"/>',
+        unsafe_allow_html=True,
+    )
+    st.divider()

utils/models.py

@@ -201,3 +201,28 @@ def _getProfileIconObjects(profile_icons: List[Dict[str, Union[str, Dict]]]):
             print(e)
             print(icon)
     return profileIconList
+
+def getUrl(text: str):
+    return text.split("'")[1]
+
+class Backgrounds(BaseModel):
+    id: str
+    rarity: str
+    imageNight: Image
+    imageNightWide: Image
+    imageNightSmall: Image
+    imageDay: Image
+    imageDayWide: Image
+    imageDaySmall: Image
+
+
+def _getBackgroundObjects(backgrounds: List[Dict[str, Union[str, Dict]]]):
+    backgroundList = []
+    for icon in backgrounds:
+        try:
+            entry = Backgrounds(**icon)
+            backgroundList.append(entry)
+        except Exception as e:
+            print(e)
+            print(icon)
+    return backgroundList

utils/wolvesville.py

@@ -10,6 +10,7 @@ from utils.models import (
     _mapEmojiCollections,
     _getAvatarItemSetObjects,
     _mapItemItemSets,
+    _getBackgroundObjects
 )
 from dotenv import load_dotenv
 from os import getenv
@@ -80,6 +81,14 @@ class Wolvesville:
         ).json()
         resp = _getScreenObjects(screens=screens)
         return resp
+    
+    @lru_cache(maxsize=None)
+    def getBackgrounds(self):
+        backgrounds = requests.get(
+            f"{self.url}/items/backgrounds", headers=self.headers
+        ).json()
+        resp = _getBackgroundObjects(backgrounds=backgrounds)
+        return resp
 
     @lru_cache(maxsize=None)
     def getItems(self):