Update model usage in verification view.

This commit is contained in:
2025-03-03 23:14:31 -05:00
parent dbe652dcce
commit 528af14e00
+8 -7
View File
@@ -3,8 +3,10 @@
import nextcord
from sqlalchemy import delete, select
from sqlalchemy.orm import Session
from sqlmodel import col
from database.db import Submission, engine
from database.db import engine
from database.models import Submission
# pylint: disable=import-error
from utilities.utils import submit_score
@@ -22,8 +24,8 @@ class VerificationView(nextcord.ui.View):
with Session(engine) as session:
stmt = (
delete(Submission)
.where(Submission.GuildId == interaction.guild_id)
.where(Submission.MessageId == interaction.message.id)
.where(col(Submission.guild_id) == interaction.guild_id)
.where(col(Submission.message_id) == interaction.message.id)
)
session.execute(stmt)
session.commit()
@@ -35,8 +37,8 @@ class VerificationView(nextcord.ui.View):
with Session(engine) as session:
stmt = (
select(Submission)
.where(Submission.GuildId == interaction.guild_id)
.where(Submission.MessageId == interaction.message.id)
.where(col(Submission.guild_id) == interaction.guild_id)
.where(col(Submission.message_id) == interaction.message.id)
)
submission = session.scalar(stmt)
@@ -47,11 +49,10 @@ class VerificationView(nextcord.ui.View):
return
await submit_score(
interaction, submission.GroupId, submission.CategoryId, submission.UserName, submission.Score
interaction, submission.group_id, submission.category_id, 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.
await interaction.message.edit(content=content, view=None)