From 457a5e39fcb0f34477c7ac22c5ed8919f6522eec Mon Sep 17 00:00:00 2001 From: Cerapter Date: Sun, 19 Aug 2018 08:03:39 +0200 Subject: [PATCH] Jukebox fixes: check if jukebox exists + blockDJ removes vote. --- server/area_manager.py | 6 ++++++ server/commands.py | 1 + 2 files changed, 7 insertions(+) diff --git a/server/area_manager.py b/server/area_manager.py index 583e149..90229d0 100644 --- a/server/area_manager.py +++ b/server/area_manager.py @@ -115,6 +115,8 @@ class AreaManager: return True def add_jukebox_vote(self, client, music_name, length=-1, showname=''): + if not self.jukebox: + return if length <= 0: self.remove_jukebox_vote(client, False) else: @@ -125,6 +127,8 @@ class AreaManager: self.start_jukebox() def remove_jukebox_vote(self, client, silent): + if not self.jukebox: + return for current_vote in self.jukebox_votes: if current_vote.client.id == client.id: self.jukebox_votes.remove(current_vote) @@ -132,6 +136,8 @@ class AreaManager: client.send_host_message('You removed your song from the jukebox.') def get_jukebox_picked(self): + if not self.jukebox: + return if len(self.jukebox_votes) == 0: return None elif len(self.jukebox_votes) == 1: diff --git a/server/commands.py b/server/commands.py index a3c3ce8..5dfe030 100644 --- a/server/commands.py +++ b/server/commands.py @@ -812,6 +812,7 @@ def ooc_cmd_blockdj(client, arg): for target in targets: target.is_dj = False target.send_host_message('A moderator muted you from changing the music.') + target.area.remove_jukebox_vote(target, True) client.send_host_message('blockdj\'d {}.'.format(targets[0].get_char_name())) def ooc_cmd_unblockdj(client, arg):