profitroll/QuizBot
1
0
Fork 0
Code Issues 5 Pull Requests Actions Projects 1 Releases 2 Activity

Fixed _set() and _remove() methods

Browse Source
This commit is contained in:
Profitroll
2025-04-22 23:28:58 +02:00
parent 5bf1537968
commit 004b5336d8
4 changed files with 30 additions and 34 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
classes/pycord_event.py
View File

@@ -129,14 +129,12 @@ class PycordEvent:
return cls(**db_entry) return cls(**db_entry)
# TODO Update the docstring
async def _set(self, cache: Optional[Cache] = None, **kwargs) -> None: async def _set(self, cache: Optional[Cache] = None, **kwargs) -> None:
"""Set attribute data and save it into the database. """Set attribute data and save it into the database.
Args: Args:
key (str): Attribute to change
value (Any): Value to set
cache (:obj:`Cache`, optional): Cache engine to write the update into cache (:obj:`Cache`, optional): Cache engine to write the update into
**kwargs (str): Mapping of attribute names and respective values to be set
""" """
for key, value in kwargs.items(): for key, value in kwargs.items():
if not hasattr(self, key): if not hasattr(self, key):
@@ -150,13 +148,12 @@ class PycordEvent:
logger.info("Set attributes of event %s to %s", self._id, kwargs) logger.info("Set attributes of event %s to %s", self._id, kwargs)
# TODO Update the docstring async def _remove(self, *args: str, cache: Optional[Cache] = None) -> None:
async def _remove(self, cache: Optional[Cache] = None, *args: str) -> None:
"""Remove attribute data and save it into the database. """Remove attribute data and save it into the database.
Args: Args:
key (str): Attribute to remove
cache (:obj:`Cache`, optional): Cache engine to write the update into cache (:obj:`Cache`, optional): Cache engine to write the update into
*args (str): List of attributes to remove
""" """
attributes: Dict[str, Any] = {} attributes: Dict[str, Any] = {}

9
classes/pycord_event_stage.py
View File

@@ -106,14 +106,12 @@ class PycordEventStage:
return cls(**db_entry) return cls(**db_entry)
# TODO Update the docstring
async def _set(self, cache: Optional[Cache] = None, **kwargs) -> None: async def _set(self, cache: Optional[Cache] = None, **kwargs) -> None:
"""Set attribute data and save it into the database. """Set attribute data and save it into the database.
Args: Args:
key (str): Attribute to change
value (Any): Value to set
cache (:obj:`Cache`, optional): Cache engine to write the update into cache (:obj:`Cache`, optional): Cache engine to write the update into
**kwargs (str): Mapping of attribute names and respective values to be set
""" """
for key, value in kwargs.items(): for key, value in kwargs.items():
if not hasattr(self, key): if not hasattr(self, key):
@@ -127,13 +125,12 @@ class PycordEventStage:
logger.info("Set attributes of event stage %s to %s", self._id, kwargs) logger.info("Set attributes of event stage %s to %s", self._id, kwargs)
# TODO Update the docstring async def _remove(self, *args: str, cache: Optional[Cache] = None) -> None:
async def _remove(self, cache: Optional[Cache] = None, *args: str) -> None:
"""Remove attribute data and save it into the database. """Remove attribute data and save it into the database.
Args: Args:
key (str): Attribute to remove
cache (:obj:`Cache`, optional): Cache engine to write the update into cache (:obj:`Cache`, optional): Cache engine to write the update into
*args (str): List of attributes to remove
""" """
attributes: Dict[str, Any] = {} attributes: Dict[str, Any] = {}

9
classes/pycord_guild.py
View File

@@ -67,14 +67,12 @@ class PycordGuild:
return cls(**db_entry) return cls(**db_entry)
# TODO Update the docstring
async def _set(self, cache: Optional[Cache] = None, **kwargs) -> None: async def _set(self, cache: Optional[Cache] = None, **kwargs) -> None:
"""Set attribute data and save it into the database. """Set attribute data and save it into the database.
Args: Args:
key (str): Attribute to change
value (Any): Value to set
cache (:obj:`Cache`, optional): Cache engine to write the update into cache (:obj:`Cache`, optional): Cache engine to write the update into
**kwargs (str): Mapping of attribute names and respective values to be set
""" """
for key, value in kwargs.items(): for key, value in kwargs.items():
if not hasattr(self, key): if not hasattr(self, key):
@@ -88,13 +86,12 @@ class PycordGuild:
logger.info("Set attributes of guild %s to %s", self.id, kwargs) logger.info("Set attributes of guild %s to %s", self.id, kwargs)
# TODO Update the docstring async def _remove(self, *args: str, cache: Optional[Cache] = None) -> None:
async def _remove(self, cache: Optional[Cache] = None, *args: str) -> None:
"""Remove attribute data and save it into the database. """Remove attribute data and save it into the database.
Args: Args:
key (str): Attribute to remove
cache (:obj:`Cache`, optional): Cache engine to write the update into cache (:obj:`Cache`, optional): Cache engine to write the update into
*args (str): List of attributes to remove
""" """
attributes: Dict[str, Any] = {} attributes: Dict[str, Any] = {}

37
classes/pycord_user.py
View File

@@ -77,44 +77,49 @@ class PycordUser:
"id": self.id, "id": self.id,
} }
async def _set(self, key: str, value: Any, cache: Optional[Cache] = None) -> None: async def _set(self, cache: Optional[Cache] = None, **kwargs) -> None:
"""Set attribute data and save it into the database. """Set attribute data and save it into the database.
Args: Args:
key (str): Attribute to change
value (Any): Value to set
cache (:obj:`Cache`, optional): Cache engine to write the update into cache (:obj:`Cache`, optional): Cache engine to write the update into
**kwargs (str): Mapping of attribute names and respective values to be set
""" """
if not hasattr(self, key): for key, value in kwargs.items():
raise AttributeError() 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) self._update_cache(cache)
logger.info("Set attribute '%s' of user %s to '%s'", key, self.id, value) logger.info("Set attributes of user %s to %s", self.id, kwargs)
async def _remove(self, key: str, cache: Optional[Cache] = None) -> None: async def _remove(self, *args: str, cache: Optional[Cache] = None) -> None:
"""Remove attribute data and save it into the database. """Remove attribute data and save it into the database.
Args: Args:
key (str): Attribute to remove
cache (:obj:`Cache`, optional): Cache engine to write the update into cache (:obj:`Cache`, optional): Cache engine to write the update into
*args (str): List of attributes to remove
""" """
if not hasattr(self, key): attributes: Dict[str, Any] = {}
raise AttributeError()
default_value: Any = PycordUser.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) self._update_cache(cache)
logger.info("Removed attribute '%s' of user %s", key, self.id) logger.info("Reset attributes %s of user %s to default values", args, self.id)
def _get_cache_key(self) -> str: def _get_cache_key(self) -> str:
return f"{self.__short_name__}_{self.id}" return f"{self.__short_name__}_{self.id}"