cleaning up some math

communitycenter · Jun 1, 2024 · 5865c2c · 5865c2c
1 parent 6c283bb
commit 5865c2c
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/scripts/helpers/utils.py b/scripts/helpers/utils.py
@@ -274,16 +274,16 @@ def get_tv_airing_date(key: int) -> str:
     Returns:
         str: A string in the format "Spring 7, Year 1".
     """
-    seasons = ["Spring", "Summer", "Fall", "Winter"]
-    day = (key * 7) % 28
-    if day == 0:
-        day = 28
+    # Dropping key from 1 indexed to 0 indexed
+    key_idx = key - 1
+    day_key = key_idx % 4
+    day = (day_key + 1) * 7
 
-    season_idx = (key - 1) // 4
+    seasons = ["Spring", "Summer", "Fall", "Winter"]
+    season_idx = key_idx // 4
     season = seasons[season_idx % 4]
 
-    # Dropping key from 1 indexed to 0 indexed
-    year = (key - 1) // 16 + 1
+    year = key_idx // 16 + 1
 
     return f"{season} {day}, Year {year}"