diff --git a/cogs/commands/test.py b/cogs/commands/test.py index a53fa9e..208e99b 100644 --- a/cogs/commands/test.py +++ b/cogs/commands/test.py @@ -5,7 +5,7 @@ from nextcord.ext import application_checks, commands from classes.election import Election from DakinySettings import settings -from utils import GUILDS, get_active_election_cycle +from utils import GUILDS, get_active_election_cycle, get_quarter_name ME = 206602713511886849 @@ -141,3 +141,15 @@ class TestCog(commands.Cog, name="Test commands."): ) await cycle.term_start_action() await interaction.send("Test command sent.", ephemeral=True) + + @test.subcommand() + @is_me() + async def get_quarter_name(self, interaction: nextcord.Interaction) -> None: + """ + Tests getting the quarter name. + + Args: + interaction (nextcord.Interaction): Interaction with the message. + """ + name = get_quarter_name() + await interaction.send(name) diff --git a/utils.py b/utils.py index 5bdf3af..66fda0d 100644 --- a/utils.py +++ b/utils.py @@ -48,10 +48,10 @@ def get_datetime(day: str) -> datetime: day = int(day.split("/")[1]) # ensure that year rolls over if current month is December and the next month is less than 12. - if today.month == calendar.DECEMBER and month < calendar.DECEMBER: + if today.month == calendar.DECEMBER and month < calendar.OCTOBER: year += 1 - if today.month == calendar.JANUARY and month > calendar.JANUARY: + if today.month == calendar.JANUARY and month > calendar.MARCH: year -= 1 return datetime(year, month, day, tzinfo=UTC)