From 1b4c5780dfa8e9186b1cfbaec96c453c2ddb57a6 Mon Sep 17 00:00:00 2001 From: imcompiling Date: Tue, 25 Jul 2023 14:51:17 -0400 Subject: [PATCH] Refactor other function to use get_election_info. --- classes/election_cycle.py | 22 ++++++---------------- 1 file changed, 6 insertions(+), 16 deletions(-) diff --git a/classes/election_cycle.py b/classes/election_cycle.py index 057425f..671b47d 100644 --- a/classes/election_cycle.py +++ b/classes/election_cycle.py @@ -159,14 +159,9 @@ class ElectionCycle: """ message = self.build_vote_start() clan_member_role: nextcord.Role = self.guild.get_role(self.config.params.clan_member_role_id) - with Session(engine) as session: - stmt = select(ElectionInformation).where( - ElectionInformation.ElectionCycle == self.get_election_cycle_name() - ) - election_info = session.scalar(stmt) - election_channel = self.guild.get_channel(election_info.ElectionChannel) - await election_channel.send(message) - await election_channel.set_permissions(clan_member_role, send_messages=False) + election_channel = self.guild.get_channel(self.get_election_info().ElectionChannel) + await election_channel.send(message) + await election_channel.set_permissions(clan_member_role, send_messages=False) async def voting_end_action(self) -> None: """ @@ -174,14 +169,9 @@ class ElectionCycle: """ message = self.build_highcouncil_election_announce() council_elect_role: nextcord.Role = self.guild.get_role(self.config.params.clan_council_elect_role_id) - with Session(engine) as session: - stmt = select(ElectionInformation).where( - ElectionInformation.ElectionCycle == self.get_election_cycle_name() - ) - election_info = session.scalar(stmt) - election_channel = self.guild.get_channel(election_info.ElectionChannel) - await election_channel.send(message) - await election_channel.set_permissions(council_elect_role, send_messages=True) + election_channel = self.guild.get_channel(self.get_election_info().ElectionChannel) + await election_channel.send(message) + await election_channel.set_permissions(council_elect_role, send_messages=True) async def highcouncil_election_action(self) -> None: """