feat: Support Loading Screens

pages/roles.py

@@ -10,15 +10,22 @@ roles: List[Role] = api.getRoleRoleIcons()
 st.title("Roles")
 
 for role in roles[:5]:
-    st.markdown(f"<img src=\"{role.image.url}\" style=\"height: 100px; width:100px;\"/>", unsafe_allow_html=True)
-    
+    st.markdown(
+        f'<img src="{role.image.url}" style="height: 100px; width:100px;"/>',
+        unsafe_allow_html=True,
+    )
+
     st.subheader(role.name)
-    st.text(role.aura)
-    st.text(role.description)
+    aura = role.aura.title()
+    st.markdown(aura)
+    st.markdown(role.description)
 
     if role.icons != []:
         with st.expander("See role icons"):
             for icon in role.icons:
-                st.markdown(f"<img src=\"{icon}\" style=\"height: 100px; width:100px;\"/>", unsafe_allow_html=True)
+                st.markdown(
+                    f'<img src="{icon}" style="height: 100px; width:100px;"/>',
+                    unsafe_allow_html=True,
+                )
 
-    st.divider()
+    st.divider()

pages/screens.py

@@ -0,0 +1,21 @@
+import streamlit as st
+from utils.wolvesville import Wolvesville
+from utils.models import LoadingScreen
+from typing import List
+
+api = Wolvesville()
+
+screens: List[LoadingScreen] = api.getScreens()
+
+st.title("Loading Screens")
+
+for screen in screens[:5]:
+    st.markdown(
+        f'<img src="{screen.image.url}"  style="height:75%; width:75%;"/>',
+        unsafe_allow_html=True,
+    )
+
+    rarity = screen.rarity.title()
+    st.markdown(rarity)
+
+    st.divider()

utils/models.py

@@ -17,12 +17,14 @@ class Role(BaseModel):
     image: Image
     icons: List[str] = None
 
+
 class RoleIcon(BaseModel):
     id: str
     rarity: str
     image: Image
     roleId: str
 
+
 def _getRoleObjects(roles: List[Dict[str, Union[str, Dict]]]):
     roleList = []
     for role in roles:
@@ -55,6 +57,7 @@ def _getEmojiObjects(emojis: List[Dict[str, Union[str, Dict]]]):
             print(emoji)
     return emojiList
 
+
 def _getRoleIconObjects(icons: List[Dict[str, Union[str, Dict]]]):
     roleIconList = []
     for icon in icons:
@@ -66,9 +69,29 @@ def _getRoleIconObjects(icons: List[Dict[str, Union[str, Dict]]]):
             print(icon)
     return roleIconList
 
+
 def _mapRoleRoleIcons(roles, icons):
     for role in roles:
         id = role.id
         iconList = [icon.image.url for icon in icons if icon.roleId == id]
         role.icons = iconList
-    return roles
+    return roles
+
+
+class LoadingScreen(BaseModel):
+    id: str
+    rarity: str
+    image: Image
+    imageWide: Image
+    imagePrimaryColor: str
+
+def _getScreenObjects(screens: List[Dict[str, Union[str, Dict]]]):
+    screenList = []
+    for screen in screens:
+        try:
+            entry = LoadingScreen(**screen)
+            screenList.append(entry)
+        except Exception as e:
+            print(e)
+            print(screen)
+    return screenList

utils/wolvesville.py

@@ -1,4 +1,10 @@
-from utils.models import _getRoleObjects, _getEmojiObjects, _getRoleIconObjects, _mapRoleRoleIcons
+from utils.models import (
+    _getRoleObjects,
+    _getEmojiObjects,
+    _getRoleIconObjects,
+    _mapRoleRoleIcons,
+    _getScreenObjects
+)
 from dotenv import load_dotenv
 from os import getenv
 import requests
@@ -20,7 +26,7 @@ class Wolvesville:
         roles = data["roles"]
         resp = _getRoleObjects(roles=roles)
         return resp
-    
+
     def getRoleIcons(self):
         data = requests.get(f"{self.url}/items/roleIcons", headers=self.headers).json()
         resp = _getRoleIconObjects(icons=data)
@@ -31,8 +37,13 @@ class Wolvesville:
         icons = self.getRoleIcons()
         mapping = _mapRoleRoleIcons(roles=roles, icons=icons)
         return mapping
-    
+
     def getEmojis(self):
         emojis = requests.get(f"{self.url}/items/emojis", headers=self.headers).json()
         resp = _getEmojiObjects(emojis=emojis)
         return resp
+
+    def getScreens(self):
+        screens = requests.get(f"{self.url}/items/loadingScreens", headers=self.headers).json()
+        resp = _getScreenObjects(screens=screens)
+        return resp