Cache fix and Better pages

Author: gccody

lib/bot/__init__.py

@@ -16,6 +16,7 @@
 from discord.ext.commands import AutoShardedBot
 from discord.ext.commands.errors import CommandNotFound, BadArgument
 
+from lib.cache import Cache
 from lib.config import Config
 from lib.db import DB
 from lib.progress import Progress, Timer, Loading
@@ -41,7 +42,7 @@ class Bot(AutoShardedBot):
     def __init__(self, shards: list[int], version: str):
         self.config: Config = Config()
         self.db: DB = DB()
-        self.cache: dict = dict()
+        self.cache: Cache = Cache()
         self.tasks = AsyncIOScheduler(timezone=str(tzlocal.get_localzone()))
         self.shards_ready = False
         self.ready = False

lib/cache/__init__.py

@@ -1,4 +1,45 @@
-from lib.levels import Player
+import json
+
+
+class Get:
+    def __init__(self, parent):
+        self.parent: Cache = parent
+
+    def __getattribute__(self, key):
+        parent: Cache = super().__getattribute__("parent")
+        if key in parent.data.keys():
+            _get_item = super().__getattribute__("_get_item")
+            return lambda: _get_item(key)
+        return super().__getattribute__(key)
+
+    def _get_item(self, key):
+        return self.parent.data.get(key)
+
+
+class Set:
+    def __init__(self, parent):
+        self.parent: Cache = parent
+
+    def __getattribute__(self, key):
+        if key in super().__getattribute__("__dict__"):
+            return super().__getattribute__(key)
+        _set_item = super().__getattribute__("_set_item")
+        return lambda value: _set_item(key, value)
+
+    def _set_item(self, key, value):
+        self.parent._data[key] = value
+
 
 class Cache:
-    levels: dict[str, Player]
+    _data = {}
+
+    def __init__(self):
+        self.get: Get = Get(self)
+        self.set: Set = Set(self)
+
+    @property
+    def data(self):
+        return self._data
+
+    def __str__(self):
+        return json.dumps(self._data, indent=4)

lib/pages/__init__.py

@@ -48,7 +48,7 @@ def handle_buttons(self):
         self.previous.disabled = self.current_page == 0
         self.next.disabled = self.current_page == self.total_page_count - 1
 
-    @discord.ui.button(emoji=discord.PartialEmoji(name="\U000023ee"), disabled=True)
+    @discord.ui.button(label="<<", disabled=True, style=discord.ButtonStyle.blurple)
     async def first(self, interaction: discord.Interaction, _):
         if interaction.user != self.ctx.user:
             embed = discord.Embed(description="You cannot control this pagination because you did not execute it.",
@@ -59,7 +59,7 @@ async def first(self, interaction: discord.Interaction, _):
         self.counter.label = f"{self.current_page + 1}/{self.total_page_count}"
         await interaction.response.edit_message(embed=self.pages[self.current_page], view=self)
 
-    @discord.ui.button(emoji=discord.PartialEmoji(name="\U000025c0"), disabled=True)
+    @discord.ui.button(label="<", disabled=True, style=discord.ButtonStyle.red)
     async def previous(self, interaction: discord.Interaction, _):
         if interaction.user != self.ctx.user:
             embed = discord.Embed(description="You cannot control this pagination because you did not execute it.",
@@ -74,7 +74,7 @@ async def previous(self, interaction: discord.Interaction, _):
     async def counter(self, interaction: discord.Interaction, button: discord.Button):
         pass
 
-    @discord.ui.button(emoji=discord.PartialEmoji(name="\U000025b6"))
+    @discord.ui.button(label=">", style=discord.ButtonStyle.green)
     async def next(self, interaction: discord.Interaction, _):
         if interaction.user != self.ctx.user:
             embed = discord.Embed(description="You cannot control this pagination because you did not execute it.",
@@ -85,7 +85,7 @@ async def next(self, interaction: discord.Interaction, _):
         self.counter.label = f"{self.current_page + 1}/{self.total_page_count}"
         await interaction.response.edit_message(embed=self.pages[self.current_page], view=self)
 
-    @discord.ui.button(emoji=discord.PartialEmoji(name="\U000023ed"))
+    @discord.ui.button(label=">>", style=discord.ButtonStyle.blurple)
     async def last(self, interaction: discord.Interaction, _):
         if interaction.user != self.ctx.user:
             embed = discord.Embed(description="You cannot control this pagination because you did not execute it.",