From 67090e0d6fbc3310504afc6ce366354a9bbcd030 Mon Sep 17 00:00:00 2001 From: imcompiling Date: Thu, 20 Feb 2025 17:53:14 -0500 Subject: [PATCH] Remove submission from table once handled by admins. --- cogs/submission/verification_view.py | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) diff --git a/cogs/submission/verification_view.py b/cogs/submission/verification_view.py index 1d8dd3d..df9b2ee 100644 --- a/cogs/submission/verification_view.py +++ b/cogs/submission/verification_view.py @@ -1,7 +1,7 @@ """View for verification buttons.""" import nextcord -from sqlalchemy import select +from sqlalchemy import delete, select from sqlalchemy.orm import Session from database_setup.sql_data_classes import Submission, engine @@ -19,6 +19,14 @@ class VerificationView(nextcord.ui.View): @nextcord.ui.button(label="Deny", style=nextcord.ButtonStyle.red, custom_id="deny-button-submission") async def deny_button(self, button: nextcord.ui.Button, interaction: nextcord.Interaction) -> None: # noqa: ARG002 """Button for denying score submission.""" + with Session(engine) as session: + stmt = ( + delete(Submission) + .where(Submission.GuildId == interaction.guild_id) + .where(Submission.MessageId == interaction.message.id) + ) + session.execute(stmt) + session.commit() await interaction.message.delete() @nextcord.ui.button(label="Approve", style=nextcord.ButtonStyle.green, custom_id="approve-button-submission") @@ -41,6 +49,8 @@ class VerificationView(nextcord.ui.View): await submit_score( interaction, submission.GroupId, submission.CategoryId, submission.UserName, submission.Score ) + session.delete(submission) + session.commit() content = interaction.message.content content += f"\nApproved by <@{interaction.user.id}>." # remove the bot information embed. it should always be the first one.