cswimr/SeaCogs
cswimr/SeaCogs
1
1
Fork 1
Code Issues 11 Pull requests 9 Projects 1 Wiki Activity Actions

feat(aurora): added a return_obj parameter to Moderation.execute()

Browse source
This commit is contained in:
Seaswimmer 2024-06-05 01:31:40 -04:00
parent 42f7f9f69b
commit e988917319
Signed by: cswimr
GPG key ID: 5D671B5D03D65A7F
2 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

4
aurora/models/moderation.py
View file

@ -208,7 +208,7 @@ class Moderation(AuroraGuildModel):
return cls.from_dict(bot=bot, data=case)
@classmethod
async def execute(cls, bot: Red, guild_id: int, query: str, parameters: tuple | None = None, cursor: Cursor | None = None) -> Tuple["Moderation"]:
async def execute(cls, bot: Red, guild_id: int, query: str, parameters: tuple | None = None, cursor: Cursor | None = None, return_obj: bool = True) -> Tuple["Moderation"]:
from ..utilities.database import connect
logger.trace("Executing query: %s", query)
logger.trace("With parameters: %s", parameters)
@ -227,7 +227,7 @@ class Moderation(AuroraGuildModel):
await cursor.close()
await database.close()
if results:
if results and return_obj:
cases = []
for result in results:
case = cls.from_result(bot=bot, result=result, guild_id=guild_id)