FIX: Register server if it is not already registered

Author: staticaron

data/base_db_entries.py

@@ -1,5 +1,5 @@
 base_servers_entry = {
     "server_id": 0,
     "server_name": "",
-    "spotify": {"enabled": True, "style": "embed"},
+    "spotify": {"enabled": False, "style": "text"},
 }

managers/cache_manager.py

@@ -63,10 +63,15 @@ def cache_servers_data(self):
         for data_item in data:
             self.server_cache[data_item.get("server_id")] = data_item
 
-    async def get_server(self, server_id: int) -> dict:
+    async def get_server(self, server_id: int, register_if_not_found: bool = False) -> dict:
         """Fetch server from cache"""
 
-        return self.server_cache.get(server_id, None)
+        server_details = self.server_cache.get(server_id, None)
+
+        if server_details is None and register_if_not_found is True:
+            server_details = self.register_server(server_id, "<none>")
+
+        return server_details
 
     async def register_server(self, server_id: int, server_name: str) -> None:
         """Register the guild on both db and cache"""

utils/bot.py

@@ -47,7 +47,7 @@ async def on_message(self, message: Message):
         SPOTIFY_TRACK_BASE = "https://open.spotify.com/track/"
 
         if message.content.strip().startswith(SPOTIFY_TRACK_BASE):
-            server_details = await cache_manager.manager.get_server(message.guild.id)
+            server_details = await cache_manager.manager.get_server(message.guild.id, True)
 
             if server_details.get("spotify").get("enabled") is True:
                 splits = message.content.strip().split()