Rosie-Val/SeaCogs
1
0
Fork 0
Code Pull requests Activity

feat(aurora): allow the Moderation.execute() classmethod to accept a cursor, to prevent opening a new database connection on every call

Browse source
This commit is contained in:
Seaswimmer 2024-06-03 23:48:21 -04:00
parent bbe8b281d1
commit 460d5a31fc
Signed by untrusted user: cswimr
GPG key ID: 5D671B5D03D65A7F
1 changed files with 19 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

31
aurora/models/moderation.py
View file

@ -202,23 +202,30 @@ class Moderation(AuroraGuildModel):
return cls.from_dict(bot=bot, data=case)
@classmethod
def execute(cls, bot: Red, guild_id: int, query: str, parameters: tuple | None = None) -> Tuple["Moderation"]:
def execute(cls, bot: Red, guild_id: int, query: str, parameters: tuple | None = None, cursor: Cursor | None = None) -> Tuple["Moderation"]:
from ..utilities.database import connect
if not parameters:
parameters = ()
with connect() as database:
if not cursor:
no_cursor = True
database = connect()
cursor = database.cursor()
cursor.execute(query, parameters)
results = cursor.fetchall()
cursor.close()
else:
no_cursor = False
if results:
cases = []
for result in results:
case = cls.from_result(bot=bot, result=result, guild_id=guild_id)
if case.moderation_id != 0:
cases.append(case)
return tuple(cases)
cursor.execute(query, parameters)
results = cursor.fetchall()
if no_cursor:
cursor.close()
database.close()
if results:
cases = []
for result in results:
case = cls.from_result(bot=bot, result=result, guild_id=guild_id)
if case.moderation_id != 0:
cases.append(case)
return tuple(cases)
return ()
@classmethod