From 3e6cb29425ada2056fefe1f334b4d40c2f197101 Mon Sep 17 00:00:00 2001
From: Chris Smith <ole088@gmail.com>
Date: Thu, 5 Mar 2026 18:40:53 -0500
Subject: [PATCH] Update check_table_name function to pass tests

---
 util/drawing.py | 24 +++++++++++++++++++++---
 1 file changed, 21 insertions(+), 3 deletions(-)

diff --git a/util/drawing.py b/util/drawing.py
index b99952a..ef1265f 100644
--- a/util/drawing.py
+++ b/util/drawing.py
@@ -3,17 +3,35 @@
 
     Utility functions for drawing data processing
 """
+import os
+
 from datetime import datetime, timedelta
+from dotenv import load_dotenv
 
 from data.database import PowerballDraw, MegaMillionsDraw
 
+load_dotenv()
 
-def check_table_name(table_name: str) -> bool:
+
+def check_table_name(table_name: str) -> dict | None:
     """
     :param table_name: table name
-    :return: True if table name is valid
+    :return: dict of game info or None
     """
-    return table_name in ['PowerballDraw', 'MegaMillionsDraw']
+    game_dict: dict = {
+        'PowerballDraw': {
+            'game_name': 'powerball',
+            'db_table_name': 'powerball_draws',
+            'inception_year': os.getenv('LP_POWERBALL_INCEPTION'),
+        },
+        'MegaMillionsDraw': {
+            'game_name': 'mega-millions',
+            'db_table_name': 'mega_millions_draws',
+            'inception_year': os.getenv('LP_MEGA_MILLIONS_INCEPTION'),
+        },
+    }
+
+    return game_dict[table_name] if table_name in game_dict.keys() else None
 
 
 def get_next_draw_date(game: str | None, from_date: str | None = None) -> str | None: