diff --git a/cogs/submission/verification_view.py b/cogs/submission/verification_view.py index 7cdd522..37ac2ef 100644 --- a/cogs/submission/verification_view.py +++ b/cogs/submission/verification_view.py @@ -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)