diff --git a/bot/modules/sponsors/module.py b/bot/modules/sponsors/module.py index ae9ba299..232eafa6 100644 --- 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() diff --git a/bot/modules/topgg/module.py b/bot/modules/topgg/module.py index a3872192..3632e9f1 100644 --- a/bot/modules/topgg/module.py +++ b/bot/modules/topgg/module.py @@ -47,6 +47,8 @@ async def topgg_reply_wrapper(func, ctx: LionContext, *args, suggest_vote=True, pass elif ctx.guild and ctx.guild.id in ctx.client.settings.topgg_guild_whitelist.value: pass + elif ctx.guild and ctx.client.data.premium_guilds.queries.fetch_guild(ctx.guild.id): + pass elif not get_last_voted_timestamp(ctx.author.id): upvote_info_formatted = upvote_info.format(lion_yayemote, ctx.best_prefix, lion_loveemote)