chore: formatting

pages/Offers.py

@@ -26,7 +26,10 @@ for roles in chunks(offerList, CHUNK_SIZE):
         advancedRoleId = role.advancedRoleId
         advancedRoleId = " ".join(advancedRoleId.split("-"))
         col.subheader(advancedRoleId)
-        col.button(f"Ability exchange vouchers: {role.abilityExchangeVoucherCount}", key=uuid4().hex)
+        col.button(
+            f"Ability exchange vouchers: {role.abilityExchangeVoucherCount}",
+            key=uuid4().hex,
+        )
         col.button(f"Talismans: {role.talismanCount}", key=uuid4().hex)
         col.button(f"Loyalty Tokens: {role.loyaltyTokenCount}", key=uuid4().hex)
 

pages/Roles.py

@@ -3,17 +3,32 @@ from utils.wolvesville import Wolvesville
 from utils.models import Role, AdvancedRoleCardOffers
 from utils.helper import chunks, getTitle
 from typing import List
+
 st.set_page_config(layout="wide")
 
 
 api = Wolvesville()
 
 roles: List[Role] = api.getRoleRoleIcons()
+
 offers: List[AdvancedRoleCardOffers] = api.getRoleCardOffers()
 
 st.markdown(getTitle("Roles"), unsafe_allow_html=True)
 st.divider()
 
+search_bar = st.text_input(
+    "Search roles",
+    "",
+    key="search_roles",
+)
+
+name: str = st.session_state.search_roles
+if name == "":
+    roles = api.getRoleRoleIcons()
+else:
+    roles = [role for role in roles if name.lower() in role.name.lower()]
+
+
 CHUNK_SIZE = 4
 
 for roles in chunks(roles, CHUNK_SIZE):
@@ -41,4 +56,4 @@ for roles in chunks(roles, CHUNK_SIZE):
 
     st.divider()
 
-st.toast("Loaded roles", icon="🕵️")
+st.toast("Loaded roles", icon="🕵️")

utils/helper.py

@@ -11,6 +11,7 @@ def chunks(lst: List, n: int):
 def getTitle(text: str):
     return f"<h1 style='text-align: center;'>{text}</h1>"
 
+
 def groupItemCategories(items: Tuple[AvatarItems]):
     groups = {}
     for item in items:
@@ -19,4 +20,4 @@ def groupItemCategories(items: Tuple[AvatarItems]):
         else:
             groups[item.type].append(item)
 
-    return groups
+    return groups

utils/models.py

@@ -202,9 +202,11 @@ def _getProfileIconObjects(profile_icons: List[Dict[str, Union[str, Dict]]]):
             print(icon)
     return profileIconList
 
+
 def getUrl(text: str):
     return text.split("'")[1]
 
+
 class Backgrounds(BaseModel):
     id: str
     rarity: str
@@ -237,6 +239,7 @@ class AdvancedRoleCardOffers(BaseModel):
     loyaltyTokenCount: int
     promoImageUrl: str
 
+
 def _getRoleCardOfferObjects(offers: List[Dict[str, Union[str, Dict]]]):
     roleCardOfferList = []
     for icon in offers:
@@ -246,4 +249,4 @@ def _getRoleCardOfferObjects(offers: List[Dict[str, Union[str, Dict]]]):
         except Exception as e:
             print(e)
             print(icon)
-    return roleCardOfferList
+    return roleCardOfferList

utils/wolvesville.py

@@ -11,7 +11,7 @@ from utils.models import (
     _getAvatarItemSetObjects,
     _mapItemItemSets,
     _getBackgroundObjects,
-    _getRoleCardOfferObjects
+    _getRoleCardOfferObjects,
 )
 from dotenv import load_dotenv
 from os import getenv
@@ -82,7 +82,7 @@ class Wolvesville:
         ).json()
         resp = _getScreenObjects(screens=screens)
         return resp
-    
+
     @lru_cache(maxsize=None)
     def getBackgrounds(self):
         backgrounds = requests.get(
@@ -128,4 +128,4 @@ class Wolvesville:
             f"{self.url}/items/advancedRoleCardOffers", headers=self.headers
         ).json()
         resp = _getRoleCardOfferObjects(offers=icons)
-        return resp
+        return resp