didier/cogs/other.py

55 lines
1.8 KiB
Python
Raw Normal View History

2021-06-19 23:08:42 +02:00
from discord.ext import commands
2021-06-22 00:36:22 +02:00
from data.embeds.snipe import EditSnipe, DeleteSnipe
2021-06-19 23:08:42 +02:00
from data.menus import custom_commands
2021-06-22 00:36:22 +02:00
from data.snipe import Action, Snipe
2021-06-19 23:08:42 +02:00
from decorators import help
from enums.help_categories import Category
from functions.database.custom_commands import get_all
from functions.stringFormatters import capitalize
2021-06-22 00:36:22 +02:00
from startup.didier import Didier
2021-06-19 23:08:42 +02:00
class Other(commands.Cog):
2021-06-22 00:36:22 +02:00
def __init__(self, client: Didier):
self.client: Didier = client
2021-06-19 23:08:42 +02:00
2021-06-22 00:36:22 +02:00
# TODO add locked field to Didier instead of client
# # Don't allow any commands to work when locked
# def cog_check(self, ctx):
# return not self.client.locked
2021-06-19 23:08:42 +02:00
@commands.command(name="Custom")
@help.Category(category=Category.Didier)
async def list_custom(self, ctx):
"""
Get a list of all custom commands
"""
all_commands = get_all()
formatted = list(sorted(map(lambda x: capitalize(x["name"]), all_commands)))
src = custom_commands.CommandsList(formatted)
await custom_commands.Pages(source=src, clear_reactions_after=True).start(ctx)
2021-06-22 00:36:22 +02:00
@commands.command(name="Snipe")
@help.Category(category=Category.Other)
async def snipe(self, ctx):
"""
Shame people for editing & removing messages.
The dict is stored in memory so it will be cleared whenever the bot restarts.
"""
if ctx.guild is None:
return
if ctx.channel.id not in self.client.snipe:
return await ctx.send("Er is hier niemand om uit te lachen.")
s: Snipe = self.client.snipe[ctx.channel.id]
embed_class = (EditSnipe(s) if s.action == Action.Edit else DeleteSnipe(s))
return await ctx.send(embed=embed_class.to_embed(self.client))
2021-06-19 23:08:42 +02:00
def setup(client):
client.add_cog(Other(client))