Compare commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
a8697b0187 | ||
|
|
8c8dc2c638 | ||
|
|
54933b427e | ||
|
|
5d05006a23 | ||
|
|
2596b9fa7b | ||
|
|
8c654612a9 | ||
|
|
e836ead399 | ||
|
|
fa691ced5d |
@@ -239,7 +239,8 @@ class ElectionCycle:
|
||||
)
|
||||
await election_thread.send(message)
|
||||
vote_msg = await election_thread.send(
|
||||
build_votes_str(self.config), view=VoteView(self.config, self.guild)
|
||||
build_votes_str(self.config, self.guild),
|
||||
view=VoteView(self.config, self.guild),
|
||||
)
|
||||
|
||||
with Session(engine) as session:
|
||||
@@ -324,7 +325,7 @@ class ElectionCycle:
|
||||
)
|
||||
await election_thread.send(message)
|
||||
vote_msg = await election_thread.send(
|
||||
build_highcouncil_votes_str(self.config),
|
||||
build_highcouncil_votes_str(self.config, self.guild),
|
||||
view=VoteHighcouncilView(self.config, self.guild),
|
||||
)
|
||||
|
||||
|
||||
@@ -82,7 +82,7 @@ class VoteSelect(nextcord.ui.StringSelect):
|
||||
session.execute(stmt)
|
||||
session.commit()
|
||||
|
||||
vote_message_str = build_highcouncil_votes_str(self.config)
|
||||
vote_message_str = build_highcouncil_votes_str(self.config, interaction.guild)
|
||||
vote_msg: nextcord.Message = await interaction.channel.fetch_message(
|
||||
vote_msg_id
|
||||
)
|
||||
@@ -159,7 +159,10 @@ class VoteHighcouncilView(nextcord.ui.View):
|
||||
else f"{self.guild.get_member(nominee).display_name}",
|
||||
value=str(nominee),
|
||||
)
|
||||
for nominee in session.scalars(stmt)
|
||||
for nominee in sorted(
|
||||
session.scalars(stmt),
|
||||
key=lambda nominee: self.guild.get_member(nominee).display_name,
|
||||
)
|
||||
]
|
||||
options.append(
|
||||
SelectOption(
|
||||
@@ -232,7 +235,10 @@ class VoteHighcouncilView(nextcord.ui.View):
|
||||
else f"{self.guild.get_member(nominee).display_name}",
|
||||
value=str(nominee),
|
||||
)
|
||||
for nominee in session.scalars(stmt)
|
||||
for nominee in sorted(
|
||||
session.scalars(stmt),
|
||||
key=lambda nominee: self.guild.get_member(nominee).display_name,
|
||||
)
|
||||
]
|
||||
options.append(
|
||||
SelectOption(
|
||||
|
||||
@@ -64,7 +64,7 @@ class VoteSelect(nextcord.ui.StringSelect):
|
||||
session.execute(stmt)
|
||||
session.commit()
|
||||
|
||||
vote_message_str = build_votes_str(self.config)
|
||||
vote_message_str = build_votes_str(self.config, interaction.guild)
|
||||
vote_msg: nextcord.Message = await interaction.channel.fetch_message(
|
||||
vote_msg_id
|
||||
)
|
||||
@@ -135,7 +135,10 @@ class VoteView(nextcord.ui.View):
|
||||
else f"{self.guild.get_member(nominee).display_name}",
|
||||
value=str(nominee),
|
||||
)
|
||||
for nominee in session.scalars(stmt)
|
||||
for nominee in sorted(
|
||||
session.scalars(stmt),
|
||||
key=lambda nominee: self.guild.get_member(nominee).display_name,
|
||||
)
|
||||
]
|
||||
options.append(
|
||||
SelectOption(
|
||||
|
||||
@@ -49,7 +49,7 @@ class RequestCog(commands.Cog, name="Request Commands"):
|
||||
"""
|
||||
|
||||
config: DakinyConfig = self.bot.config
|
||||
content = f'<@{interaction.user.id}> would like to add presence" {presence}."'
|
||||
content = f'<@{interaction.user.id}> would like to add presence "{presence}."'
|
||||
view = PresenceChange(presence)
|
||||
await self.bot.get_channel(config.params.admin_channel_id).send(
|
||||
content=content, view=view
|
||||
|
||||
@@ -7,12 +7,7 @@ from nextcord.ext import commands
|
||||
from sqlalchemy import select
|
||||
from sqlalchemy.orm import Session
|
||||
|
||||
from database_setup.sql_data_classes import (
|
||||
ElectionInformation,
|
||||
Presences,
|
||||
Votes,
|
||||
engine,
|
||||
)
|
||||
from database_setup.sql_data_classes import Presences, Votes, engine
|
||||
|
||||
|
||||
class ViewCog(commands.Cog, name="View Commands"):
|
||||
@@ -121,7 +116,7 @@ class ViewCog(commands.Cog, name="View Commands"):
|
||||
"""
|
||||
|
||||
with Session(engine) as session:
|
||||
stmt = select(ElectionInformation.ElectionCycle)
|
||||
stmt = select(Votes.ElectionCycle).distinct()
|
||||
elections = session.scalars(stmt)
|
||||
|
||||
# if empty compare string
|
||||
|
||||
@@ -79,6 +79,10 @@ class DailyLoop(commands.Cog, name="Daily Loop"):
|
||||
config: DakinyConfig = self.bot.config
|
||||
cycle = get_current_cycle(config)
|
||||
|
||||
# prevent exception from using cycle as an Int in an array
|
||||
if cycle is None:
|
||||
return None
|
||||
|
||||
return ElectionCycle(cycle, config, self.bot.get_guild(config.params.guild_id))
|
||||
|
||||
def update_birthdays(self) -> None:
|
||||
@@ -107,10 +111,14 @@ class DailyLoop(commands.Cog, name="Daily Loop"):
|
||||
|
||||
today = datetime.today()
|
||||
guild: nextcord.Guild = self.bot.get_guild(self.bot.config.params.guild_id)
|
||||
announce_channel: nextcord.TextChannel = guild.get_channel(self.bot.config.params.general_channel_id)
|
||||
announce_channel: nextcord.TextChannel = guild.get_channel(
|
||||
self.bot.config.params.general_channel_id
|
||||
)
|
||||
|
||||
cakedays: list[nextcord.Member] = [
|
||||
human for human in guild.humans if human.joined_at.month == today.month and human.joined_at.day == today.day
|
||||
human
|
||||
for human in guild.humans
|
||||
if human.joined_at.month == today.month and human.joined_at.day == today.day
|
||||
]
|
||||
|
||||
for member in cakedays:
|
||||
|
||||
@@ -4,6 +4,7 @@ File contains util functions.
|
||||
|
||||
from datetime import datetime
|
||||
|
||||
import nextcord
|
||||
from sqlalchemy import select
|
||||
from sqlalchemy.orm import Session
|
||||
|
||||
@@ -31,6 +32,9 @@ def get_datetime(day: str) -> datetime:
|
||||
if datetime.today().month == 12 and month < 12:
|
||||
year += 1
|
||||
|
||||
if datetime.today().month == 1 and month > 1:
|
||||
year -= 1
|
||||
|
||||
return datetime(year, month, day)
|
||||
|
||||
|
||||
@@ -88,7 +92,7 @@ def get_current_cycle(config: DakinyConfig) -> int:
|
||||
config (DakinyConfig): Config object to use
|
||||
|
||||
Returns:
|
||||
int: cycle index for active election. -1 for no active cycle
|
||||
int: cycle index for active election. None for no active cycle
|
||||
"""
|
||||
|
||||
today = datetime.today()
|
||||
@@ -100,15 +104,16 @@ def get_current_cycle(config: DakinyConfig) -> int:
|
||||
):
|
||||
return cycle
|
||||
|
||||
return -1
|
||||
return None
|
||||
|
||||
|
||||
def build_votes_str(config: DakinyConfig) -> str:
|
||||
def build_votes_str(config: DakinyConfig, guild: nextcord.Guild) -> str:
|
||||
"""
|
||||
Creates a string that contains all nominees and their votes to be sent as a discord message.
|
||||
|
||||
Args:
|
||||
config (DakinyConfig): Daking config object
|
||||
guild (nextcord.Guild): The guild object. Used to get display names for sorting.
|
||||
|
||||
Returns:
|
||||
str: String that contains all nominees and their votes.
|
||||
@@ -121,7 +126,10 @@ def build_votes_str(config: DakinyConfig) -> str:
|
||||
CouncilMembers.ElectionCycle == get_election_cycle_name(config)
|
||||
)
|
||||
|
||||
for nominee in session.scalars(stmt):
|
||||
for nominee in sorted(
|
||||
session.scalars(stmt),
|
||||
key=lambda nominee: guild.get_member(nominee.MemberId).display_name,
|
||||
):
|
||||
votes[nominee.MemberId] = 0
|
||||
|
||||
stmt = select(Votes).where(
|
||||
@@ -138,7 +146,7 @@ def build_votes_str(config: DakinyConfig) -> str:
|
||||
return votes_str
|
||||
|
||||
|
||||
def build_highcouncil_votes_str(config: DakinyConfig) -> str:
|
||||
def build_highcouncil_votes_str(config: DakinyConfig, guild: nextcord.Guild) -> str:
|
||||
"""
|
||||
Creates a string that contains all high council nominees and their votes to be sent as a discord message.
|
||||
|
||||
@@ -147,6 +155,7 @@ def build_highcouncil_votes_str(config: DakinyConfig) -> str:
|
||||
|
||||
Returns:
|
||||
str: String that contains all nominees and their votes.
|
||||
guild (nextcord.Guild): The guild object. Used to get display names for sorting.
|
||||
"""
|
||||
|
||||
cl_votes: dict[int, int] = {}
|
||||
@@ -159,7 +168,10 @@ def build_highcouncil_votes_str(config: DakinyConfig) -> str:
|
||||
.filter(CouncilMembers.ClanLeaderSpeechId.is_not(None))
|
||||
)
|
||||
|
||||
for nominee in session.scalars(stmt):
|
||||
for nominee in sorted(
|
||||
session.scalars(stmt),
|
||||
key=lambda nominee: guild.get_member(nominee.MemberId).display_name,
|
||||
):
|
||||
cl_votes[nominee.MemberId] = 0
|
||||
|
||||
stmt = select(Votes).where(Votes.ElectionCycle == get_cl_election_name(config))
|
||||
@@ -174,7 +186,10 @@ def build_highcouncil_votes_str(config: DakinyConfig) -> str:
|
||||
.filter(CouncilMembers.DiscordManagerSpeechId.is_not(None))
|
||||
)
|
||||
|
||||
for nominee in session.scalars(stmt):
|
||||
for nominee in sorted(
|
||||
session.scalars(stmt),
|
||||
key=lambda nominee: guild.get_member(nominee.MemberId).display_name,
|
||||
):
|
||||
dm_votes[nominee.MemberId] = 0
|
||||
|
||||
stmt = select(Votes).where(Votes.ElectionCycle == get_dm_election_name(config))
|
||||
|
||||
Reference in New Issue
Block a user