Update model usage in verification view.
This commit is contained in:
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user