Allow users to view and delete tournaments

main
sgoudham 3 years ago
parent 642b1f5d75
commit a8c8b49fe9
Signed by: hammy
GPG Key ID: 44E818FD5457EEA4

@ -1,12 +1,19 @@
import asyncio
import logging
import interactions
import table2ascii
from table2ascii import PresetStyle
from exclamation_mark_charity.constants import GUILD_ID, OWNER_IDS
from exclamation_mark_charity.database import Database
from exclamation_mark_charity.logger_factory import LoggerFactory
def create_button(style: interactions.ButtonStyle, label: str, custom_id: str) -> interactions.Button:
return interactions.Button(style=style, label=label, custom_id=custom_id)
class Tournament(interactions.Extension):
def __init__(self, bot: interactions.Client):
self.bot: interactions.Client = bot
@ -17,45 +24,121 @@ class Tournament(interactions.Extension):
description="Tournament Command",
scope=GUILD_ID,
options=[
interactions.Option(
name="update",
description="Update The Tournament Details",
type=interactions.OptionType.SUB_COMMAND,
options=[]
),
interactions.Option(
name="create",
description="Create A !charity Tournament",
description="Create a !charity tournament",
type=interactions.OptionType.SUB_COMMAND,
options=[
interactions.Option(
name="name",
description="The name of the tournament",
name="title",
description="Title of the tournament",
type=interactions.OptionType.STRING,
required=True,
)
]
),
interactions.Option(
name="all",
description="View all existing tournaments",
type=interactions.OptionType.SUB_COMMAND,
options=[]
),
interactions.Option(
name="update",
description="Update the tournament details",
type=interactions.OptionType.SUB_COMMAND,
options=[]
),
interactions.Option(
name="delete",
description="Delete a tournament",
type=interactions.OptionType.SUB_COMMAND,
options=[]
)
]
)
async def tournament_command(self, ctx: interactions.CommandContext, sub_command, name: str = None):
async def tournament_command(self, ctx: interactions.CommandContext, sub_command, title: str = None):
# TODO: Remove when interactions.py implement application_command_permissions
if int(ctx.author.id) not in OWNER_IDS:
await ctx.send("**[ERROR 0069]** -> Not Allowed!")
return
await ctx.defer()
# Check function to ensure inputs are from the original user who invoked slash command
async def check(button_ctx):
if int(button_ctx.author.user.id) == int(ctx.author.user.id):
return True
await button_ctx.send("I wasn't asking you!", ephemeral=True)
return False
if sub_command == "create":
success, tournament_id = Database.create_tournament(name)
row = interactions.ActionRow(
components=[
create_button(interactions.ButtonStyle.PRIMARY, "Yes", "success"),
create_button(interactions.ButtonStyle.DANGER, "Cancel", "cancel")
]
)
message: interactions.Message = await ctx.send(
f"Create A Tournament With The Name **{title}**?", components=row)
try:
button_ctx: interactions.ComponentContext = await self.bot.wait_for_component(
components=row, check=check, timeout=25
)
if button_ctx.label == "Yes":
success, tournament_id = Database.create_tournament(title)
if success:
await ctx.send(f"Tournament **{name}** Successfully Created!")
self.logger.info(f"Tournament '{name}' Created Successfully With ID '{tournament_id}'!")
await message.edit(f"Tournament **{title}** Successfully Created!", components=None)
self.logger.info(f"Tournament '{title}' Created Successfully With ID '{tournament_id}'!")
else:
await message.edit("Tournament Creation Failed! **Check Logs!**", components=None)
else:
await message.edit("Tournament Creation Aborted!", components=None)
except asyncio.TimeoutError:
await message.edit("You didn't click it :(", components=None)
elif sub_command == "all":
success, all_tournaments = Database.view_all_tournaments()
if success:
output = table2ascii.table2ascii(
header=all_tournaments[0],
body=all_tournaments[1:],
style=PresetStyle.thick_compact
)
await ctx.send(f"```\n{output}\n```")
self.logger.info("All Tournaments Successfully Retrieved!")
else:
await ctx.send(f"Tournament Creation Failed! Check Logs!")
await ctx.send("Could Not List All Tournaments! **Check Logs!**")
elif sub_command == "update":
await ctx.send("**[INFO 0069]** Not Implemented Yet!")
elif sub_command == "delete":
success, all_tournaments = Database.view_all_tournaments()
tables = [table[0] for table in all_tournaments[1:]]
menu = interactions.SelectMenu(
custom_id="tournament_deletes_menu",
options=[interactions.SelectOption(label=table, value=table) for table in tables],
placeholder="Select Row(s)",
max_values=len(tables)
)
message: interactions.Message = await ctx.send(
f"Select Tournaments To **DELETE FOREVER**", components=menu)
try:
menu_ctx: interactions.ComponentContext = await self.bot.wait_for_component(
components=menu, check=check, timeout=25
)
success = Database.delete_many_tournaments([tuple([row]) for row in menu_ctx.data.values])
if success:
await message.edit("Chosen Table(s) Deleted", components=None)
else:
await message.edit("Table(s) Could Not Be Deleted! **Check Logs!**", components=None)
except asyncio.TimeoutError:
await message.edit("You didn't choose anything :(", components=None)
def setup(bot):
Tournament(bot)

Loading…
Cancel
Save