imcompiling/dakiny-bot
1
0
Fork 0
Code Issues 22 Pull Requests 9 Actions Packages Projects Activity

Refactor other function to use get_election_info.

Browse Source
This commit is contained in:
imcompiling
2023-07-25 14:51:17 -04:00
parent d67940aa26
commit 1b4c5780df
1 changed files with 6 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files
classes/election_cycle.py
+2 -12
View File
@@ -159,12 +159,7 @@ 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(
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.send(message)
await election_channel.set_permissions(clan_member_role, send_messages=False) await election_channel.set_permissions(clan_member_role, send_messages=False)
@@ -174,12 +169,7 @@ 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(
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.send(message)
await election_channel.set_permissions(council_elect_role, send_messages=True) await election_channel.set_permissions(council_elect_role, send_messages=True)