From 8e27e03709be0272bc4840a014a3fecbdd7c9f67 Mon Sep 17 00:00:00 2001 From: imcompiling Date: Sat, 16 Dec 2023 22:30:45 -0500 Subject: [PATCH] Need a message to track highcouncil votes. --- classes/election_cycle.py | 11 ++++++++++- database_setup/database.sql | 1 + 2 files changed, 11 insertions(+), 1 deletion(-) diff --git a/classes/election_cycle.py b/classes/election_cycle.py index a9e1789..f167cc1 100644 --- a/classes/election_cycle.py +++ b/classes/election_cycle.py @@ -297,7 +297,16 @@ class ElectionCycle: election_thread: nextcord.Thread = self.guild.get_thread( self.get_election_info().ThreadId ) - await election_thread.send(message) + vote_msg = await election_thread.send(message) + + with Session(engine) as session: + stmt = select(ElectionInformation).where( + ElectionInformation.ElectionCycle + == get_election_cycle_name(self.config) + ) + info = session.scalar(stmt) + info.HighCouncilVoteMsgId = vote_msg.id + session.commit() async def term_start_action(self) -> None: """ diff --git a/database_setup/database.sql b/database_setup/database.sql index 7d3b40b..e976e69 100644 --- a/database_setup/database.sql +++ b/database_setup/database.sql @@ -2,6 +2,7 @@ CREATE TABLE ElectionInformation( ElectionCycle varchar(30) not null unique, ThreadId bigint unsigned not null, CouncilVoteMsgId bigint unsigned, + HighCouncilVoteMsgId bigint unsigned, PRIMARY KEY (ElectionCycle) );