34 lines
1021 B
Python
34 lines
1021 B
Python
import discord
|
|
|
|
from LionModule import LionModule
|
|
from LionContext import LionContext
|
|
|
|
from meta import client
|
|
|
|
|
|
module = LionModule("Sponsor")
|
|
|
|
|
|
sponsored_commands = {'profile', 'stats', 'weekly', 'monthly'}
|
|
|
|
|
|
@LionContext.reply.add_wrapper
|
|
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 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()
|
|
)
|
|
if 'embed' not in kwargs:
|
|
kwargs['embed'] = sponsor_hint
|
|
|
|
return await func(ctx, *args, **kwargs)
|