Skip to content

Commit

Permalink
No real point in having a different set for this
Browse files Browse the repository at this point in the history
Refactor to use ssd instead of a separate list for guild ID's and timers.
  • Loading branch information
BabyBoySnow committed Oct 14, 2023
1 parent 764bc3b commit b5b6f32
Showing 1 changed file with 8 additions and 5 deletions.
13 changes: 8 additions & 5 deletions musicbot/bot.py
Original file line number Diff line number Diff line change
Expand Up @@ -74,7 +74,6 @@ def __init__(self, config_file=None, perms_file=None, aliases_file=None):
aliases_file = AliasesDefault.aliases_file

self.players = {}
self.guild_data = {}
self.exit_signal = None
self.init_ok = False
self.cached_app_info = None
Expand Down Expand Up @@ -114,6 +113,8 @@ def __init__(self, config_file=None, perms_file=None, aliases_file=None):
"last_np_msg": None,
"auto_paused": False,
"availability_paused": False,
"guild_id": None,
"voice_channel_timers": None,
}
self.server_specific_data = defaultdict(ssd_defaults.copy)

Expand Down Expand Up @@ -4430,7 +4431,9 @@ async def gen_cmd_list(self, message, list_all_cmds=False):

async def on_timer_expired(self, voice_channel):
guild_id = voice_channel.guild.id
timers = self.guild_data.get(guild_id, {}).get("voice_channel_timers", {})
timers = self.server_specific_data.get(guild_id, {}).get(
"voice_channel_timers", {}
)
if voice_channel.id in timers:
guild = self.get_guild(guild_id)
vc = guild.get_channel(voice_channel.id)
Expand Down Expand Up @@ -4459,10 +4462,10 @@ async def on_voice_state_update(self, member, before, after):
guild_id = member.guild.id

# Ensure timers are initialized for this guild
if guild_id not in self.guild_data:
self.guild_data[guild_id] = {"voice_channel_timers": {}}
if guild_id not in self.server_specific_data:
self.server_specific_data[guild_id] = {"voice_channel_timers": {}}

timers = self.guild_data[guild_id]["voice_channel_timers"]
timers = self.server_specific_data[guild_id]["voice_channel_timers"]

if before.channel and member != self.user:
if not any(not user.bot for user in before.channel.members):
Expand Down

0 comments on commit b5b6f32

Please sign in to comment.