114 lines
3.8 KiB
Python
114 lines
3.8 KiB
Python
from discord import SlashCommandGroup, option, ApplicationContext, Attachment
|
|
from discord.ext.commands import Cog
|
|
from discord.utils import basic_autocomplete
|
|
|
|
from classes import PycordGuild, PycordEventStage, PycordEvent
|
|
from classes.pycord_bot import PycordBot
|
|
from modules.utils import autofill_active_events
|
|
|
|
|
|
class Stage(Cog):
|
|
"""Cog with event stage management commands."""
|
|
|
|
def __init__(self, bot: PycordBot):
|
|
self.bot: PycordBot = bot
|
|
|
|
command_group: SlashCommandGroup = SlashCommandGroup("stage", "Event stage management")
|
|
|
|
# TODO Introduce i18n
|
|
# TODO Maybe add an option for order?
|
|
@command_group.command(
|
|
name="add",
|
|
description="Add new event stage",
|
|
)
|
|
@option(
|
|
"event",
|
|
description="Name of the event",
|
|
autocomplete=basic_autocomplete(autofill_active_events),
|
|
required=True,
|
|
)
|
|
@option("question", description="Question to be answered", required=True)
|
|
@option("answer", description="Answer to the stage's question", required=True)
|
|
@option("media", description="Media file to be attached", required=False)
|
|
async def command_stage_add(
|
|
self,
|
|
ctx: ApplicationContext,
|
|
event: str,
|
|
question: str,
|
|
answer: str,
|
|
media: Attachment = None,
|
|
) -> None:
|
|
guild: PycordGuild = await self.bot.find_guild(ctx.guild.id)
|
|
pycord_event: PycordEvent = await self.bot.find_event(event)
|
|
|
|
if not guild.is_configured():
|
|
# TODO Make a nice message
|
|
await ctx.respond("Guild is not configured.")
|
|
return
|
|
|
|
event_stage: PycordEventStage = await self.bot.create_event_stage(
|
|
event=pycord_event,
|
|
event_id=pycord_event._id,
|
|
guild_id=guild.id,
|
|
sequence=len(pycord_event.stage_ids),
|
|
creator_id=ctx.author.id,
|
|
question=question,
|
|
answer=answer,
|
|
media=None if media is None else media.id,
|
|
)
|
|
|
|
# TODO Make a nice message
|
|
await ctx.respond("Event stage has been created.")
|
|
|
|
# TODO Implement the command
|
|
# /stage edit <event> <stage> <order> <question> <answer> <media>
|
|
@command_group.command(
|
|
name="edit",
|
|
description="Edit the event stage",
|
|
)
|
|
@option(
|
|
"event",
|
|
description="Name of the event",
|
|
autocomplete=basic_autocomplete(autofill_active_events),
|
|
required=True,
|
|
)
|
|
@option("stage", description="Stage to edit", required=True)
|
|
@option("order", description="Number in the event stages' order", required=False)
|
|
@option("question", description="Question to be answered", required=False)
|
|
@option("answer", description="Answer to the stage's question", required=False)
|
|
@option("media", description="Media file to be attached", required=False)
|
|
async def command_stage_edit(
|
|
self,
|
|
ctx: ApplicationContext,
|
|
event: str,
|
|
stage: str,
|
|
order: int,
|
|
question: str,
|
|
answer: str,
|
|
media: Attachment = None,
|
|
) -> None:
|
|
await ctx.respond("Not implemented.")
|
|
|
|
# TODO Implement the command
|
|
# /stage delete <event> <stage> <confirm>
|
|
@command_group.command(
|
|
name="delete",
|
|
description="Delete the event stage",
|
|
)
|
|
@option(
|
|
"event",
|
|
description="Name of the event",
|
|
autocomplete=basic_autocomplete(autofill_active_events),
|
|
required=True,
|
|
)
|
|
@option("stage", description="Stage to edit", required=True)
|
|
@option("confirm", description="Confirmation of the operation", required=False)
|
|
async def command_stage_delete(
|
|
self, ctx: ApplicationContext, event: str, stage: str, confirm: bool = False
|
|
) -> None:
|
|
await ctx.respond("Not implemented.")
|
|
|
|
|
|
def setup(bot: PycordBot) -> None:
|
|
bot.add_cog(Stage(bot))
|