Refactor other function to use get_election_info.

This commit is contained in:
2023-07-25 14:51:17 -04:00
parent d67940aa26
commit 1b4c5780df
+6 -16
View File
@@ -159,14 +159,9 @@ class ElectionCycle:
""" """
message = self.build_vote_start() message = self.build_vote_start()
clan_member_role: nextcord.Role = self.guild.get_role(self.config.params.clan_member_role_id) clan_member_role: nextcord.Role = self.guild.get_role(self.config.params.clan_member_role_id)
with Session(engine) as session: election_channel = self.guild.get_channel(self.get_election_info().ElectionChannel)
stmt = select(ElectionInformation).where( await election_channel.send(message)
ElectionInformation.ElectionCycle == self.get_election_cycle_name() await election_channel.set_permissions(clan_member_role, send_messages=False)
)
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)
async def voting_end_action(self) -> None: async def voting_end_action(self) -> None:
""" """
@@ -174,14 +169,9 @@ class ElectionCycle:
""" """
message = self.build_highcouncil_election_announce() message = self.build_highcouncil_election_announce()
council_elect_role: nextcord.Role = self.guild.get_role(self.config.params.clan_council_elect_role_id) council_elect_role: nextcord.Role = self.guild.get_role(self.config.params.clan_council_elect_role_id)
with Session(engine) as session: election_channel = self.guild.get_channel(self.get_election_info().ElectionChannel)
stmt = select(ElectionInformation).where( await election_channel.send(message)
ElectionInformation.ElectionCycle == self.get_election_cycle_name() await election_channel.set_permissions(council_elect_role, send_messages=True)
)
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)
async def highcouncil_election_action(self) -> None: async def highcouncil_election_action(self) -> None:
""" """