update summer to get days for year

Author: onerandomusername

bot/exts/summer_aoc.py

@@ -11,13 +11,13 @@
 
 from bot.bot import SirRobin
 from bot.constants import Bot, Channels, Roles
+from bot.exts.advent_of_code._helpers import days_in_year
 from bot.utils.time import time_until
 
 log = logging.get_logger(__name__)
 
 # TODO: Add support for different years in accordance with AOC changes
 AOC_URL = "https://adventofcode.com/{year}/day/{day}"
-LAST_DAY = 25
 FIRST_YEAR = 2015
 LAST_YEAR = arrow.get().year - 1
 PUBLIC_NAME = "Revival of Code"
@@ -64,8 +64,9 @@ def __init__(self, bot: SirRobin):
         self.wait_task: asyncio.Task | None = None
         self.loop_task: tasks.Loop | None = None
 
-        self.is_running = False
+        self.is_running: bool = False
         self.year: int | None = None
+        self.days_this_year: int | None = None
         self.current_day: int | None = None
         self.day_interval: int | None = None
         self.post_time = 0
@@ -146,6 +147,7 @@ async def start(self, ctx: commands.Context, year: int, day_interval: int, post_
 
         self.is_running = True
         self.year = year
+        self.days_this_year = days_in_year(year)
         self.current_day = 1
         self.day_interval = day_interval
         self.post_time = post_time
@@ -175,8 +177,8 @@ async def force_day(self, ctx: commands.Context, day: int, now: Literal["now"] |
             await ctx.send(embed=embed)
             return
 
-        if not 1 <= day <= LAST_DAY:
-            raise commands.BadArgument(f"Start day must be between 1 and {LAST_DAY}, inclusive")
+        if not 1 <= day <= self.days_this_year:
+            raise commands.BadArgument(f"Start day must be between 1 and {self.days_this_year}, inclusive")
 
         log.info(f"Setting the current day of Summer AoC to {day}")
         await self.stop_event()
@@ -188,7 +190,7 @@ async def force_day(self, ctx: commands.Context, day: int, now: Literal["now"] |
 
         embed = self.get_info_embed()
         if now:
-            if self.current_day > LAST_DAY:
+            if self.current_day > self.days_this_year:
                 title = "Puzzle posted and event is now ending"
             else:
                 title = "Puzzle posted and event is now running"
@@ -198,7 +200,7 @@ async def force_day(self, ctx: commands.Context, day: int, now: Literal["now"] |
         embed.title = title
         embed.color = discord.Color.green()
         await ctx.send(embed=embed)
-        if self.current_day <= LAST_DAY:
+        if self.current_day <= self.days_this_year:
             await self.start_event()
 
     @summer_aoc_group.command(name="stop")
@@ -291,7 +293,7 @@ async def stop_event(self) -> bool:
 
     async def post_puzzle(self) -> None:
         """Create a thread for the current day's puzzle."""
-        if self.current_day > LAST_DAY:
+        if self.current_day > self.days_this_year:
             log.error("Attempted to post puzzle after last day, stopping event")
             await self.stop_event()
             return
@@ -306,7 +308,7 @@ async def post_puzzle(self) -> None:
 
         self.current_day += 1
         await self.save_event_state()
-        if self.current_day > LAST_DAY:
+        if self.current_day > self.days_this_year:
             await self.stop_event()
 
     def get_info_embed(self) -> discord.Embed:
@@ -326,7 +328,7 @@ def get_info_embed(self) -> discord.Embed:
 
     def get_puzzle_embed(self) -> discord.Embed:
         """Generate an embed for the day's puzzle post."""
-        if self.current_day == LAST_DAY:
+        if self.current_day == self.days_this_year:
             next_puzzle_text = LAST_PUZZLE_TEXT.format(timestamp=int(arrow.get(REAL_AOC_START).timestamp()))
         else:
             next_puzzle_text = NEXT_PUZZLE_TEXT.format(