Croc/croccybot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix (schedule): Various bug fixes.

Browse Source
This commit is contained in:
Conatum
2023-06-30 12:34:21 +03:00
parent a7c5af59a7
commit 65c17f11b2
3 changed files with 20 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/modules/schedule/cog.py
View File

@@ -489,7 +489,8 @@ class ScheduleCog(LionCog):
session = slot.sessions.get(guildid, None)
if session is None:
# Create a new session in the slot and set it up
session = await slot.load_sessions(session_data[guildid, slotid])
sessions = await slot.load_sessions([session_data[guildid, slotid]])
session = sessions[guildid]
slot.sessions[guildid] = session
if slot.closing.is_set():
# This should never happen

29
src/modules/schedule/core/timeslot.py
View File

@@ -210,20 +210,21 @@ class TimeSlot:
await batchrun_per_second(coros, 5)
# Save messageids
tmptable = TemporaryTable(
'_gid', '_sid', '_mid',
types=('BIGINT', 'INTEGER', 'BIGINT')
)
tmptable.values = [
(sg.data.guildid, sg.data.slotid, sg.messageid)
for sg in sessions
if sg.messageid is not None
]
await Data.ScheduleSession.table.update_where(
guildid=tmptable['_gid'], slotid=tmptable['_sid']
).set(
messageid=tmptable['_mid']
).from_expr(tmptable)
if sessions:
tmptable = TemporaryTable(
'_gid', '_sid', '_mid',
types=('BIGINT', 'INTEGER', 'BIGINT')
)
tmptable.values = [
(sg.data.guildid, sg.data.slotid, sg.messageid)
for sg in sessions
if sg.messageid is not None
]
await Data.ScheduleSession.table.update_where(
guildid=tmptable['_gid'], slotid=tmptable['_sid']
).set(
messageid=tmptable['_mid']
).from_expr(tmptable)
except Exception:
logger.exception(
f"Unhandled exception while preparing timeslot <slotid: {self.slotid}>."