Conatum
9ea40e5433
Add `sponsor_hidden_in` app setting. Add new `GuildID` and `GuildIDList` setting types.
28 lines
836 B
Python
28 lines
836 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 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)
|