feat: Map role icons to roles

utils/models.py

@@ -15,7 +15,13 @@ class Role(BaseModel):
     name: str
     description: str
     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 = []
@@ -48,3 +54,21 @@ def _getEmojiObjects(emojis: List[Dict[str, Union[str, Dict]]]):
             print(e)
             print(emoji)
     return emojiList
+
+def _getRoleIconObjects(icons: List[Dict[str, Union[str, Dict]]]):
+    roleIconList = []
+    for icon in icons:
+        try:
+            entry = RoleIcon(**icon)
+            roleIconList.append(entry)
+        except Exception as e:
+            print(e)
+            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

utils/wolvesville.py

@@ -1,4 +1,4 @@
-from utils.models import _getRoleObjects, _getEmojiObjects
+from utils.models import _getRoleObjects, _getEmojiObjects, _getRoleIconObjects, _mapRoleRoleIcons
 from dotenv import load_dotenv
 from os import getenv
 import requests
@@ -20,7 +20,18 @@ 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)
+        return resp
 
+    def getRoleRoleIcons(self):
+        roles = self.getRoles()
+        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)