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

fix (rooms): Avoid loading rooms in dead guilds.

Browse Source
This commit is contained in:
Conatum
2021-11-07 03:16:56 +02:00
parent b8702c14c3
commit 10f048fabc
2 changed files with 19 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
bot/modules/accountability/TimeSlot.py
View File

@@ -218,6 +218,9 @@ class TimeSlot:
""" """
Load data and update applicable caches. Load data and update applicable caches.
""" """
if not self.guild:
return self
# Load setting data # Load setting data
self.category = GuildSettings(self.guild.id).accountability_category.value self.category = GuildSettings(self.guild.id).accountability_category.value
self.lobby = GuildSettings(self.guild.id).accountability_lobby.value self.lobby = GuildSettings(self.guild.id).accountability_lobby.value
@@ -389,6 +392,7 @@ class TimeSlot:
pass pass
# Reward members appropriately # Reward members appropriately
if self.guild:
guild_settings = GuildSettings(self.guild.id) guild_settings = GuildSettings(self.guild.id)
reward = guild_settings.accountability_reward.value reward = guild_settings.accountability_reward.value
if all(mem.has_attended for mem in self.members.values()): if all(mem.has_attended for mem in self.members.values()):

5
bot/modules/accountability/tracker.py
View File

@@ -374,14 +374,15 @@ async def _accountability_system_resume():
None, mow.slotid, mow.userid) None, mow.slotid, mow.userid)
for mow in slot_members[row.slotid] if mow.last_joined_at for mow in slot_members[row.slotid] if mow.last_joined_at
) )
if client.get_guild(row.guildid):
slot = TimeSlot(client.get_guild(row.guildid), row.start_at, data=row).load( slot = TimeSlot(client.get_guild(row.guildid), row.start_at, data=row).load(
memberids=[mow.userid for mow in slot_members[row.slotid]] memberids=[mow.userid for mow in slot_members[row.slotid]]
) )
row.closed_at = now
try: try:
await slot.close() await slot.close()
except discord.HTTPException: except discord.HTTPException:
pass pass
row.closed_at = now
# Load the in-progress room data # Load the in-progress room data
if current_room_data: if current_room_data:
@@ -451,7 +452,7 @@ async def launch_accountability_system(client):
guilds = tables.guild_config.fetch_rows_where( guilds = tables.guild_config.fetch_rows_where(
accountability_category=NOTNULL accountability_category=NOTNULL
) )
[AccountabilityGuild(guild.guildid) for guild in guilds] [AccountabilityGuild(guild.guildid) for guild in guilds if client.get_guild(guild.guildid)]
await _accountability_system_resume() await _accountability_system_resume()
asyncio.create_task(_accountability_loop()) asyncio.create_task(_accountability_loop())