profitroll/QuizBot
1
0
Fork 0
Code Issues 5 Pull Requests Actions Projects 1 Releases 2 Activity

Added validation for ongoing events (#2)

Browse Source
This commit is contained in:
Profitroll
2025-04-22 23:22:45 +02:00
parent 5230ac9ace
commit 5bf1537968
1 changed files with 36 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

43
cogs/stage.py
View File

@@ -1,12 +1,30 @@
from discord import SlashCommandGroup, option, ApplicationContext, Attachment
from datetime import datetime
from zoneinfo import ZoneInfo
from discord import ApplicationContext, Attachment, SlashCommandGroup, option
from discord.ext.commands import Cog
from discord.utils import basic_autocomplete
from classes import PycordGuild, PycordEventStage, PycordEvent
from classes import PycordEvent, PycordEventStage, PycordGuild
from classes.pycord_bot import PycordBot
from modules.utils import autocomplete_active_events, autocomplete_event_stages
async def validate_event_status(
ctx: ApplicationContext,
event: PycordEvent,
) -> None:
if event.cancelled:
# TODO Make a nice message
await ctx.respond("This event was cancelled.")
return
if event.starts <= datetime.now(tz=ZoneInfo("UTC")) <= event.ends:
# TODO Make a nice message
await ctx.respond("Ongoing events cannot be modified.")
return
class Stage(Cog):
"""Cog with event stage management commands."""
@@ -39,13 +57,16 @@ class Stage(Cog):
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
pycord_event: PycordEvent = await self.bot.find_event(event)
await validate_event_status(ctx, pycord_event)
event_stage: PycordEventStage = await self.bot.create_event_stage(
event=pycord_event,
event_id=pycord_event._id,
@@ -96,14 +117,18 @@ class Stage(Cog):
remove_media: bool = False,
) -> None:
guild: PycordGuild = await self.bot.find_guild(ctx.guild.id)
pycord_event: PycordEvent = await self.bot.find_event(event)
event_stage: PycordEventStage = await self.bot.find_event_stage(stage)
if not guild.is_configured():
# TODO Make a nice message
await ctx.respond("Guild is not configured.")
return
pycord_event: PycordEvent = await self.bot.find_event(event)
await validate_event_status(ctx, pycord_event)
event_stage: PycordEventStage = await self.bot.find_event_stage(stage)
if order is not None and order > len(pycord_event.stage_ids):
# TODO Make a nice message
await ctx.respond("Stage sequence out of range.")
@@ -149,14 +174,18 @@ class Stage(Cog):
return
guild: PycordGuild = await self.bot.find_guild(ctx.guild.id)
pycord_event: PycordEvent = await self.bot.find_event(event)
event_stage: PycordEventStage = await self.bot.find_event_stage(stage)
if not guild.is_configured():
# TODO Make a nice message
await ctx.respond("Guild is not configured.")
return
pycord_event: PycordEvent = await self.bot.find_event(event)
await validate_event_status(ctx, pycord_event)
event_stage: PycordEventStage = await self.bot.find_event_stage(stage)
await pycord_event.remove_stage(self.bot, event_stage._id, cache=self.bot.cache)
await event_stage.purge(cache=self.bot.cache)