feat: Helper functions

utils/models.py

@@ -0,0 +1,50 @@
+from pydantic import BaseModel
+from typing import Dict, List, Union
+
+
+class Image(BaseModel):
+    url: str
+    width: int
+    height: int
+
+
+class Role(BaseModel):
+    id: str
+    team: str
+    aura: str
+    name: str
+    description: str
+    image: Image
+
+
+def _getRoleObjects(roles: List[Dict[str, Union[str, Dict]]]):
+    roleList = []
+    for role in roles:
+        try:
+            entry = Role(**role)
+            roleList.append(entry)
+        except Exception as e:
+            print(e)
+            print(role)
+    return roleList
+
+
+class Emoji(BaseModel):
+    id: str
+    name: str
+    rarity: str
+    urlPreview: str
+    urlAnimation: str
+    event: str = None
+
+
+def _getEmojiObjects(emojis: List[Dict[str, Union[str, Dict]]]):
+    emojiList = []
+    for emoji in emojis:
+        try:
+            entry = Emoji(**emoji)
+            emojiList.append(entry)
+        except Exception as e:
+            print(e)
+            print(emoji)
+    return emojiList

utils/wolvesville.py

@@ -0,0 +1,24 @@
+from utils.models import _getRoleObjects, _getEmojiObjects
+from dotenv import load_dotenv
+from os import getenv
+import requests
+
+load_dotenv()
+
+
+class Wolvesville:
+    def __init__(self):
+        API_KEY = getenv("API_KEY")
+        self.headers = {"Authorization": f"Bot {API_KEY}"}
+        self.url = "https://api.wolvesville.com"
+
+    def getRoles(self):
+        data = requests.get(f"{self.url}/roles", headers=self.headers).json()
+        roles = data["roles"]
+        resp = _getRoleObjects(roles=roles)
+        return resp
+
+    def getEmojis(self):
+        emojis = requests.get(f"{self.url}/items/emojis", headers=self.headers).json()
+        resp = _getEmojiObjects(emojis=emojis)
+        return resp