Convert bot to use a submissions table.
This commit is contained in:
@@ -1,6 +1,10 @@
|
||||
"""View for verification buttons"""
|
||||
|
||||
import nextcord
|
||||
from sqlalchemy import select
|
||||
from sqlalchemy.orm import Session
|
||||
|
||||
from utilities.sql_data_classes import Submissions, engine
|
||||
|
||||
# pylint: disable=import-error
|
||||
from utilities.utils import submit_score
|
||||
@@ -17,33 +21,34 @@ class VerificationView(nextcord.ui.View):
|
||||
self, button: nextcord.ui.Button, interaction: nextcord.Interaction
|
||||
): # pylint: disable=unused-argument
|
||||
"""Button for denying score submission."""
|
||||
with Session(engine) as session:
|
||||
stmt = select(Submissions).where(Submissions.VerificationMessageId == interaction.message.id)
|
||||
submission = session.scalar(stmt)
|
||||
# let user know submission was denied
|
||||
|
||||
# TODO need to access bot here
|
||||
|
||||
# delete the submission
|
||||
session.delete(stmt)
|
||||
session.commit()
|
||||
await interaction.message.delete()
|
||||
|
||||
@nextcord.ui.button(label="Approve", style=nextcord.ButtonStyle.green, custom_id="approve-button-submission")
|
||||
async def approve_button(
|
||||
self, button: nextcord.ui.Button, interaction: nextcord.Interaction
|
||||
): # pylint: disable=unused-argument
|
||||
"""Button for approving score submission."""
|
||||
for embed in interaction.message.embeds:
|
||||
content = embed.to_dict()
|
||||
|
||||
for field in content["fields"]:
|
||||
if field["name"] == "GroupId":
|
||||
group_id = int(field["value"])
|
||||
elif field["name"] == "CategoryId":
|
||||
category_id = int(field["value"])
|
||||
elif field["name"] == "Score":
|
||||
score = field["value"]
|
||||
elif field["name"] == "User Name":
|
||||
user = field["value"]
|
||||
with Session(engine) as session:
|
||||
stmt = select(Submissions).where(Submissions.VerificationMessageId == interaction.message.id)
|
||||
submission = session.scalar(stmt)
|
||||
|
||||
try:
|
||||
|
||||
await submit_score(interaction, group_id, category_id, user, score)
|
||||
await submit_score(
|
||||
interaction, submission.GroupId, submission.CategoryId, submission.UserName, submission.Score
|
||||
)
|
||||
content = interaction.message.content
|
||||
content += f"\nApproved by <@{interaction.user.id}>."
|
||||
# remove the bot information embed. it should always be the first one.
|
||||
await interaction.message.edit(content=content, view=None, embed=interaction.message.embeds[0])
|
||||
await interaction.message.edit(content=content, view=None)
|
||||
# TODO send message to let user know submission was accepted. Need access to bot here.
|
||||
except UnboundLocalError:
|
||||
await interaction.send("Instruct user to submit with all fields.")
|
||||
await interaction.message.edit(view=None)
|
||||
|
||||
Reference in New Issue
Block a user