Change bot to use update query.

2023-11-07 14:59:07 -05:00
parent 8c1eb8d11f
commit b0e3162123
1 changed files with 4 additions and 4 deletions
@@ -7,7 +7,7 @@ from typing import List
 import nextcord
 from nextcord import SelectOption
 from nextcord.interactions import Interaction
-from sqlalchemy import exc, select
+from sqlalchemy import exc, select, update
 from sqlalchemy.orm import Session

 from config import DakinyConfig
@@ -56,12 +56,12 @@ class VoteSelect(nextcord.ui.StringSelect):
            except exc.IntegrityError:
                session.rollback()
                stmt = (
-                    select(Votes)
+                    update(Votes)
                    .where(Votes.ElectionCycle == election_name)
                    .where(Votes.VoterId == interaction.user.id)
+                    .values(VotedForId=voted_for)
                )
-                vote = session.scalar(stmt)
-                vote.VotedForId = voted_for
+                session.execute(stmt)
                session.commit()

        vote_message_str = build_votes_str(self.config)