From d17cfbf323725a37646dbbce2f2540a5b73f94ec Mon Sep 17 00:00:00 2001 From: Aevann1 Date: Tue, 20 Dec 2022 05:52:00 +0200 Subject: [PATCH] allow PCM and WPD users to opt into event music --- files/classes/user.py | 2 +- files/events/__init__.py | 2 ++ 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/files/classes/user.py b/files/classes/user.py index 9b55650fbb..eeaaa704bd 100644 --- a/files/classes/user.py +++ b/files/classes/user.py @@ -1144,4 +1144,4 @@ class User(Base): @property @lazy def can_toggle_event_music(self): - return self.has_badge(91) + return SITE_NAME != 'rDrama' or self.has_badge(91) diff --git a/files/events/__init__.py b/files/events/__init__.py index 5deb3f0509..6f1398fe3c 100644 --- a/files/events/__init__.py +++ b/files/events/__init__.py @@ -1,6 +1,7 @@ from sqlalchemy import inspect from files.helpers.config.awards import AWARDS, AWARDS_ENABLED, AWARDS_DISABLED +from files.helpers.config.const import SITE_NAME from files.__main__ import engine from files.events.classes import * @@ -19,6 +20,7 @@ def _populate_awards(): AWARDS_ENABLED.update(temp) for award in EVENT_AWARDS: + if award == 'grinch' and SITE_NAME != 'rDrama': continue EVENT_AWARDS[award]['ghost'] = EVENT_AWARDS[award]['cosmetic'] if award in AWARDS_DISABLED: AWARDS_DISABLED.remove(award)