From ab88652cdb9d3b368b6bdc44205d7682e6e54e63 Mon Sep 17 00:00:00 2001 From: imcompiling Date: Tue, 7 Nov 2023 15:35:08 -0500 Subject: [PATCH] Automatic reformat. Guessing from vscode update. --- classes/election_cycle.py | 117 ++++++++++++++++++++++++++++---------- 1 file changed, 87 insertions(+), 30 deletions(-) diff --git a/classes/election_cycle.py b/classes/election_cycle.py index 3638217..fff8fd5 100644 --- a/classes/election_cycle.py +++ b/classes/election_cycle.py @@ -12,7 +12,12 @@ from sqlalchemy.orm import Session from classes.nominate_views import NominateView from classes.voting_views import VoteView from config import DakinyConfig -from database_setup.sql_data_classes import CouncilMembers, ElectionInformation, Votes, engine +from database_setup.sql_data_classes import ( + CouncilMembers, + ElectionInformation, + Votes, + engine, +) from utils import build_votes_str, get_datetime, get_election_cycle_name @@ -40,9 +45,15 @@ class ElectionCycle: self.term_end: datetime = get_datetime(self.election_cycle.term_end) self.voting_start: datetime = get_datetime(self.election_cycle.voting_start) self.voting_end: datetime = get_datetime(self.election_cycle.voting_end) - self.highcouncil_election_announce: datetime = get_datetime(self.election_cycle.highcouncil_election_announce) - self.highcouncil_election_start: datetime = get_datetime(self.election_cycle.highcouncil_election_start) - self.highcouncil_election_end: datetime = get_datetime(self.election_cycle.highcouncil_election_end) + self.highcouncil_election_announce: datetime = get_datetime( + self.election_cycle.highcouncil_election_announce + ) + self.highcouncil_election_start: datetime = get_datetime( + self.election_cycle.highcouncil_election_start + ) + self.highcouncil_election_end: datetime = get_datetime( + self.election_cycle.highcouncil_election_end + ) self.guild: nextcord.Guild = guild self.election_over: bool = False @@ -55,11 +66,15 @@ class ElectionCycle: long. """ - with open("bot_messages/election_announcement.txt", mode="r", encoding="utf8") as file: + with open( + "bot_messages/election_announcement.txt", mode="r", encoding="utf8" + ) as file: message = file.read() next_election = get_datetime( - self.config.election_cycles[(self.cycle + 1) % len(self.config.election_cycles)].election_start + self.config.election_cycles[ + (self.cycle + 1) % len(self.config.election_cycles) + ].election_start ) message = message.format( @@ -84,14 +99,22 @@ class ElectionCycle: str: Message to send. """ - with open("bot_messages/highcouncil_election.txt", mode="r", encoding="utf8") as file: + with open( + "bot_messages/highcouncil_election.txt", mode="r", encoding="utf8" + ) as file: message = file.read() return message.format( clan_council_elect_role_id=self.config.params.clan_council_elect_role_id, - highcouncil_election_announce=self.highcouncil_election_announce.strftime("%B %d"), - highcouncil_election_start=self.highcouncil_election_start.strftime("%B %d"), - highcouncil_election_end=self.highcouncil_election_end.strftime("%B %d"), + highcouncil_election_announce=self.highcouncil_election_announce.strftime( + "%B %d" + ), + highcouncil_election_start=self.highcouncil_election_start.strftime( + "%B %d" + ), + highcouncil_election_end=self.highcouncil_election_end.strftime( + "%B %d" + ), term_start=self.term_start.strftime("%B %d"), ) @@ -103,7 +126,9 @@ class ElectionCycle: str: Message to send. """ - with open("bot_messages/highcouncil_vote.txt", mode="r", encoding="utf8") as file: + with open( + "bot_messages/highcouncil_vote.txt", mode="r", encoding="utf8" + ) as file: message = file.read() return message.format( @@ -121,7 +146,9 @@ class ElectionCycle: with open("bot_messages/vote_start.txt", mode="r", encoding="utf8") as file: message = file.read() - vote_end: datetime = get_datetime(self.config.election_cycles[self.cycle].voting_end) + vote_end: datetime = get_datetime( + self.config.election_cycles[self.cycle].voting_end + ) return message.format( clan_member_id=self.config.params.clan_member_role_id, @@ -140,7 +167,11 @@ class ElectionCycle: await self.election_start_action() elif today.date() == self.voting_start.date(): await self.voting_start_action() - elif (self.voting_end - timedelta(3)).date() <= today.date() < self.voting_end.date(): + elif ( + (self.voting_end - timedelta(3)).date() + <= today.date() + < self.voting_end.date() + ): await self.election_reminder() elif today.date() == self.voting_end.date(): await self.voting_end_action() @@ -158,9 +189,9 @@ class ElectionCycle: quarter_name: str = get_election_cycle_name(self.config) # create new entry into database. - election_channel: nextcord.TextChannel = await category.create_text_channel(quarter_name) - clan_member_role: nextcord.Role = self.guild.get_role(self.config.params.clan_member_role_id) - await election_channel.set_permissions(clan_member_role, add_reactions=True) + clan_member_role: nextcord.Role = self.guild.get_role( + self.config.params.clan_member_role_id + ) with Session(engine) as session: new_cycle = ElectionInformation(ElectionCycle=quarter_name, ElectionChannel=election_channel.id) session.add(new_cycle) @@ -183,14 +214,21 @@ class ElectionCycle: Removes write permission from the general clan member role. """ 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) + 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) - vote_msg = await election_channel.send(build_votes_str(self.config), view=VoteView(self.config, self.guild)) + 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( - ElectionInformation.ElectionCycle == get_election_cycle_name(self.config) + ElectionInformation.ElectionCycle + == get_election_cycle_name(self.config) ) info = session.scalar(stmt) info.CouncilVoteMsgId = vote_msg.id @@ -204,9 +242,15 @@ 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) - election_channel: nextcord.TextChannel = self.guild.get_channel(self.get_election_info().ElectionChannel) - clan_member_role: nextcord.Role = self.guild.get_role(self.config.params.clan_member_role_id) + council_elect_role: nextcord.Role = self.guild.get_role( + self.config.params.clan_council_elect_role_id + ) + election_channel: nextcord.TextChannel = self.guild.get_channel( + self.get_election_info().ElectionChannel + ) + clan_member_role: nextcord.Role = self.guild.get_role( + self.config.params.clan_member_role_id + ) # remove vote button from vote message election_name = get_election_cycle_name(self.config) @@ -239,7 +283,9 @@ class ElectionCycle: await user.add_roles(council_elect_role) await election_channel.send(message) - await election_channel.set_permissions(council_elect_role, send_messages=True, add_reactions=True) + await election_channel.set_permissions( + council_elect_role, send_messages=True, add_reactions=True + ) await election_channel.set_permissions(clan_member_role, add_reactions=False) async def highcouncil_election_start_action(self) -> None: @@ -248,7 +294,9 @@ class ElectionCycle: """ message = self.build_highcouncil_vote_start() - election_channel: nextcord.TextChannel = self.guild.get_channel(self.get_election_info().ElectionChannel) + election_channel: nextcord.TextChannel = self.guild.get_channel( + self.get_election_info().ElectionChannel + ) await election_channel.send(message) async def term_start_action(self) -> None: @@ -257,8 +305,12 @@ class ElectionCycle: """ self.election_over = True - election_channel: nextcord.TextChannel = self.guild.get_channel(self.get_election_info().ElectionChannel) - archive: nextcord.CategoryChannel = self.guild.get_channel(self.config.params.archive_channel_group_id) + election_channel: nextcord.TextChannel = self.guild.get_channel( + self.get_election_info().ElectionChannel + ) + archive: nextcord.CategoryChannel = self.guild.get_channel( + self.config.params.archive_channel_group_id + ) await election_channel.move(beginning=True, category=archive) @@ -272,7 +324,8 @@ class ElectionCycle: with Session(engine) as session: stmt = select(ElectionInformation).where( - ElectionInformation.ElectionCycle == get_election_cycle_name(self.config) + ElectionInformation.ElectionCycle + == get_election_cycle_name(self.config) ) return session.scalar(stmt) @@ -281,9 +334,13 @@ class ElectionCycle: Sends a message to general chat reminder clan about election. """ - general: nextcord.TextChannel = self.guild.get_channel(self.config.params.general_channel_id) + general: nextcord.TextChannel = self.guild.get_channel( + self.config.params.general_channel_id + ) message = f"Reminder to <@&{self.config.params.clan_member_role_id}>! Clan elections are happening! Head over " - message += f"to <#{self.get_election_info().ElectionChannel}> to cast your vote!" + message += ( + f"to <#{self.get_election_info().ElectionChannel}> to cast your vote!" + ) await general.send(content=message)