Improved permission loading #446

bot/src/modules/level/command/level_group.py

@@ -24,7 +24,7 @@ from bot_data.model.level import Level
 from bot_data.model.server_config import ServerConfig
 from modules.level.level_seeder import LevelSeeder
 from modules.level.service.level_service import LevelService
-from modules.permission.abc.permission_service_abc import PermissionServiceABC
+from bot_core.abc.permission_service_abc import PermissionServiceABC
 
 
 class LevelGroup(DiscordCommandABC):

bot/src/modules/level/service/level_service.py

@@ -6,6 +6,8 @@ from cpl_discord.container import Guild, Role, Member
 from cpl_discord.service import DiscordBotServiceABC
 from cpl_translation import TranslatePipe
 
+from bot_core.configuration.feature_flags_enum import FeatureFlagsEnum
+from bot_core.configuration.feature_flags_settings import FeatureFlagsSettings
 from bot_core.service.message_service import MessageService
 from bot_data.model.level import Level
 from bot_data.model.server_config import ServerConfig
@@ -38,7 +40,14 @@ class LevelService:
         self._message_service = message_service
         self._t = t
 
+    def _check_for_feature(self, user: User):
+        server_config: ServerConfig = self._config.get_configuration(f"ServerConfig_{user.server.discord_id}")
+        if not FeatureFlagsSettings.get_flag_from_dict(server_config.feature_flags, FeatureFlagsEnum.level_module):
+            raise Exception(f"Feature {FeatureFlagsEnum.level_module.value} disabled")
+
     def get_level(self, user: User) -> Level:
+        self._check_for_feature(user)
+
         levels_by_server = self._levels.get_levels_by_server_id(user.server.id)
         if user.xp < 0:
             return levels_by_server.order_by(lambda l: l.min_xp).first()
@@ -51,6 +60,7 @@ class LevelService:
         return levels.last()
 
     async def set_level(self, user: User):
+        self._check_for_feature(user)
         level_names = self._levels.get_levels_by_server_id(user.server.id).select(lambda l: l.name)
         guild: Guild = self._bot.guilds.where(lambda g: g.id == user.server.discord_id).single()
         member: Member = guild.members.where(lambda m: m.id == user.discord_id).single()