imcompiling
b9621a67c9
Add a discord max message length constant. Move boolean argument to not be positional.
209 lines
8.1 KiB
Python
209 lines
8.1 KiB
Python
"""Contained commands and views for record commands."""
|
|
|
|
import nextcord
|
|
from discord import TextInputStyle
|
|
from nextcord.ext import commands
|
|
|
|
from config import DakinyConfig
|
|
from utils import record_event_points
|
|
|
|
|
|
class RecordCog(commands.Cog, name="Record Commands"):
|
|
"""Class holds record commands."""
|
|
|
|
def __init__(self, bot: commands.Bot) -> None:
|
|
"""Default constructor for RecordCog class. Does nothing."""
|
|
super().__init__()
|
|
self.bot: commands.Bot = bot
|
|
|
|
@nextcord.slash_command()
|
|
async def record(self, interaction: nextcord.Interaction) -> None:
|
|
"""Parent slash command for record. Never called."""
|
|
|
|
@record.subcommand(description="Record an event and its participants.")
|
|
async def event(
|
|
self,
|
|
interaction: nextcord.Interaction,
|
|
name: str = nextcord.SlashOption(required=True, description="Name of the event that was hosted."),
|
|
host: nextcord.Member = nextcord.SlashOption(
|
|
required=False,
|
|
description="An alternative host if you are recording event for someone else.",
|
|
),
|
|
) -> None:
|
|
"""
|
|
Create and send recording event.
|
|
|
|
This command creates the base event recording embed and sends a message with it and the EventView
|
|
for gathering more information on the event.
|
|
|
|
Args:
|
|
interaction (nextcord.Interaction): interaction with this command.
|
|
name (str): Name of the event.
|
|
host (nextcord.Member, optional): Alternative host to user using command.
|
|
"""
|
|
# check role
|
|
if interaction.guild.get_role(self.bot.config.params.event_host_role_id) not in interaction.user.roles:
|
|
await interaction.response.send_message(
|
|
f"You must have the <@&{self.bot.config.params.event_host_role_id}> role to record an event.",
|
|
ephemeral=True,
|
|
)
|
|
return
|
|
|
|
if host is None:
|
|
host = interaction.user
|
|
|
|
embed: nextcord.Embed = nextcord.Embed(title=name, description="")
|
|
embed.add_field(name="Participants", value="")
|
|
embed.add_field(name="Non-Discord Participants", value="")
|
|
embed.add_field(name="Hosted by", value=f"<@{host.id}>", inline=False)
|
|
|
|
await interaction.response.send_message(
|
|
"Fill out event details and select all members who participated!",
|
|
view=EventView(name, self.bot.config, host.id),
|
|
ephemeral=True,
|
|
embed=embed,
|
|
)
|
|
|
|
|
|
class EventView(nextcord.ui.View):
|
|
"""
|
|
Class handles gathering information on an event.
|
|
|
|
Has buttons to edit the information, cancel, and submit.
|
|
Also contains a UserSelect object to pick discord members.
|
|
"""
|
|
|
|
def __init__(self, name: str, config: DakinyConfig, host_id: int):
|
|
super().__init__()
|
|
|
|
self.name: str = name
|
|
self.config: DakinyConfig = config
|
|
self.host_id: int = host_id
|
|
|
|
self.event_participants = ParticipantSelect()
|
|
|
|
self.add_item(self.event_participants)
|
|
|
|
@nextcord.ui.button(label="Cancel", style=nextcord.ButtonStyle.red, row=1)
|
|
async def cancel(self, button: nextcord.ui.Button, interaction: nextcord.Interaction) -> None: # noqa: ARG002
|
|
"""Cancels event submission. Removes embed and view."""
|
|
await interaction.response.edit_message("Canceled", ephemeral=True, embed=None, view=None)
|
|
self.stop()
|
|
|
|
@nextcord.ui.button(label="Edit", style=nextcord.ButtonStyle.blurple, row=1)
|
|
async def edit(self, button: nextcord.ui.Button, interaction: nextcord.Interaction) -> None: # noqa: ARG002
|
|
"""Calls modal to update the text input information of the event."""
|
|
await interaction.response.send_modal(ParticipantModal(interaction.message.embeds[0]))
|
|
|
|
@nextcord.ui.button(label="Confirm", style=nextcord.ButtonStyle.green, row=1)
|
|
async def confirm(self, button: nextcord.ui.Button, interaction: nextcord.Interaction) -> None: # noqa: ARG002
|
|
"""
|
|
Submits the event embed to the event channel.
|
|
|
|
Removes view and embed from recording message.
|
|
Sends point information to database.
|
|
"""
|
|
discord_users = self.event_participants.values.members # noqa: PD011 not a pandas object
|
|
if discord_users:
|
|
discord_users_list: list[str] = [str(user.id) for user in discord_users if user.id is not self.host_id]
|
|
record_event_points(
|
|
self.config,
|
|
discord_users_list,
|
|
True, # noqa: FBT003
|
|
self.config.params.event_participate_points,
|
|
)
|
|
|
|
non_discord_users = interaction.message.embeds[0].fields[1].value
|
|
if non_discord_users:
|
|
non_discord_users_list: list[str] = non_discord_users.split("\n")
|
|
record_event_points(
|
|
self.config,
|
|
non_discord_users_list,
|
|
self.config.params.event_participate_points,
|
|
is_member=False,
|
|
)
|
|
|
|
record_event_points(self.config, [self.host_id], self.config.params.event_host_points, is_member=True)
|
|
|
|
event_channel: nextcord.TextChannel = interaction.guild.get_channel(self.config.params.event_post_channel_id)
|
|
await event_channel.send(embed=interaction.message.embeds[0])
|
|
await interaction.response.edit_message(content="Event has been submitted.", view=None, embed=None)
|
|
|
|
self.stop()
|
|
|
|
|
|
class ParticipantModal(nextcord.ui.Modal):
|
|
"""
|
|
This modal contains all the text edits for an event.
|
|
|
|
You can change the name, description, and add the usernames of participants who are not in the discord.
|
|
"""
|
|
|
|
# consider turning this embed into a dict
|
|
def __init__(self, embed: nextcord.Embed):
|
|
super().__init__(
|
|
"Edit the event.",
|
|
timeout=5 * 60,
|
|
)
|
|
self.embed: nextcord.Embed = embed
|
|
self.name: nextcord.ui.TextInput = nextcord.ui.TextInput(
|
|
required=True,
|
|
label="Event Name",
|
|
default_value=self.embed.title,
|
|
)
|
|
self.description: nextcord.ui.TextInput = nextcord.ui.TextInput(
|
|
required=False,
|
|
label="Event Description",
|
|
default_value=self.embed.description,
|
|
style=TextInputStyle.paragraph,
|
|
placeholder="Add a description if desired.",
|
|
max_length=1000,
|
|
)
|
|
self.participants: nextcord.ui.TextInput = nextcord.ui.TextInput(
|
|
required=False,
|
|
label="Non-Discord Usernames",
|
|
style=TextInputStyle.paragraph,
|
|
placeholder="Please separate usernames with a new line.",
|
|
default_value=self.embed.fields[1].value,
|
|
)
|
|
self.add_item(self.name)
|
|
self.add_item(self.description)
|
|
self.add_item(self.participants)
|
|
|
|
async def callback(self, interaction: nextcord.Interaction) -> None:
|
|
"""
|
|
Edits the interaction message to contain the Non-Discord participants.
|
|
|
|
Args:
|
|
interaction (nextcord.Interaction): Interaction of the callback.
|
|
"""
|
|
self.embed.title = self.name.value
|
|
self.embed.description = self.description.value
|
|
self.embed.set_field_at(index=1, name="Non-Discord Participants", value=self.participants.value)
|
|
|
|
await interaction.response.edit_message(embed=self.embed)
|
|
|
|
|
|
class ParticipantSelect(nextcord.ui.UserSelect):
|
|
"""A selection drop down for discord members. Updates the embed with all users selected."""
|
|
|
|
def __init__(self) -> None:
|
|
super().__init__(
|
|
min_values=1,
|
|
max_values=25,
|
|
placeholder="Select all participants",
|
|
row=0,
|
|
)
|
|
|
|
async def callback(self, interaction: nextcord.Interaction) -> None:
|
|
"""
|
|
Edits the interaction message to contain the Discord participants.
|
|
|
|
Args:
|
|
interaction (nextcord.Interaction): Interaction of the callback.
|
|
"""
|
|
embed: nextcord.Embed = interaction.message.embeds[0]
|
|
participants: str = "\n".join([f"<@{participant.id}>" for participant in self.values.members])
|
|
embed.set_field_at(0, name="Participants", value=participants)
|
|
await interaction.response.edit_message(embed=embed)
|