(premium): Remove ad messages from premium guilds.

2022-05-12 00:43:49 +03:00
parent 8c9e2e6cd8
commit 5be84facd3
2 changed files with 9 additions and 1 deletions
--- a/bot/modules/sponsors/module.py
+++ b/bot/modules/sponsors/module.py
@@ -16,7 +16,13 @@ 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):
-            if not ctx.guild or ctx.guild.id not in ctx.client.settings.sponsor_guild_whitelist.value:
+            if ctx.guild:
+                show = ctx.guild.id not in ctx.client.settings.sponsor_guild_whitelist.value
+                show = show and not ctx.client.data.premium_guilds.queries.fetch_guild(ctx.guild.id)
+            else:
+                show = True
+
+            if show:
                sponsor_hint = discord.Embed(
                    description=prompt,
                    colour=discord.Colour.dark_theme()