Implemented /event edit and /edit cancel

classes/pycord_guild.py

@@ -17,15 +17,16 @@ logger: Logger = get_logger(__name__)
 class PycordGuild:
     """Dataclass of DB entry of a guild"""
 
-    __slots__ = ("_id", "id", "channel_id", "category_id", "timezone")
+    __slots__ = ("_id", "id", "channel_id", "category_id", "timezone", "language")
     __short_name__ = "guild"
     __collection__ = col_guilds
 
     _id: ObjectId
     id: int
-    channel_id: Optional[int]
-    category_id: Optional[int]
+    channel_id: int | None
+    category_id: int | None
     timezone: str
+    language: str | None
 
     @classmethod
     async def from_id(
@@ -67,7 +68,8 @@ class PycordGuild:
 
         return cls(**db_entry)
 
-    async def _set(self, key: str, value: Any, cache: Optional[Cache] = None) -> None:
+    # TODO Update the docstring
+    async def _set(self, cache: Optional[Cache] = None, **kwargs) -> None:
         """Set attribute data and save it into the database.
 
         Args:
@@ -75,36 +77,43 @@ class PycordGuild:
             value (Any): Value to set
             cache (:obj:`Cache`, optional): Cache engine to write the update into
         """
-        if not hasattr(self, key):
-            raise AttributeError()
+        for key, value in kwargs.items():
+            if not hasattr(self, key):
+                raise AttributeError()
 
-        setattr(self, key, value)
+            setattr(self, key, value)
 
-        await self.__collection__.update_one({"_id": self._id}, {"$set": {key: value}}, upsert=True)
+        await self.__collection__.update_one({"_id": self._id}, {"$set": kwargs}, upsert=True)
 
         self._update_cache(cache)
 
-        logger.info("Set attribute '%s' of guild %s to '%s'", key, self.id, value)
+        logger.info("Set attributes of guild %s to %s", self.id, kwargs)
 
-    async def _remove(self, key: str, cache: Optional[Cache] = None) -> None:
+    # TODO Update the docstring
+    async def _remove(self, cache: Optional[Cache] = None, *args: str) -> None:
         """Remove attribute data and save it into the database.
 
         Args:
             key (str): Attribute to remove
             cache (:obj:`Cache`, optional): Cache engine to write the update into
         """
-        if not hasattr(self, key):
-            raise AttributeError()
+        attributes: Dict[str, Any] = {}
 
-        default_value: Any = PycordGuild.get_default_value(key)
+        for key in args:
+            if not hasattr(self, key):
+                raise AttributeError()
 
-        setattr(self, key, default_value)
+            default_value: Any = self.get_default_value(key)
 
-        await self.__collection__.update_one({"_id": self._id}, {"$set": {key: default_value}}, upsert=True)
+            setattr(self, key, default_value)
+
+            attributes[key] = default_value
+
+        await self.__collection__.update_one({"_id": self._id}, {"$set": attributes}, upsert=True)
 
         self._update_cache(cache)
 
-        logger.info("Removed attribute '%s' of guild %s", key, self.id)
+        logger.info("Reset attributes %s of guild %s to default values", args, self.id)
 
     def _get_cache_key(self) -> str:
         return f"{self.__short_name__}_{self.id}"
@@ -141,11 +150,18 @@ class PycordGuild:
             "channel_id": self.channel_id,
             "category_id": self.category_id,
             "timezone": self.timezone,
+            "language": self.language,
         }
 
     @staticmethod
     def get_defaults(guild_id: Optional[int] = None) -> Dict[str, Any]:
-        return {"id": guild_id, "channel_id": None, "category_id": None, "timezone": "UTC"}
+        return {
+            "id": guild_id,
+            "channel_id": None,
+            "category_id": None,
+            "timezone": "UTC",
+            "language": None,
+        }
 
     @staticmethod
     def get_default_value(key: str) -> Any:
@@ -154,6 +170,22 @@ class PycordGuild:
 
         return PycordGuild.get_defaults()[key]
 
+    # TODO Add documentation
+    async def update(
+        self,
+        cache: Optional[Cache] = None,
+        **kwargs,
+    ):
+        await self._set(cache=cache, **kwargs)
+
+    # TODO Add documentation
+    async def reset(
+        self,
+        cache: Optional[Cache] = None,
+        *args,
+    ):
+        await self._remove(cache, *args)
+
     async def purge(self, cache: Optional[Cache] = None) -> None:
         """Completely remove guild data from database. Currently only removes the guild record from guilds collection.
 
@@ -161,8 +193,11 @@ class PycordGuild:
             cache (:obj:`Cache`, optional): Cache engine to write the update into
         """
         await self.__collection__.delete_one({"_id": self._id})
+
         self._delete_cache(cache)
 
+        logger.info("Purged guild %s (%s) from the database", self.id, self._id)
+
     # TODO Add documentation
     def is_configured(self) -> bool:
         return (
@@ -174,24 +209,32 @@ class PycordGuild:
 
     # TODO Add documentation
     async def set_channel(self, channel_id: Optional[int] = None, cache: Optional[Cache] = None) -> None:
-        await self._set("channel_id", channel_id, cache)
+        await self._set(cache, channel_id=channel_id)
 
     # TODO Add documentation
     async def reset_channel(self, cache: Optional[Cache] = None) -> None:
-        await self._remove("channel_id", cache)
+        await self._remove(cache, "channel_id")
 
     # TODO Add documentation
     async def set_category(self, category_id: Optional[int] = None, cache: Optional[Cache] = None) -> None:
-        await self._set("category_id", category_id, cache)
+        await self._set(cache, category_id=category_id)
 
     # TODO Add documentation
     async def reset_category(self, cache: Optional[Cache] = None) -> None:
-        await self._remove("category_id", cache)
+        await self._remove(cache, "category_id")
 
     # TODO Add documentation
     async def set_timezone(self, timezone: str, cache: Optional[Cache] = None) -> None:
-        await self._set("timezone", timezone, cache)
+        await self._set(cache, timezone=timezone)
 
     # TODO Add documentation
     async def reset_timezone(self, cache: Optional[Cache] = None) -> None:
-        await self._remove("timezone", cache)
+        await self._remove(cache, "timezone")
+
+    # TODO Add documentation
+    async def set_language(self, language: str, cache: Optional[Cache] = None) -> None:
+        await self._set(cache, language=language)
+
+    # TODO Add documentation
+    async def reset_language(self, cache: Optional[Cache] = None) -> None:
+        await self._remove(cache, "language")