imcompiling/agon
1
0
Fork 0
Code Issues 17 Pull Requests 4 Actions Packages Activity

Update model usage in verification view.

Browse Source
This commit is contained in:
imcompiling
2025-03-03 23:14:31 -05:00
parent dbe652dcce
commit 528af14e00
1 changed files with 8 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
cogs/submission/verification_view.py
+8 -7
View File
@@ -3,8 +3,10 @@
import nextcord import nextcord
from sqlalchemy import delete, select from sqlalchemy import delete, select
from sqlalchemy.orm import Session 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 # pylint: disable=import-error
from utilities.utils import submit_score from utilities.utils import submit_score
@@ -22,8 +24,8 @@ class VerificationView(nextcord.ui.View):
with Session(engine) as session: with Session(engine) as session:
stmt = ( stmt = (
delete(Submission) delete(Submission)
.where(Submission.GuildId == interaction.guild_id) .where(col(Submission.guild_id) == interaction.guild_id)
.where(Submission.MessageId == interaction.message.id) .where(col(Submission.message_id) == interaction.message.id)
) )
session.execute(stmt) session.execute(stmt)
session.commit() session.commit()
@@ -35,8 +37,8 @@ class VerificationView(nextcord.ui.View):
with Session(engine) as session: with Session(engine) as session:
stmt = ( stmt = (
select(Submission) select(Submission)
.where(Submission.GuildId == interaction.guild_id) .where(col(Submission.guild_id) == interaction.guild_id)
.where(Submission.MessageId == interaction.message.id) .where(col(Submission.message_id) == interaction.message.id)
) )
submission = session.scalar(stmt) submission = session.scalar(stmt)
@@ -47,11 +49,10 @@ class VerificationView(nextcord.ui.View):
return return
await submit_score( 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.delete(submission)
session.commit() session.commit()
content = interaction.message.content content = interaction.message.content
content += f"\nApproved by <@{interaction.user.id}>." 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) await interaction.message.edit(content=content, view=None)