From dcea91156470fc905358933ea93cfe64f85a6525 Mon Sep 17 00:00:00 2001 From: imcompiling Date: Fri, 3 Nov 2023 12:05:36 -0400 Subject: [PATCH] Fix bug getting ElectionInformation object correctly. --- classes/election_cycle.py | 8 +++++--- classes/voting_views.py | 2 +- 2 files changed, 6 insertions(+), 4 deletions(-) diff --git a/classes/election_cycle.py b/classes/election_cycle.py index 210c2f3..2688e19 100644 --- a/classes/election_cycle.py +++ b/classes/election_cycle.py @@ -183,11 +183,13 @@ class ElectionCycle: message = self.build_vote_start() clan_member_role: nextcord.Role = self.guild.get_role(self.config.params.clan_member_role_id) election_channel: nextcord.TextChannel = self.guild.get_channel(self.get_election_info().ElectionChannel) - await election_channel.send(message, view=VoteView(self.config, self.guild)) - vote_msg = await election_channel.send(build_votes_str(self.config)) + await election_channel.send(message) + vote_msg = await election_channel.send(build_votes_str(self.config), view=VoteView(self.config, self.guild)) with Session(engine) as session: - stmt = select(ElectionInformation).where(ElectionCycle == get_election_cycle_name(self.config)) + stmt = select(ElectionInformation).where( + ElectionInformation.ElectionCycle == get_election_cycle_name(self.config) + ) info = session.scalar(stmt) info.CouncilVoteMsgId = vote_msg.id session.commit() diff --git a/classes/voting_views.py b/classes/voting_views.py index 9c34966..542eb95 100644 --- a/classes/voting_views.py +++ b/classes/voting_views.py @@ -30,7 +30,7 @@ class VoteSelect(nextcord.ui.StringSelect): vote_message_str = build_votes_str(self.config) with Session(engine) as session: stmt = select(ElectionInformation.CouncilVoteMsgId).where( - ElectionCycle == get_election_cycle_name(self.config) + ElectionInformation.ElectionCycle == get_election_cycle_name(self.config) ) vote_msg_id = session.scalar(stmt) vote_msg = await interaction.channel.fetch_message(vote_msg_id)