Closes #13

classes/pycord_user.py

@@ -17,7 +17,7 @@ from discord.abc import GuildChannel
 from libbot.cache.classes import Cache
 from pymongo.results import InsertOneResult
 
-from classes.abstract import Cacheable
+from classes.base.base_cacheable import BaseCacheable
 from classes.errors import (
     DiscordCategoryNotFoundError,
     DiscordChannelNotFoundError,
@@ -34,17 +34,9 @@ logger: Logger = get_logger(__name__)
 
 
 @dataclass
-class PycordUser(Cacheable):
+class PycordUser(BaseCacheable):
     """Dataclass of DB entry of a user"""
 
-    # TODO Implement this
-    async def update(self, cache: Optional[Cache] = None, **kwargs: Any) -> None:
-        pass
-
-    # TODO Implement this
-    async def reset(self, *args: str, cache: Optional[Cache] = None) -> None:
-        pass
-
     __slots__ = (
         "_id",
         "id",
@@ -145,68 +137,19 @@ class PycordUser(Cacheable):
         }
 
     async def _set(self, cache: Optional[Cache] = None, **kwargs: Any) -> None:
-        """Set attribute data and save it into the database.
-
-        Args:
-            cache (:obj:`Cache`, optional): Cache engine to write the update into
-            **kwargs (Any): Mapping of attribute names and respective values to be set
-        """
-        for key, value in kwargs.items():
-            if not hasattr(self, key):
-                raise AttributeError()
-
-            setattr(self, key, value)
-
-        await self.__collection__.update_one({"_id": self._id}, {"$set": kwargs})
-
-        self._update_cache(cache)
-
-        logger.info("Set attributes of user %s to %s", self.id, kwargs)
+        await super()._set(cache, **kwargs)
 
     async def _remove(self, *args: str, cache: Optional[Cache] = None) -> None:
-        """Remove attribute data and save it into the database.
-
-        Args:
-            cache (:obj:`Cache`, optional): Cache engine to write the update into
-            *args (str): List of attributes to remove
-        """
-        attributes: Dict[str, Any] = {}
-
-        for key in args:
-            if not hasattr(self, key):
-                raise AttributeError()
-
-            default_value: Any = self.get_default_value(key)
-
-            setattr(self, key, default_value)
-
-            attributes[key] = default_value
-
-        await self.__collection__.update_one({"_id": self._id}, {"$set": attributes})
-
-        self._update_cache(cache)
-
-        logger.info("Reset attributes %s of user %s to default values", args, self.id)
+        await super()._remove(*args, cache=cache)
 
     def _get_cache_key(self) -> str:
         return f"{self.__short_name__}_{self.id}_{self.guild_id}"
 
     def _update_cache(self, cache: Optional[Cache] = None) -> None:
-        if cache is None:
-            return
-
-        object_dict: Dict[str, Any] = self.to_dict(json_compatible=True)
-
-        if object_dict is not None:
-            cache.set_json(self._get_cache_key(), object_dict)
-        else:
-            self._delete_cache(cache)
+        super()._update_cache(cache)
 
     def _delete_cache(self, cache: Optional[Cache] = None) -> None:
-        if cache is None:
-            return
-
-        cache.delete(self._get_cache_key())
+        super()._delete_cache(cache)
 
     @staticmethod
     def _entry_to_cache(db_entry: Dict[str, Any]) -> Dict[str, Any]:
@@ -270,14 +213,22 @@ class PycordUser(Cacheable):
 
         return PycordUser.get_defaults()[key]
 
-    async def purge(self, cache: Optional[Cache] = None) -> None:
-        """Completely remove user data from database. Currently only removes the user record from users collection.
+    async def update(
+        self,
+        cache: Optional[Cache] = None,
+        **kwargs: Any,
+    ) -> None:
+        await super().update(cache=cache, **kwargs)
 
-        Args:
-            cache (:obj:`Cache`, optional): Cache engine to write the update into
-        """
-        await self.__collection__.delete_one({"_id": self._id})
-        self._delete_cache(cache)
+    async def reset(
+        self,
+        *args: str,
+        cache: Optional[Cache] = None,
+    ) -> None:
+        await super().reset(*args, cache=cache)
+
+    async def purge(self, cache: Optional[Cache] = None) -> None:
+        await super().purge(cache)
 
     # TODO Add documentation
     async def event_register(self, event_id: str | ObjectId, cache: Optional[Cache] = None) -> None: