Automatic reformat. Guessing from vscode update.

2023-11-07 15:35:43 -05:00
parent b0e3162123
commit ab88652cdb
1 changed files with 87 additions and 30 deletions
@@ -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)