feature (sponsors): Add guild whitelist.

Add `sponsor_hidden_in` app setting.
Add new `GuildID` and `GuildIDList` setting types.

bot/modules/sponsors/config.py

@@ -1,11 +1,12 @@
 from cmdClient.checks import is_owner
 
 from settings import AppSettings, Setting, KeyValueData, ListData
-from settings.setting_types import Message, String
+from settings.setting_types import Message, String, GuildIDList
 
 from meta import client
 from core.data import app_config
 
+from .data import guild_whitelist
 
 @AppSettings.attach_setting
 class sponsor_prompt(String, KeyValueData, Setting):
@@ -37,6 +38,13 @@ class sponsor_prompt(String, KeyValueData, Setting):
         else:
             return None
 
+    @property
+    def success_response(self):
+        if self.value:
+            return "The sponsor prompt has been update."
+        else:
+            return "The sponsor prompt has been cleared."
+
 
 @AppSettings.attach_setting
 class sponsor_message(Message, KeyValueData, Setting):
@@ -60,3 +68,25 @@ class sponsor_message(Message, KeyValueData, Setting):
     _key = 'sponsor_message'
 
     _cmd_str = "{prefix}sponsors --edit"
+
+    @property
+    def success_response(self):
+        return "The `sponsors` command message has been updated."
+
+
+@AppSettings.attach_setting
+class sponsor_guild_whitelist(GuildIDList, ListData, Setting):
+    attr_name = 'sponsor_guild_whitelist'
+    write_ward = is_owner
+
+    category = 'Sponsors'
+    display_name = 'sponsor_hidden_in'
+    desc = "Guilds where the sponsor prompt is not displayed."
+    long_desc = (
+        "A list of guilds where the sponsor prompt hint will be hidden (see the `sponsor_prompt` setting)."
+    )
+
+    _table_interface = guild_whitelist
+    _id_column = 'appid'
+    _data_column = 'guildid'
+    _force_unique = True

bot/modules/sponsors/data.py

@@ -1,4 +1,4 @@
 from data import Table
 
 
-sponsor_text = Table("sponsor_text")
+guild_whitelist = Table("sponsor_guild_whitelist")

bot/modules/sponsors/module.py

@@ -16,11 +16,12 @@ sponsored_commands = {'profile', 'stats', 'weekly', 'monthly'}
 async def sponsor_reply_wrapper(func, ctx: LionContext, *args, **kwargs):
     if ctx.cmd and ctx.cmd.name in sponsored_commands:
         if (prompt := ctx.client.settings.sponsor_prompt.value):
-            sponsor_hint = discord.Embed(
-                description=prompt,
-                colour=discord.Colour.dark_theme()
-            )
-            if 'embed' not in kwargs:
-                kwargs['embed'] = sponsor_hint
+            if not ctx.guild or ctx.guild.id not in ctx.client.settings.sponsor_guild_whitelist.value:
+                sponsor_hint = discord.Embed(
+                    description=prompt,
+                    colour=discord.Colour.dark_theme()
+                )
+                if 'embed' not in kwargs:
+                    kwargs['embed'] = sponsor_hint
 
     return await func(ctx, *args, **kwargs)