Implemented #7
This commit is contained in:
@@ -1,5 +1,10 @@
|
||||
from discord import ApplicationContext, Cog, option, slash_command
|
||||
from typing import List
|
||||
|
||||
from bson import ObjectId
|
||||
from bson.errors import InvalidId
|
||||
from discord import ApplicationContext, Cog, option, slash_command, File
|
||||
|
||||
from classes import PycordEvent, PycordUser, PycordGuild, PycordEventStage
|
||||
from classes.pycord_bot import PycordBot
|
||||
|
||||
|
||||
@@ -16,7 +21,64 @@ class Guess(Cog):
|
||||
)
|
||||
@option("answer", description="An answer to the current stage")
|
||||
async def command_guess(self, ctx: ApplicationContext, answer: str) -> None:
|
||||
await ctx.respond("Not implemented.")
|
||||
guild: PycordGuild = await self.bot.find_guild(ctx.guild.id)
|
||||
|
||||
if not guild.is_configured():
|
||||
# TODO Make a nice message
|
||||
await ctx.respond("Guild is not configured.")
|
||||
return
|
||||
|
||||
user: PycordUser = await self.bot.find_user(ctx.author)
|
||||
|
||||
if user.current_event_id is None or user.current_stage_id is None:
|
||||
# TODO Make a nice message
|
||||
await ctx.respond("You have no ongoing events.")
|
||||
return
|
||||
|
||||
try:
|
||||
event: PycordEvent = await self.bot.find_event(event_id=user.current_event_id)
|
||||
stage: PycordEventStage = await self.bot.find_event_stage(user.current_stage_id)
|
||||
except (InvalidId, RuntimeError):
|
||||
# TODO Make a nice message
|
||||
await ctx.respond(
|
||||
"Your event could not be found. Please, report this issue to the event's management."
|
||||
)
|
||||
return
|
||||
|
||||
if answer.lower() != stage.answer.lower():
|
||||
# TODO Make a nice message
|
||||
await ctx.respond("Provided answer is wrong.")
|
||||
return
|
||||
|
||||
next_stage_index = stage.sequence + 1
|
||||
next_stage_id: ObjectId | None = (
|
||||
None if len(event.stage_ids) < next_stage_index + 1 else event.stage_ids[next_stage_index]
|
||||
)
|
||||
|
||||
if next_stage_id is None:
|
||||
# TODO Make a nice message
|
||||
user.completed_event_ids.append(event._id)
|
||||
|
||||
await user._set(
|
||||
self.bot.cache,
|
||||
current_event_id=None,
|
||||
current_stage_id=None,
|
||||
completed_event_ids=user.completed_event_ids,
|
||||
)
|
||||
|
||||
await ctx.respond("Event is completed.")
|
||||
return
|
||||
|
||||
next_stage: PycordEventStage = await self.bot.find_event_stage(next_stage_id)
|
||||
|
||||
files: List[File] | None = next_stage.get_media_files()
|
||||
|
||||
await ctx.respond(
|
||||
f"Provided answer is correct! Next stage...\n\n{next_stage.question}",
|
||||
files=files,
|
||||
)
|
||||
|
||||
await user.set_event_stage(next_stage._id, cache=self.bot.cache)
|
||||
|
||||
|
||||
def setup(bot: PycordBot) -> None:
|
||||
|
Reference in New Issue
Block a user