Implemented /event edit and /edit cancel

classes/pycord_event.py

@@ -1,7 +1,8 @@
 from dataclasses import dataclass
-from datetime import datetime, timezone
+from datetime import datetime
 from logging import Logger
 from typing import Any, Dict, List, Optional
+from zoneinfo import ZoneInfo
 
 from bson import ObjectId
 from libbot.cache.classes import Cache
@@ -21,6 +22,7 @@ class PycordEvent:
         "guild_id",
         "created",
         "ended",
+        "cancelled",
         "creator_id",
         "starts",
         "ends",
@@ -35,6 +37,7 @@ class PycordEvent:
     guild_id: int
     created: datetime
     ended: datetime | None
+    cancelled: bool
     creator_id: int
     starts: datetime
     ends: datetime
@@ -85,28 +88,31 @@ class PycordEvent:
         if db_entry is None:
             raise RuntimeError(f"Event with name {event_name} not found")
 
+            # TODO Add a unique exception
+            # raise EventNotFoundError(event_name)
+
         if cache is not None:
             cache.set_json(f"{cls.__short_name__}_{db_entry['_id']}", db_entry)
 
         return cls(**db_entry)
 
-    # TODO Implement this method
     @classmethod
     async def create(
-            cls,
-            name: str,
-            guild_id: int,
-            creator_id: int,
-            starts: datetime,
-            ends: datetime,
-            thumbnail_id: str | None,
-            cache: Optional[Cache] = None,
+        cls,
+        name: str,
+        guild_id: int,
+        creator_id: int,
+        starts: datetime,
+        ends: datetime,
+        thumbnail_id: str | None,
+        cache: Optional[Cache] = None,
     ) -> "PycordEvent":
         db_entry: Dict[str, Any] = {
             "name": name,
             "guild_id": guild_id,
-            "created": datetime.now(tz=timezone.utc),
+            "created": datetime.now(tz=ZoneInfo("UTC")),
             "ended": None,
+            "cancelled": False,
             "creator_id": creator_id,
             "starts": starts,
             "ends": ends,
@@ -123,7 +129,8 @@ class PycordEvent:
 
         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:
@@ -131,36 +138,43 @@ class PycordEvent:
             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 event %s to '%s'", key, self._id, value)
+        logger.info("Set attributes of event %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 = PycordEvent.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 event %s", key, self._id)
+        logger.info("Reset attributes %s of event %s to default values", args, self._id)
 
     def _get_cache_key(self) -> str:
         return f"{self.__short_name__}_{self._id}"
@@ -197,6 +211,7 @@ class PycordEvent:
             "guild_id": self.guild_id,
             "created": self.created,
             "ended": self.ended,
+            "cancelled": self.cancelled,
             "creator_id": self.creator_id,
             "starts": self.starts,
             "ends": self.ends,
@@ -211,6 +226,7 @@ class PycordEvent:
             "guild_id": None,
             "created": None,
             "ended": None,
+            "cancelled": False,
             "creator_id": None,
             "starts": None,
             "ends": None,
@@ -225,6 +241,22 @@ class PycordEvent:
 
         return PycordEvent.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 event data from database. Currently only removes the event record from events collection.
 
@@ -233,3 +265,6 @@ class PycordEvent:
         """
         await self.__collection__.delete_one({"_id": self._id})
         self._delete_cache(cache)
+
+    async def cancel(self, cache: Optional[Cache] = None):
+        await self._set(cache, cancelled=True)